fix brekages

Author: gaborfeher

tests/contrib/tasks/test_vertex_task_store.py

@@ -545,38 +545,43 @@ async def test_metadata_empty_transitions(
     task = Task(
         id=task_id,
         context_id='session-meta-empty',
-        status=TaskStatus(state=TaskState.submitted),
-        kind='task',
+        status=TaskStatus(state=TaskState.TASK_STATE_SUBMITTED),
         metadata={},
     )
-    await vertex_store.save(task)
+    await vertex_store.save(task, ServerCallContext())
 
     full_name = f'{vertex_store._agent_engine_resource_id}/a2aTasks/{task_id}'
 
     # Get initial event sequence number
     stored_task_before = (
-        await vertex_store._client.aio.agent_engines.a2a_tasks.get(full_name)
+        await vertex_store._client.aio.agent_engines.a2a_tasks.get(
+            name=full_name
+        )
     )
     initial_seq = stored_task_before.next_event_sequence_number
 
     # Step 2: Update metadata to None
-    updated_task = task.model_copy(deep=True)
-    updated_task.metadata = None
-    await vertex_store.save(updated_task)
+    updated_task = Task()
+    updated_task.CopyFrom(task)
+    updated_task.metadata.Clear()
+    await vertex_store.save(updated_task, ServerCallContext())
 
     # Step 3: Update back to {}
-    task_back = updated_task.model_copy(deep=True)
+    task_back = Task()
+    task_back.CopyFrom(updated_task)
     task_back.metadata = {}
-    await vertex_store.save(task_back)
+    await vertex_store.save(task_back, ServerCallContext())
 
     # Verify that retrieved task still has {} (due to mapping)
-    retrieved = await vertex_store.get(task_id)
+    retrieved = await vertex_store.get(task_id, ServerCallContext())
     assert retrieved is not None
     assert retrieved.metadata == {}
 
     # Verify that next_event_sequence_number did NOT increase (no events generated)
     stored_task_after = (
-        await vertex_store._client.aio.agent_engines.a2a_tasks.get(full_name)
+        await vertex_store._client.aio.agent_engines.a2a_tasks.get(
+            name=full_name
+        )
     )
     assert stored_task_after.next_event_sequence_number == initial_seq