Fix A2A unit testing and QueryRequest after rebase

Author: luis5tb

src/app/endpoints/a2a.py

@@ -307,6 +307,7 @@ async def _process_task_streaming(  # pylint: disable=too-many-locals
             no_tools=False,
             generate_topic_summary=True,
             media_type=None,
+            vector_store_ids=None,
         )
 
         # Get LLM client and select model

src/app/main.py

@@ -17,6 +17,7 @@
 from configuration import configuration
 from log import get_logger
 from models.responses import InternalServerErrorResponse
+from a2a_storage import A2AStorageFactory
 from utils.common import register_mcp_servers_async
 from utils.llama_stack_version import check_llama_stack_version
 
@@ -53,6 +54,10 @@ async def lifespan(_app: FastAPI) -> AsyncIterator[None]:
 
     yield
 
+    # Cleanup resources on shutdown
+    await A2AStorageFactory.cleanup()
+    logger.info("App shutdown complete")
+
 
 app = FastAPI(
     title=f"{service_name} service - OpenAPI",

tests/unit/models/config/test_dump_configuration.py

@@ -520,6 +520,10 @@ def test_dump_configuration_with_quota_limiters(tmp_path: Path) -> None:
                 },
                 "enable_token_history": True,
             },
+            "a2a_state": {
+                "sqlite": None,
+                "postgres": None,
+            },
         }
 
 
@@ -627,6 +631,7 @@ def test_dump_configuration_with_quota_limiters_different_values(
             "service": {
                 "host": "localhost",
                 "port": 8080,
+                "base_url": None,
                 "auth_enabled": False,
                 "workers": 1,
                 "color_log": True,
@@ -919,6 +924,10 @@ def test_dump_configuration_byok(tmp_path: Path) -> None:
                 },
                 "enable_token_history": False,
             },
+            "a2a_state": {
+                "sqlite": None,
+                "postgres": None,
+            },
         }
 
 
@@ -1001,6 +1010,7 @@ def test_dump_configuration_pg_namespace(tmp_path: Path) -> None:
             "service": {
                 "host": "localhost",
                 "port": 8080,
+                "base_url": None,
                 "auth_enabled": False,
                 "workers": 1,
                 "color_log": True,