feat: Bump a2a-python version (#24)

Contains pkg update to latest https://github.com/a2aproject/a2a-python
and all the required changes

---------

Signed-off-by: Janos Sarusi-Kis <janossk@cisco.com>

Author: janosSarusiKis

examples/echo_agent/client.py

@@ -10,10 +10,10 @@
 from a2a.client import (
     A2ACardResolver,
     Client,
-    create_text_message_object,
     minimal_agent_card,
 )
-from a2a.types.a2a_pb2 import AgentCard, SendMessageRequest
+from a2a.helpers import new_text_message
+from a2a.types.a2a_pb2 import AgentCard, Role, SendMessageRequest
 from a2a.utils.constants import (
     AGENT_CARD_WELL_KNOWN_PATH,
 )
@@ -160,25 +160,25 @@ async def send_message(
     client: Client,
     text: str,
 ) -> str:
-    message = create_text_message_object(content=text)
+    message = new_text_message(text, role=Role.ROLE_USER)
     request = SendMessageRequest(message=message)
 
     output = ""
     try:
-        async for stream_response, task in client.send_message(request=request):
+        async for stream_response in client.send_message(request=request):
             which = stream_response.WhichOneof("payload")
             if which == "message":
                 for part in stream_response.message.parts:
                     if part.WhichOneof("content") == "text":
                         output += part.text
             elif which == "task":
-                if task:
-                    logger.info(f"task ({task.id}) status: {task.status.state}")
-                    if task.artifacts:
-                        for artifact in task.artifacts:
-                            for part in artifact.parts:
-                                if part.WhichOneof("content") == "text":
-                                    output += part.text
+                task = stream_response.task
+                logger.info(f"task ({task.id}) status: {task.status.state}")
+                if task.artifacts:
+                    for artifact in task.artifacts:
+                        for part in artifact.parts:
+                            if part.WhichOneof("content") == "text":
+                                output += part.text
             elif which == "artifact_update":
                 artifact = stream_response.artifact_update.artifact
                 for part in artifact.parts:
@@ -198,7 +198,7 @@ async def send_message_multicast(
     client: MulticastClient,
     text: str,
 ) -> None:
-    message = create_text_message_object(content=text)
+    message = new_text_message(text, role=Role.ROLE_USER)
     request = SendMessageRequest(message=message)
 
     try:

examples/echo_agent/echo_agent_executor.py

@@ -1,5 +1,6 @@
 import logging
 
+from a2a.helpers import new_task_from_user_message
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events import EventQueue
 from a2a.server.tasks.task_updater import TaskUpdater
@@ -26,12 +27,18 @@ async def execute(
 
         logging.debug(f"received message: {context.message}")
 
+        # The V2 request handler requires an initial Task to be enqueued
+        # before any status/artifact update events are emitted.
+        task = context.current_task
+        if task is None:
+            task = new_task_from_user_message(context.message)
+            await event_queue.enqueue_event(task)
+
         task_updater = TaskUpdater(
             event_queue=event_queue,
-            task_id=context.message.task_id,
-            context_id=context.message.context_id,
+            task_id=task.id,
+            context_id=task.context_id,
         )
-        await task_updater.submit(message=context.message)
 
         if context.message.parts[0].WhichOneof("content") != "text":
             raise Exception("only text parts are supported")

examples/echo_agent/server.py

@@ -52,6 +52,7 @@ async def main() -> None:
     default_request_handler = DefaultRequestHandler(
         agent_executor=agent_executor,
         task_store=task_store,
+        agent_card=agent_card,
     )
 
     match args.type:
@@ -91,14 +92,23 @@ async def main() -> None:
             await server.serve_async()
         case "starlette":
             import uvicorn
-            from a2a.server.apps import A2AStarletteApplication
+            from a2a.server.routes import (
+                create_agent_card_routes,
+                create_jsonrpc_routes,
+            )
+            from starlette.applications import Starlette
 
-            servicer = A2AStarletteApplication(
-                agent_card=agent_card,
-                http_handler=default_request_handler,
+            routes = create_agent_card_routes(agent_card=agent_card)
+            routes += create_jsonrpc_routes(
+                request_handler=default_request_handler,
+                rpc_url="/",
             )
+            app = Starlette(routes=routes)
 
-            uvicorn.run(servicer.build(), host="0.0.0.0", port=9999)
+            # main() already runs inside asyncio.run(), so use the awaitable
+            # uvicorn Server API instead of the blocking uvicorn.run().
+            config = uvicorn.Config(app, host="0.0.0.0", port=9999)
+            await uvicorn.Server(config).serve()
         case _:
             raise ValueError(f"Invalid server type: {args.type}")
 

examples/travel_planner_agent/agent_executor.py

@@ -1,3 +1,4 @@
+from a2a.helpers import new_text_artifact
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events import EventQueue
 from a2a.types import (
@@ -6,7 +7,6 @@
     TaskStatus,
     TaskStatusUpdateEvent,
 )
-from a2a.utils import new_text_artifact
 from typing_extensions import override
 
 from examples.travel_planner_agent.agent import TravelPlannerAgent

examples/travel_planner_agent/client.py

@@ -14,10 +14,10 @@
 from a2a.client import (
     Client,
     ClientFactory,
-    create_text_message_object,
     minimal_agent_card,
 )
-from a2a.types.a2a_pb2 import SendMessageRequest
+from a2a.helpers import new_text_message
+from a2a.types.a2a_pb2 import Role, SendMessageRequest
 
 from slima2a import setup_slim_client
 from slima2a.client_transport import (
@@ -43,19 +43,20 @@ async def interact_with_server(client: Client) -> None:
             print("bye!~")
             break
 
-        message = create_text_message_object(content=user_input)
+        message = new_text_message(user_input, role=Role.ROLE_USER)
         request = SendMessageRequest(message=message)
 
         output = ""
         try:
-            async for stream_response, task in client.send_message(request=request):
+            async for stream_response in client.send_message(request=request):
                 which = stream_response.WhichOneof("payload")
                 if which == "message":
                     for part in stream_response.message.parts:
                         if part.WhichOneof("content") == "text":
                             output += part.text
                 elif which == "task":
-                    if task and task.artifacts:
+                    task = stream_response.task
+                    if task.artifacts:
                         for artifact in task.artifacts:
                             for part in artifact.parts:
                                 if part.WhichOneof("content") == "text":

examples/travel_planner_agent/server.py

@@ -49,6 +49,7 @@ async def main() -> None:
     request_handler = DefaultRequestHandler(
         agent_executor=TravelPlannerAgentExecutor(),
         task_store=InMemoryTaskStore(),
+        agent_card=agent_card,
     )
 
     # Initialize and connect to SLIM

pyproject.toml

@@ -8,7 +8,7 @@ description = "A2A protocol over slimrpc"
 readme = "README.md"
 requires-python = ">=3.10, <4.0"
 dependencies = [
-    "a2a-sdk[telemetry,sqlite]==1.0.0-alpha.0",
+    "a2a-sdk[telemetry,sqlite]==1.1.0",
     "slim-bindings>=1.4,<2",
 ]
 classifiers = [
@@ -34,6 +34,7 @@ Changelog = "https://github.com/agntcy/slim/blob/main/data-plane/python/integrat
 linting = ["ruff>=0.12", "mypy>=1.17.0", "types-protobuf>=6.30.2.20250822"]
 testing = ["pytest>=8.3.4"]
 examples = [
+    "a2a-sdk[http-server]",
     "langchain-core",
     "langchain-openai",
     "typing-extensions",

slima2a/client_transport.py

@@ -267,9 +267,8 @@ class MultiAgentClientFactory(ClientFactory):
     def __init__(
         self,
         config: ClientConfig,
-        consumers: list | None = None,
     ) -> None:
-        super().__init__(config, consumers)
+        super().__init__(config)
         self._config: ClientConfig = config
         self._multiagent_labels: set[str] = set()
         self.register("slimrpc", SRPCTransport.create, multiagent=True)  # type: ignore[arg-type]
@@ -287,14 +286,13 @@ def register(  # type: ignore[override]
     def create(  # type: ignore[override]
         self,
         card: AgentCard | list[AgentCard],
-        consumers: list | None = None,
         interceptors: list | None = None,
     ) -> "Client | MulticastClient":
         if not isinstance(card, list):
-            return super().create(card, consumers, interceptors)
+            return super().create(card, interceptors)
 
         if len(card) == 1:
-            return super().create(card[0], consumers, interceptors)
+            return super().create(card[0], interceptors)
 
         protocol = self._find_common_multiagent_protocol(card)
         producer = self._registry[protocol]

slima2a/handler.py

@@ -13,12 +13,11 @@
     get_requested_extensions,
 )
 from a2a.server.context import ServerCallContext
-from a2a.server.request_handlers.request_handler import RequestHandler
+from a2a.server.request_handlers.request_handler import RequestHandler, validate
 from a2a.types import a2a_pb2
 from a2a.types.a2a_pb2 import AgentCard
 from a2a.utils import proto_utils
 from a2a.utils.errors import A2AError, TaskNotFoundError
-from a2a.utils.helpers import validate, validate_async_generator
 from google.protobuf import empty_pb2
 from google.rpc import code_pb2
 
@@ -132,7 +131,7 @@ async def SendMessage(
             await self.raise_error_response(e)
         return a2a_pb2.SendMessageResponse()
 
-    @validate_async_generator(
+    @validate(
         lambda self: self.agent_card.capabilities.streaming,
         "Streaming is not supported by the agent",
     )
@@ -168,7 +167,7 @@ async def CancelTask(
             await self.raise_error_response(e)
         return a2a_pb2.Task()
 
-    @validate_async_generator(
+    @validate(
         lambda self: self.agent_card.capabilities.streaming,
         "Streaming is not supported by the agent",
     )