revert unwanted changes

Author: guglielmo-san

src/a2a/contrib/tasks/vertex_task_converter.py

@@ -1,4 +1,5 @@
 try:
+    from google.genai import types as genai_types
     from vertexai import types as vertexai_types
 except ImportError as e:
     raise ImportError(
@@ -25,70 +26,70 @@
 
 
 _TO_SDK_TASK_STATE = {
-    vertexai_types.State.STATE_UNSPECIFIED: TaskState.unknown,
-    vertexai_types.State.SUBMITTED: TaskState.submitted,
-    vertexai_types.State.WORKING: TaskState.working,
-    vertexai_types.State.COMPLETED: TaskState.completed,
-    vertexai_types.State.CANCELLED: TaskState.canceled,
-    vertexai_types.State.FAILED: TaskState.failed,
-    vertexai_types.State.REJECTED: TaskState.rejected,
-    vertexai_types.State.INPUT_REQUIRED: TaskState.input_required,
-    vertexai_types.State.AUTH_REQUIRED: TaskState.auth_required,
+    vertexai_types.A2aTaskState.STATE_UNSPECIFIED: TaskState.unknown,
+    vertexai_types.A2aTaskState.SUBMITTED: TaskState.submitted,
+    vertexai_types.A2aTaskState.WORKING: TaskState.working,
+    vertexai_types.A2aTaskState.COMPLETED: TaskState.completed,
+    vertexai_types.A2aTaskState.CANCELLED: TaskState.canceled,
+    vertexai_types.A2aTaskState.FAILED: TaskState.failed,
+    vertexai_types.A2aTaskState.REJECTED: TaskState.rejected,
+    vertexai_types.A2aTaskState.INPUT_REQUIRED: TaskState.input_required,
+    vertexai_types.A2aTaskState.AUTH_REQUIRED: TaskState.auth_required,
 }
 
 _SDK_TO_STORED_TASK_STATE = {v: k for k, v in _TO_SDK_TASK_STATE.items()}
 
 
-def to_sdk_task_state(stored_state: vertexai_types.State) -> TaskState:
+def to_sdk_task_state(stored_state: vertexai_types.A2aTaskState) -> TaskState:
     """Converts a proto A2aTask.State to a TaskState enum."""
     return _TO_SDK_TASK_STATE.get(stored_state, TaskState.unknown)
 
 
-def to_stored_task_state(task_state: TaskState) -> vertexai_types.State:
+def to_stored_task_state(task_state: TaskState) -> vertexai_types.A2aTaskState:
     """Converts a TaskState enum to a proto A2aTask.State enum value."""
     return _SDK_TO_STORED_TASK_STATE.get(
-        task_state, vertexai_types.State.STATE_UNSPECIFIED
+        task_state, vertexai_types.A2aTaskState.STATE_UNSPECIFIED
     )
 
 
-def to_stored_part(part: Part) -> vertexai_types.Part:
+def to_stored_part(part: Part) -> genai_types.Part:
     """Converts a SDK Part to a proto Part."""
     if isinstance(part.root, TextPart):
-        return vertexai_types.Part(text=part.root.text)
+        return genai_types.Part(text=part.root.text)
     if isinstance(part.root, DataPart):
         data_bytes = json.dumps(part.root.data).encode('utf-8')
-        return vertexai_types.Part(
-            inline_data=vertexai_types.Blob(
+        return genai_types.Part(
+            inline_data=genai_types.Blob(
                 mime_type='application/json', data=data_bytes
             )
         )
     if isinstance(part.root, FilePart):
         file_content = part.root.file
         if isinstance(file_content, FileWithBytes):
             decoded_bytes = base64.b64decode(file_content.bytes)
-            return vertexai_types.Part(
-                inline_data=vertexai_types.Blob(
+            return genai_types.Part(
+                inline_data=genai_types.Blob(
                     mime_type=file_content.mime_type or '', data=decoded_bytes
                 )
             )
         if isinstance(file_content, FileWithUri):
-            return vertexai_types.Part(
-                file_data=vertexai_types.FileData(
+            return genai_types.Part(
+                file_data=genai_types.FileData(
                     mime_type=file_content.mime_type or '',
                     file_uri=file_content.uri,
                 )
             )
     raise ValueError(f'Unsupported part type: {type(part.root)}')
 
 
-def to_sdk_part(stored_part: vertexai_types.Part) -> Part:
+def to_sdk_part(stored_part: genai_types.Part) -> Part:
     """Converts a proto Part to a SDK Part."""
     if stored_part.text:
         return Part(root=TextPart(text=stored_part.text))
     if stored_part.inline_data:
-        encoded_bytes = base64.b64encode(stored_part.inline_data.data).decode(
-            'utf-8'
-        )
+        encoded_bytes = base64.b64encode(
+            stored_part.inline_data.data or b''
+        ).decode('utf-8')
         return Part(
             root=FilePart(
                 file=FileWithBytes(
@@ -97,7 +98,7 @@ def to_sdk_part(stored_part: vertexai_types.Part) -> Part:
                 )
             )
         )
-    if stored_part.file_data:
+    if stored_part.file_data and stored_part.file_data.file_uri:
         return Part(
             root=FilePart(
                 file=FileWithUri(