solace-agent-mesh 1.6.1__py3-none-any.whl → 1.6.2__py3-none-any.whl

solace_agent_mesh/gateway/http_sse/component.py

@@ -115,6 +115,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
             self.ssl_keyfile = self.get_config("ssl_keyfile", "")
             self.ssl_certfile = self.get_config("ssl_certfile", "")
             self.ssl_keyfile_password = self.get_config("ssl_keyfile_password", "")
+            self.model_config = self.get_config("model", None)
 
             log.info(
                 "%s WebUI-specific configuration retrieved (Host: %s, Port: %d).",
@@ -437,7 +438,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
                     raise RuntimeError(
                         "Visualization flow setup error: BrokerInput not found."
                     )
-                log.info(
+                log.debug(
                     "%s Obtained reference to internal BrokerInput component.",
                     log_id_prefix,
                 )
@@ -883,7 +884,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
         Manages global subscription reference counts.
         """
         log_id_prefix = f"{self.log_identifier}[AddVizSub:{stream_id}]"
-        log.info(
+        log.debug(
             "%s Attempting to add subscription to topic: %s", log_id_prefix, topic_str
         )
 
@@ -1207,7 +1208,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
             self.task_logger_service = TaskLoggerService(
                 session_factory=session_factory, config=task_logging_config
             )
-            log.info(
+            log.debug(
                 "%s Services dependent on database session factory have been initialized.",
                 self.log_identifier,
             )
@@ -1238,14 +1239,14 @@ class WebUIBackendComponent(BaseGatewayComponent):
                 )
                 try:
                     self.fastapi_event_loop = asyncio.get_running_loop()
-                    log.info(
+                    log.debug(
                         "%s [_start_listener] Captured FastAPI event loop via startup event: %s",
                         self.log_identifier,
                         self.fastapi_event_loop,
                     )
 
                     if self.fastapi_event_loop:
-                        log.info(
+                        log.debug(
                             "%s Ensuring visualization flow is running...",
                             self.log_identifier,
                         )
@@ -1255,7 +1256,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
                             self._visualization_processor_task is None
                             or self._visualization_processor_task.done()
                         ):
-                            log.info(
+                            log.debug(
                                 "%s Starting visualization message processor task.",
                                 self.log_identifier,
                             )
@@ -1265,7 +1266,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
                                 )
                             )
                         else:
-                            log.info(
+                            log.debug(
                                 "%s Visualization message processor task already running.",
                                 self.log_identifier,
                             )
@@ -1812,7 +1813,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
         for agent_name in agents_to_deregister:
             self._deregister_agent(agent_name)
             
-        log.info(
+        log.debug(
             "%s Agent health check completed. Total agents: %d, De-registered: %d",
             self.log_identifier,
             total_agents,
@@ -2052,6 +2053,13 @@ class WebUIBackendComponent(BaseGatewayComponent):
             )
             return
 
+        try:
+            from solace_agent_mesh_enterprise.auth.input_required import handle_input_required_request
+            event_data = handle_input_required_request(event_data, sse_task_id, self)
+        except ImportError:
+            pass
+
+
         log.debug(
             "%s Sending update for A2A Task ID %s to SSE Task ID %s. Final chunk: %s",
             log_id_prefix,
@@ -2073,7 +2081,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
             await self.sse_manager.send_event(
                 task_id=sse_task_id, event_data=sse_payload, event_type=sse_event_type
             )
-            log.info(
+            log.debug(
                 "%s Successfully sent %s via SSE for A2A Task ID %s.",
                 log_id_prefix,
                 sse_event_type,

solace_agent_mesh/gateway/http_sse/dependencies.py

@@ -5,7 +5,6 @@ managed by the WebUIBackendComponent.
 
 import logging
 from collections.abc import Callable, Generator
-from contextlib import contextmanager
 from typing import TYPE_CHECKING, Any
 
 from fastapi import Depends, HTTPException, Request, status
@@ -18,11 +17,11 @@ from ...common.services.identity_service import BaseIdentityService
 from ...core_a2a.service import CoreA2AService
 from ...gateway.base.task_context import TaskContextManager
 from ...gateway.http_sse.services.agent_card_service import AgentCardService
-from ...gateway.http_sse.services.people_service import PeopleService
-from ...gateway.http_sse.services.task_service import TaskService
+from ...gateway.http_sse.services.data_retention_service import DataRetentionService
 from ...gateway.http_sse.services.feedback_service import FeedbackService
+from ...gateway.http_sse.services.people_service import PeopleService
 from ...gateway.http_sse.services.task_logger_service import TaskLoggerService
-from ...gateway.http_sse.services.data_retention_service import DataRetentionService
+from ...gateway.http_sse.services.task_service import TaskService
 from ...gateway.http_sse.session_manager import SessionManager
 from ...gateway.http_sse.sse_manager import SSEManager
 from .repository import SessionRepository
@@ -54,32 +53,57 @@ def set_component_instance(component: "WebUIBackendComponent"):
     global sac_component_instance
     if sac_component_instance is None:
         sac_component_instance = component
-        log.info("[Dependencies] SAC Component instance provided.")
+        log.info("SAC Component instance provided.")
     else:
-        log.warning("[Dependencies] SAC Component instance already set.")
+        log.warning("SAC Component instance already set.")
 
 
 def init_database(database_url: str):
-    """Initialize database with direct sessionmaker."""
+    """Initialize database with appropriate configuration based on database dialect."""
     global SessionLocal
     if SessionLocal is None:
-        engine = create_engine(database_url)
+        from sqlalchemy import event, pool
+        from sqlalchemy.engine.url import make_url
+
+        url = make_url(database_url)
+        dialect_name = url.get_dialect().name
+
+        engine_kwargs = {}
+
+        if dialect_name == "sqlite":
+            engine_kwargs = {
+                "poolclass": pool.StaticPool,
+                "connect_args": {"check_same_thread": False}
+            }
+            log.info("Configuring SQLite database (single-connection mode)")
+
+        elif dialect_name in ("postgresql", "mysql"):
+            engine_kwargs = {
+                "pool_size": 10,
+                "max_overflow": 20,
+                "pool_timeout": 30,
+                "pool_recycle": 1800,
+                "pool_pre_ping": True,
+            }
+            log.info(f"Configuring {dialect_name} database with connection pooling")
+
+        else:
+            log.warning(f"Using default configuration for dialect: {dialect_name}")
 
-        # Enable foreign keys for SQLite only (database-agnostic)
-        from sqlalchemy import event
+        engine = create_engine(database_url, **engine_kwargs)
 
         @event.listens_for(engine, "connect")
         def set_sqlite_pragma(dbapi_conn, connection_record):
-            # Only apply to SQLite connections
-            if database_url.startswith("sqlite"):
+            if dialect_name == "sqlite":
                 cursor = dbapi_conn.cursor()
                 cursor.execute("PRAGMA foreign_keys=ON")
                 cursor.close()
 
         SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
-        log.info("[Dependencies] Database initialized with foreign key support.")
+        log.debug(f"Database initialized: {url}")
+        log.info("Database initialized successfully")
     else:
-        log.warning("[Dependencies] Database already initialized.")
+        log.warning("Database already initialized.")
 
 
 def set_api_config(config: dict[str, Any]):
@@ -87,16 +111,16 @@ def set_api_config(config: dict[str, Any]):
     global api_config
     if api_config is None:
         api_config = config
-        log.info("[Dependencies] API configuration provided.")
+        log.debug("API configuration provided.")
     else:
-        log.warning("[Dependencies] API configuration already set.")
+        log.warning("API configuration already set.")
 
 
 def get_sac_component() -> "WebUIBackendComponent":
     """FastAPI dependency to get the SAC component instance."""
     if sac_component_instance is None:
         log.critical(
-            "[Dependencies] SAC Component instance accessed before it was set!"
+            "SAC Component instance accessed before it was set!"
         )
         raise HTTPException(
             status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
@@ -108,7 +132,7 @@ def get_sac_component() -> "WebUIBackendComponent":
 def get_api_config() -> dict[str, Any]:
     """FastAPI dependency to get the API configuration."""
     if api_config is None:
-        log.critical("[Dependencies] API configuration accessed before it was set!")
+        log.critical("API configuration accessed before it was set!")
         raise HTTPException(
             status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
             detail="API configuration not yet initialized.",
@@ -120,7 +144,7 @@ def get_agent_registry(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> AgentRegistry:
     """FastAPI dependency to get the AgentRegistry."""
-    log.debug("[Dependencies] get_agent_registry called")
+    log.debug("get_agent_registry called")
     return component.get_agent_registry()
 
 
@@ -128,7 +152,7 @@ def get_sse_manager(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> SSEManager:
     """FastAPI dependency to get the SSEManager."""
-    log.debug("[Dependencies] get_sse_manager called")
+    log.debug("get_sse_manager called")
     return component.get_sse_manager()
 
 
@@ -136,7 +160,7 @@ def get_session_manager(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> SessionManager:
     """FastAPI dependency to get the SessionManager."""
-    log.debug("[Dependencies] get_session_manager called")
+    log.debug("get_session_manager called")
     return component.get_session_manager()
 
 
@@ -144,7 +168,7 @@ def get_user_id_callable(
     session_manager: SessionManager = Depends(get_session_manager),
 ) -> Callable:
     """Dependency that provides the callable for getting user_id (client_id)."""
-    log.debug("[Dependencies] Providing user_id callable")
+    log.debug("Providing user_id callable")
     return session_manager.dep_get_client_id()
 
 
@@ -152,7 +176,7 @@ def ensure_session_id_callable(
     session_manager: SessionManager = Depends(get_session_manager),
 ) -> Callable:
     """Dependency that provides the callable for ensuring session_id."""
-    log.debug("[Dependencies] Providing ensure_session_id callable")
+    log.debug("Providing ensure_session_id callable")
     return session_manager.dep_ensure_session_id()
 
 
@@ -165,17 +189,17 @@ def get_user_id(
     When FRONTEND_USE_AUTHORIZATION is true: Fully relies on OAuth - user must be authenticated by AuthMiddleware.
     When FRONTEND_USE_AUTHORIZATION is false: Uses development fallback user.
     """
-    log.debug("[Dependencies] Resolving user_id string")
+    log.debug("Resolving user_id string")
 
     # AuthMiddleware should always set user state for both auth enabled/disabled cases
     if hasattr(request.state, "user") and request.state.user:
         user_id = request.state.user.get("id")
         if user_id:
-            log.debug(f"[Dependencies] Using user ID from AuthMiddleware: {user_id}")
+            log.debug(f"Using user ID from AuthMiddleware: {user_id}")
             return user_id
         else:
             log.error(
-                "[Dependencies] request.state.user exists but has no 'id' field: %s. This indicates a bug in AuthMiddleware.",
+                "request.state.user exists but has no 'id' field: %s. This indicates a bug in AuthMiddleware.",
                 request.state.user,
             )
 
@@ -185,7 +209,7 @@ def get_user_id(
     if use_authorization:
         # When OAuth is enabled, we should never reach here - AuthMiddleware should have handled authentication
         log.error(
-            "[Dependencies] OAuth is enabled but no authenticated user found. This indicates an authentication failure or middleware bug."
+            "OAuth is enabled but no authenticated user found. This indicates an authentication failure or middleware bug."
         )
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,
@@ -195,7 +219,7 @@ def get_user_id(
         # When auth is disabled, use development fallback user
         fallback_id = "sam_dev_user"
         log.info(
-            "[Dependencies] Authorization disabled and no user in request state, using fallback user: %s",
+            "Authorization disabled and no user in request state, using fallback user: %s",
             fallback_id,
         )
         return fallback_id
@@ -206,7 +230,7 @@ def ensure_session_id(
     session_manager: SessionManager = Depends(get_session_manager),
 ) -> str:
     """FastAPI dependency that directly returns the ensured session_id string."""
-    log.debug("[Dependencies] Resolving ensured session_id string")
+    log.debug("Resolving ensured session_id string")
     return session_manager.ensure_a2a_session(request)
 
 
@@ -214,7 +238,7 @@ def get_identity_service(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> BaseIdentityService | None:
     """FastAPI dependency to get the configured IdentityService instance."""
-    log.debug("[Dependencies] get_identity_service called")
+    log.debug("get_identity_service called")
     return component.identity_service
 
 
@@ -239,13 +263,13 @@ def get_people_service(
     identity_service: BaseIdentityService | None = Depends(get_identity_service),
 ) -> PeopleService:
     """FastAPI dependency to get an instance of PeopleService."""
-    log.debug("[Dependencies] get_people_service called")
+    log.debug("get_people_service called")
     return PeopleService(identity_service=identity_service)
 
 
 def get_task_repository() -> ITaskRepository:
     """FastAPI dependency to get an instance of TaskRepository."""
-    log.debug("[Dependencies] get_task_repository called")
+    log.debug("get_task_repository called")
     return TaskRepository()
 
 
@@ -254,7 +278,7 @@ def get_feedback_service(
     task_repo: ITaskRepository = Depends(get_task_repository),
 ) -> FeedbackService:
     """FastAPI dependency to get an instance of FeedbackService."""
-    log.debug("[Dependencies] get_feedback_service called")
+    log.debug("get_feedback_service called")
     # The session factory is needed for the existing DB save logic.
     session_factory = SessionLocal if component.database_url else None
     return FeedbackService(
@@ -277,11 +301,11 @@ def get_data_retention_service(
         data retention statistics or manual cleanup triggers. The service itself
         runs automatically via timer in the component.
     """
-    log.debug("[Dependencies] get_data_retention_service called")
+    log.debug("get_data_retention_service called")
 
     if not component.database_url:
         log.debug(
-            "[Dependencies] Database not configured, returning None for data retention service"
+            "Database not configured, returning None for data retention service"
         )
         return None
 
@@ -289,7 +313,7 @@ def get_data_retention_service(
         not hasattr(component, "data_retention_service")
         or component.data_retention_service is None
     ):
-        log.warning("[Dependencies] DataRetentionService not initialized on component")
+        log.warning("DataRetentionService not initialized on component")
         return None
 
     return component.data_retention_service
@@ -299,10 +323,10 @@ def get_task_logger_service(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> TaskLoggerService:
     """FastAPI dependency to get an instance of TaskLoggerService."""
-    log.debug("[Dependencies] get_task_logger_service called")
+    log.debug("get_task_logger_service called")
     task_logger_service = component.get_task_logger_service()
     if task_logger_service is None:
-        log.error("[Dependencies] TaskLoggerService is not available.")
+        log.error("TaskLoggerService is not available.")
         raise HTTPException(
             status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
             detail="Task logging service is not configured or available.",
@@ -317,7 +341,7 @@ def get_publish_a2a_func(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> PublishFunc:
     """FastAPI dependency to get the component's publish_a2a method."""
-    log.debug("[Dependencies] get_publish_a2a_func called")
+    log.debug("get_publish_a2a_func called")
     return component.publish_a2a
 
 
@@ -325,7 +349,7 @@ def get_namespace(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> str:
     """FastAPI dependency to get the namespace."""
-    log.debug("[Dependencies] get_namespace called")
+    log.debug("get_namespace called")
     return component.get_namespace()
 
 
@@ -333,7 +357,7 @@ def get_gateway_id(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> str:
     """FastAPI dependency to get the Gateway ID."""
-    log.debug("[Dependencies] get_gateway_id called")
+    log.debug("get_gateway_id called")
     return component.get_gateway_id()
 
 
@@ -341,7 +365,7 @@ def get_config_resolver(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> ConfigResolver:
     """FastAPI dependency to get the ConfigResolver."""
-    log.debug("[Dependencies] get_config_resolver called")
+    log.debug("get_config_resolver called")
     return component.get_config_resolver()
 
 
@@ -351,7 +375,7 @@ def get_app_config(
     """
     FastAPI dependency to safely get the application configuration dictionary.
     """
-    log.debug("[Dependencies] get_app_config called")
+    log.debug("get_app_config called")
     return component.component_config.get("app_config", {})
 
 
@@ -365,7 +389,7 @@ async def get_user_config(
     """
     FastAPI dependency to get the user-specific configuration.
     """
-    log.debug(f"[Dependencies] get_user_config called for user_id: {user_id}")
+    log.debug(f"get_user_config called for user_id: {user_id}")
     gateway_context = {
         "gateway_id": component.gateway_id,
         "gateway_app_config": app_config,
@@ -408,7 +432,7 @@ class ValidatedUserConfig:
         user_id = user_config.get("user_profile", {}).get("id")
 
         log.debug(
-            f"[Dependencies] ValidatedUserConfig called for user_id: {user_id} with required scopes: {self.required_scopes}"
+            f"ValidatedUserConfig called for user_id: {user_id} with required scopes: {self.required_scopes}"
         )
 
         # Validate scopes
@@ -418,7 +442,7 @@ class ValidatedUserConfig:
             {},
         ):
             log.warning(
-                f"[Dependencies] Authorization denied for user '{user_id}'. Required scopes: {self.required_scopes}"
+                f"Authorization denied for user '{user_id}'. Required scopes: {self.required_scopes}"
             )
             raise HTTPException(
                 status_code=status.HTTP_403_FORBIDDEN,
@@ -432,7 +456,7 @@ def get_shared_artifact_service(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> BaseArtifactService | None:
     """FastAPI dependency to get the shared ArtifactService."""
-    log.debug("[Dependencies] get_shared_artifact_service called")
+    log.debug("get_shared_artifact_service called")
     return component.get_shared_artifact_service()
 
 
@@ -440,7 +464,7 @@ def get_embed_config(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> dict[str, Any]:
     """FastAPI dependency to get embed-related configuration."""
-    log.debug("[Dependencies] get_embed_config called")
+    log.debug("get_embed_config called")
     return component.get_embed_config()
 
 
@@ -448,10 +472,10 @@ def get_core_a2a_service(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> CoreA2AService:
     """FastAPI dependency to get the CoreA2AService."""
-    log.debug("[Dependencies] get_core_a2a_service called")
+    log.debug("get_core_a2a_service called")
     core_service = component.get_core_a2a_service()
     if core_service is None:
-        log.critical("[Dependencies] CoreA2AService accessed before initialization!")
+        log.critical("CoreA2AService accessed before initialization!")
         raise HTTPException(status_code=503, detail="Core service not ready.")
     return core_service
 
@@ -460,10 +484,10 @@ def get_task_context_manager_from_component(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> TaskContextManager:
     """FastAPI dependency to get the TaskContextManager from the component."""
-    log.debug("[Dependencies] get_task_context_manager_from_component called")
+    log.debug("get_task_context_manager_from_component called")
     if component.task_context_manager is None:
         log.critical(
-            "[Dependencies] TaskContextManager accessed before initialization!"
+            "TaskContextManager accessed before initialization!"
         )
         raise HTTPException(status_code=503, detail="Task context manager not ready.")
     return component.task_context_manager
@@ -473,7 +497,7 @@ def get_agent_card_service(
     registry: AgentRegistry = Depends(get_agent_registry),
 ) -> AgentCardService:
     """FastAPI dependency to get an instance of AgentCardService."""
-    log.debug("[Dependencies] get_agent_card_service called")
+    log.debug("get_agent_card_service called")
     return AgentCardService(agent_registry=registry)
 
 
@@ -489,7 +513,7 @@ def get_task_service(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> TaskService:
     """FastAPI dependency to get an instance of TaskService."""
-    log.debug("[Dependencies] get_task_service called")
+    log.debug("get_task_service called")
     app_name = component.get_config("name", "WebUIBackendApp")
     return TaskService(
         core_a2a_service=core_a2a_service,
@@ -506,7 +530,7 @@ def get_task_service(
 def get_session_business_service(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> SessionService:
-    log.debug("[Dependencies] get_session_business_service called")
+    log.debug("get_session_business_service called")
 
     # Note: Session and message repositories will be created per request
     # when the SessionService methods receive the db parameter
@@ -516,7 +540,7 @@ def get_session_business_service(
 def get_session_validator(
     component: "WebUIBackendComponent" = Depends(get_sac_component),
 ) -> Callable[[str, str], bool]:
-    log.debug("[Dependencies] get_session_validator called")
+    log.debug("get_session_validator called")
 
     if SessionLocal:
         log.debug("Using database-backed session validation")
@@ -532,7 +556,7 @@ def get_session_validator(
                     return session_domain is not None
                 finally:
                     db.close()
-            except:
+            except Exception:
                 return False
 
         return validate_with_database
@@ -550,7 +574,7 @@ def get_session_validator(
 def get_db_optional() -> Generator[Session | None, None, None]:
     """Optional database dependency that returns None if database is not configured."""
     if SessionLocal is None:
-        log.debug("[Dependencies] Database not configured, returning None")
+        log.debug("Database not configured, returning None")
         yield None
     else:
         db = SessionLocal()
@@ -570,7 +594,7 @@ def get_session_business_service_optional(
     """Optional session service dependency that returns None if database is not configured."""
     if SessionLocal is None:
         log.debug(
-            "[Dependencies] Database not configured, returning None for session service"
+            "Database not configured, returning None for session service"
         )
         return None
     return SessionService(component=component)

solace_agent_mesh/gateway/http_sse/main.py

@@ -209,6 +209,7 @@ def _create_auth_middleware(component):
             skip_paths = [
                 "/api/v1/config",
                 "/api/v1/auth/callback",
+                "/api/v1/auth/tool/callback",
                 "/api/v1/auth/login",
                 "/api/v1/auth/refresh",
                 "/api/v1/csrf-token",
@@ -526,14 +527,14 @@ def setup_dependencies(component: "WebUIBackendComponent", database_url: str = N
     api_config_dict = _create_api_config(app_config, database_url)
 
     dependencies.set_api_config(api_config_dict)
-    log.info("API configuration extracted and stored.")
+    log.debug("API configuration extracted and stored.")
 
     _setup_middleware(component)
     _setup_routers()
     _setup_static_files()
 
     _dependencies_initialized = True
-    log.info("[setup_dependencies] Dependencies initialization complete")
+    log.debug("[setup_dependencies] Dependencies initialization complete")
 
 
 def _setup_middleware(component: "WebUIBackendComponent") -> None:

solace_agent_mesh/gateway/http_sse/routers/agent_cards.py

@@ -33,7 +33,7 @@ async def get_discovered_agent_cards(
             if agent_registry.get_agent(name)
         ]
 
-        log.info("%sReturning %d discovered agent cards.", log_prefix, len(agents))
+        log.debug("%sReturning %d discovered agent cards.", log_prefix, len(agents))
         return agents
     except Exception as e:
         log.exception("%sError retrieving discovered agent cards: %s", log_prefix, e)