feat(a2a): add persistent/redis context store for a2a server

Author: daltonnyx

AgentCrew/app.py

@@ -172,6 +172,8 @@ def run_server(
         api_key: Optional[str] = None,
         mcp_config: Optional[str] = None,
         memory_llm: Optional[str] = None,
+        store_type: str = "memory",
+        store_options: Optional[dict] = None,
     ) -> None:
         from AgentCrew.modules.a2a.server import A2AServer
         from AgentCrew.modules.mcpclient import MCPSessionManager
@@ -209,6 +211,8 @@ def run_server(
                 port=port,
                 base_url=base_url,
                 api_key=api_key,
+                store_type=store_type,
+                store_options=store_options,
             )
 
             click.echo(f"Starting A2A server on {host}:{port}")

AgentCrew/main.py

@@ -311,6 +311,17 @@ def chat(
 @common_options
 @click.option("--model-id", default=None, help="Model ID from provider")
 @click.option("--api-key", default=None, help="API key for authentication (optional)")
+@click.option(
+    "--store-type",
+    default="memory",
+    type=click.Choice(["memory", "file", "redis"]),
+    help="Task store backend: memory, file, or redis",
+)
+@click.option(
+    "--store-option",
+    multiple=True,
+    help="Store options as key=value pairs (e.g. --store-option base_dir=./data --store-option redis_url=redis://localhost:6379)",
+)
 def a2a_server(
     host,
     port,
@@ -322,13 +333,21 @@ def a2a_server(
     memory_path,
     model_id,
     api_key,
+    store_type,
+    store_option,
 ):
     """Start an A2A server exposing all SwissKnife agents"""
     from AgentCrew.app import AgentCrewApplication
 
     if memory_path:
         os.environ["MEMORYDB_PATH"] = memory_path
 
+    store_options = {}
+    for opt in store_option:
+        if "=" in opt:
+            k, v = opt.split("=", 1)
+            store_options[k.strip()] = v.strip()
+
     app = AgentCrewApplication()
     app.run_server(
         host=host,
@@ -340,6 +359,8 @@ def a2a_server(
         api_key=api_key,
         mcp_config=mcp_config,
         memory_llm=memory_llm,
+        store_type=store_type,
+        store_options=store_options if store_options else None,
     )
 
 

AgentCrew/main_docker.py

@@ -99,6 +99,17 @@ def chat(provider, agent_config, mcp_config, memory_llm, memory_path):
 @common_options
 @click.option("--model-id", default=None, help="Model ID from provider")
 @click.option("--api-key", default=None, help="API key for authentication (optional)")
+@click.option(
+    "--store-type",
+    default="memory",
+    type=click.Choice(["memory", "file", "redis"]),
+    help="Task store backend: memory, file, or redis",
+)
+@click.option(
+    "--store-option",
+    multiple=True,
+    help="Store options as key=value pairs (e.g. --store-option base_dir=./data)",
+)
 def a2a_server(
     host,
     port,
@@ -110,13 +121,21 @@ def a2a_server(
     memory_path,
     model_id,
     api_key,
+    store_type,
+    store_option,
 ):
     """Start an A2A server exposing all SwissKnife agents"""
     from AgentCrew.app import AgentCrewApplication
 
     if memory_path:
         os.environ["MEMORYDB_PATH"] = memory_path
 
+    store_options = {}
+    for opt in store_option:
+        if "=" in opt:
+            k, v = opt.split("=", 1)
+            store_options[k.strip()] = v.strip()
+
     app = AgentCrewApplication()
     app.run_server(
         host=host,
@@ -128,6 +147,8 @@ def a2a_server(
         api_key=api_key,
         mcp_config=mcp_config,
         memory_llm=memory_llm,
+        store_type=store_type,
+        store_options=store_options if store_options else None,
     )
 
 

AgentCrew/modules/a2a/__init__.py

@@ -4,5 +4,19 @@
 """
 
 from .server import A2AServer
+from .task_store import (
+    TaskStore,
+    InMemoryTaskStore,
+    FileTaskStore,
+    RedisTaskStore,
+    create_task_store,
+)
 
-__all__ = ["A2AServer"]
+__all__ = [
+    "A2AServer",
+    "TaskStore",
+    "InMemoryTaskStore",
+    "FileTaskStore",
+    "RedisTaskStore",
+    "create_task_store",
+]

AgentCrew/modules/a2a/server.py

@@ -44,6 +44,8 @@ def __init__(
         port: int = 41241,
         base_url: Optional[str] = None,
         api_key: Optional[str] = None,
+        store_type: str = "memory",
+        store_options: Optional[dict] = None,
     ):
         logger.info(f"Initializing A2A server with host={host}, port={port}")
         self.agent_manager = agent_manager
@@ -55,13 +57,14 @@ def __init__(
 
         self.exposed_url = os.getenv("A2A_SERVER_EXPOSED_URL", self.base_url)
 
-        # Create agent registry
         self.agent_registry = AgentRegistry(agent_manager, self.exposed_url)
 
-        # Create task manager
-        self.task_manager = MultiAgentTaskManager(agent_manager)
+        self.task_manager = MultiAgentTaskManager(
+            agent_manager,
+            store_type=store_type,
+            store_options=store_options,
+        )
 
-        # Create Starlette app
         self.app = self._create_app()
 
     def _create_app(self) -> Starlette: