Enhance agent_to_a2a with optional services

src/google/adk/a2a/utils/agent_to_a2a.py

@@ -36,9 +36,13 @@
 from starlette.applications import Starlette
 
 from ...agents.base_agent import BaseAgent
+from ...artifacts.base_artifact_service import BaseArtifactService
 from ...artifacts.in_memory_artifact_service import InMemoryArtifactService
+from ...auth.credential_service.base_credential_service import BaseCredentialService
 from ...auth.credential_service.in_memory_credential_service import InMemoryCredentialService
 from ...cli.utils.logs import setup_adk_logger
+from ...memory.base_memory_service import BaseMemoryService
+from ...sessions.base_session_service import BaseSessionService
 from ...memory.in_memory_memory_service import InMemoryMemoryService
 from ...runners import Runner
 from ...sessions.in_memory_session_service import InMemorySessionService
@@ -90,6 +94,10 @@ def to_a2a(
     port: int = 8000,
     protocol: str = "http",
     agent_card: Optional[Union[AgentCard, str]] = None,
+    artifact_service: Optional[BaseArtifactService] = None,
+    credential_service: Optional[BaseCredentialService] = None,
+    memory_service: Optional[BaseMemoryService] = None,
+    session_service: Optional[BaseSessionService] = None,
 ) -> Starlette:
   """Convert an ADK agent to a A2A Starlette application.
 
@@ -101,7 +109,15 @@ def to_a2a(
       agent_card: Optional pre-built AgentCard object or path to agent card
                   JSON. If not provided, will be built automatically from the
                   agent.
-
+      artifact_service: Service for artifact management (file storage, logs, etc.).
+                        Defaults to in-memory artifact service.
+      credential_service: Service for authentication/credential management.
+                        Defaults to in-memory credential service.
+      memory_service: Service for conversation or workspace memory.
+                        Defaults to in-memory memory service.
+      session_service: Service for session management.
+                        Defaults to in-memory session service.
+
   Returns:
       A Starlette application that can be run with uvicorn
 
@@ -121,11 +137,10 @@ async def create_runner() -> Runner:
     return Runner(
         app_name=agent.name or "adk_agent",
         agent=agent,
-        # Use minimal services - in a real implementation these could be configured
-        artifact_service=InMemoryArtifactService(),
-        session_service=InMemorySessionService(),
-        memory_service=InMemoryMemoryService(),
-        credential_service=InMemoryCredentialService(),
+        artifact_service=artifact_service or InMemoryArtifactService(),
+        session_service=session_service or InMemorySessionService(),
+        memory_service=memory_service or InMemoryMemoryService(),
+        credential_service=credential_service or InMemoryCredentialService(),
     )
 
   # Create A2A components