chore: Enhance a2a part converters

a. fix binary data conversion
b. support thoughts, code execution result, executable codes conversion

PiperOrigin-RevId: 775827259

Author: seanzhougoogle

src/google/adk/a2a/converters/part_converter.py

@@ -18,6 +18,7 @@
 
 from __future__ import annotations
 
+import base64
 import json
 import logging
 import sys
@@ -43,8 +44,11 @@
 logger = logging.getLogger('google_adk.' + __name__)
 
 A2A_DATA_PART_METADATA_TYPE_KEY = 'type'
+A2A_DATA_PART_METADATA_IS_LONG_RUNNING_KEY = 'is_long_running'
 A2A_DATA_PART_METADATA_TYPE_FUNCTION_CALL = 'function_call'
 A2A_DATA_PART_METADATA_TYPE_FUNCTION_RESPONSE = 'function_response'
+A2A_DATA_PART_METADATA_TYPE_CODE_EXECUTION_RESULT = 'code_execution_result'
+A2A_DATA_PART_METADATA_TYPE_EXECUTABLE_CODE = 'executable_code'
 
 
 @working_in_progress
@@ -67,7 +71,8 @@ def convert_a2a_part_to_genai_part(
     elif isinstance(part.file, a2a_types.FileWithBytes):
       return genai_types.Part(
           inline_data=genai_types.Blob(
-              data=part.file.bytes.encode('utf-8'), mime_type=part.file.mimeType
+              data=base64.b64decode(part.file.bytes),
+              mime_type=part.file.mimeType,
           )
       )
     else:
@@ -84,7 +89,11 @@ def convert_a2a_part_to_genai_part(
     # response.
     # TODO once A2A defined how to suervice such information, migrate below
     # logic accordinlgy
-    if part.metadata and A2A_DATA_PART_METADATA_TYPE_KEY in part.metadata:
+    if (
+        part.metadata
+        and _get_adk_metadata_key(A2A_DATA_PART_METADATA_TYPE_KEY)
+        in part.metadata
+    ):
       if (
           part.metadata[_get_adk_metadata_key(A2A_DATA_PART_METADATA_TYPE_KEY)]
           == A2A_DATA_PART_METADATA_TYPE_FUNCTION_CALL
@@ -103,6 +112,24 @@ def convert_a2a_part_to_genai_part(
                 part.data, by_alias=True
             )
         )
+      if (
+          part.metadata[_get_adk_metadata_key(A2A_DATA_PART_METADATA_TYPE_KEY)]
+          == A2A_DATA_PART_METADATA_TYPE_CODE_EXECUTION_RESULT
+      ):
+        return genai_types.Part(
+            code_execution_result=genai_types.CodeExecutionResult.model_validate(
+                part.data, by_alias=True
+            )
+        )
+      if (
+          part.metadata[_get_adk_metadata_key(A2A_DATA_PART_METADATA_TYPE_KEY)]
+          == A2A_DATA_PART_METADATA_TYPE_EXECUTABLE_CODE
+      ):
+        return genai_types.Part(
+            executable_code=genai_types.ExecutableCode.model_validate(
+                part.data, by_alias=True
+            )
+        )
     return genai_types.Part(text=json.dumps(part.data))
 
   logger.warning(
@@ -118,27 +145,40 @@ def convert_genai_part_to_a2a_part(
     part: genai_types.Part,
 ) -> Optional[a2a_types.Part]:
   """Convert a Google GenAI Part to an A2A Part."""
+
   if part.text:
-    return a2a_types.TextPart(text=part.text)
+    a2a_part = a2a_types.TextPart(text=part.text)
+    if part.thought is not None:
+      a2a_part.metadata = {_get_adk_metadata_key('thought'): part.thought}
+    return a2a_types.Part(root=a2a_part)
 
   if part.file_data:
-    return a2a_types.FilePart(
-        file=a2a_types.FileWithUri(
-            uri=part.file_data.file_uri,
-            mimeType=part.file_data.mime_type,
+    return a2a_types.Part(
+        root=a2a_types.FilePart(
+            file=a2a_types.FileWithUri(
+                uri=part.file_data.file_uri,
+                mimeType=part.file_data.mime_type,
+            )
         )
     )
 
   if part.inline_data:
-    return a2a_types.Part(
-        root=a2a_types.FilePart(
-            file=a2a_types.FileWithBytes(
-                bytes=part.inline_data.data,
-                mimeType=part.inline_data.mime_type,
-            )
+    a2a_part = a2a_types.FilePart(
+        file=a2a_types.FileWithBytes(
+            bytes=base64.b64encode(part.inline_data.data).decode('utf-8'),
+            mimeType=part.inline_data.mime_type,
         )
     )
 
+    if part.video_metadata:
+      a2a_part.metadata = {
+          _get_adk_metadata_key(
+              'video_metadata'
+          ): part.video_metadata.model_dump(by_alias=True, exclude_none=True)
+      }
+
+    return a2a_types.Part(root=a2a_part)
+
   # Conver the funcall and function reponse to A2A DataPart.
   # This is mainly for converting human in the loop and auth request and
   # response.
@@ -151,9 +191,9 @@ def convert_genai_part_to_a2a_part(
                 by_alias=True, exclude_none=True
             ),
             metadata={
-                A2A_DATA_PART_METADATA_TYPE_KEY: (
-                    A2A_DATA_PART_METADATA_TYPE_FUNCTION_CALL
-                )
+                _get_adk_metadata_key(
+                    A2A_DATA_PART_METADATA_TYPE_KEY
+                ): A2A_DATA_PART_METADATA_TYPE_FUNCTION_CALL
             },
         )
     )
@@ -165,9 +205,37 @@ def convert_genai_part_to_a2a_part(
                 by_alias=True, exclude_none=True
             ),
             metadata={
-                A2A_DATA_PART_METADATA_TYPE_KEY: (
-                    A2A_DATA_PART_METADATA_TYPE_FUNCTION_RESPONSE
-                )
+                _get_adk_metadata_key(
+                    A2A_DATA_PART_METADATA_TYPE_KEY
+                ): A2A_DATA_PART_METADATA_TYPE_FUNCTION_RESPONSE
+            },
+        )
+    )
+
+  if part.code_execution_result:
+    return a2a_types.Part(
+        root=a2a_types.DataPart(
+            data=part.code_execution_result.model_dump(
+                by_alias=True, exclude_none=True
+            ),
+            metadata={
+                _get_adk_metadata_key(
+                    A2A_DATA_PART_METADATA_TYPE_KEY
+                ): A2A_DATA_PART_METADATA_TYPE_CODE_EXECUTION_RESULT
+            },
+        )
+    )
+
+  if part.executable_code:
+    return a2a_types.Part(
+        root=a2a_types.DataPart(
+            data=part.executable_code.model_dump(
+                by_alias=True, exclude_none=True
+            ),
+            metadata={
+                _get_adk_metadata_key(
+                    A2A_DATA_PART_METADATA_TYPE_KEY
+                ): A2A_DATA_PART_METADATA_TYPE_EXECUTABLE_CODE
             },
         )
     )