Fixes aftert #978

Author: ishymko

tests/integration/cross_version/client_server/server_0_3.py

@@ -38,7 +38,7 @@
 from starlette.requests import Request
 from starlette.concurrency import iterate_in_threadpool
 import time
-from a2a.helpers.proto_helpers import new_task_from_user_message
+from a2a.utils.task import new_task
 from server_common import CustomLoggingMiddleware
 
 
@@ -49,7 +49,7 @@ def __init__(self):
     async def execute(self, context: RequestContext, event_queue: EventQueue):
         print(f'SERVER: execute called for task {context.task_id}')
 
-        task = new_task_from_user_message(context.message)
+        task = new_task(context.message)
         task.id = context.task_id
         task.context_id = context.context_id
         task.status.state = TaskState.working

tests/server/request_handlers/test_default_request_handler_v2.py

@@ -73,7 +73,9 @@ def create_default_agent_card():
 class MockAgentExecutor(AgentExecutor):
     async def execute(self, context: RequestContext, event_queue: EventQueue):
         if context.message:
-            await event_queue.enqueue_event(new_task(context.message))
+            await event_queue.enqueue_event(
+                new_task_from_user_message(context.message)
+            )
 
         task_updater = TaskUpdater(
             event_queue,
@@ -967,7 +969,9 @@ async def execute(
         self, context: RequestContext, event_queue: EventQueue
     ) -> None:
         if context.message:
-            await event_queue.enqueue_event(new_task(context.message))
+            await event_queue.enqueue_event(
+                new_task_from_user_message(context.message)
+            )
         updater = TaskUpdater(
             event_queue,
             task_id=context.task_id or str(uuid.uuid4()),