PyPI - solace-agent-mesh - Versions diffs - 1.6.1__py3-none-any.whl → 1.6.3__py3-none-any.whl - Mend

solace-agent-mesh 1.6.1py3-none-any.whl → 1.6.3py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of solace-agent-mesh might be problematic. Click here for more details.

Files changed (124) hide show

solace_agent_mesh/gateway/http_sse/routers/auth.py CHANGED Viewed

@@ -3,19 +3,20 @@ Router for handling authentication-related endpoints.
 """
 import logging
+import secrets
+from urllib.parse import urlencode
+import httpx
 from fastapi import (
     APIRouter,
-    Request as FastAPIRequest,
     Depends,
     HTTPException,
+    Request as FastAPIRequest,
     Response,
 )
-from fastapi.responses import RedirectResponse
-import httpx
-import secrets
-from urllib.parse import urlencode
+from fastapi.responses import HTMLResponse, RedirectResponse
-from ...http_sse.dependencies import get_sac_component, get_api_config
+from ...http_sse.dependencies import get_api_config, get_sac_component
 log = logging.getLogger(__name__)
@@ -212,3 +213,99 @@ async def refresh_token(
         "access_token": access_token,
         "refresh_token": new_refresh_token,
     }
+@router.get("/auth/tool/callback")
+async def auth_tool_callback(
+    request: FastAPIRequest,
+    component: "WebUIBackendComponent" = Depends(get_sac_component),
+):
+    """
+    Handles OAuth2 authorization code grant response for tool authentication.
+    """
+    code = request.query_params.get("code")
+    state = request.query_params.get("state")
+    error = request.query_params.get("error")
+    error_description = request.query_params.get("error_description")
+    if error:
+        log.error(f"OAuth2 tool callback received error: {error} - {error_description}")
+        return HTMLResponse(
+            content=f"""
+            <html>
+                <head>
+                    <title>Authorization Error</title>
+                </head>
+                <body>
+                    <h2>Authorization Error</h2>
+                    <p>Error: {error}</p>
+                    <p>Description: {error_description or 'No description provided'}</p>
+                    <p>Please close this window and try again.</p>
+                </body>
+            </html>
+            """,
+            status_code=400
+        )
+    # Get the current request URL for logging/debugging
+    url = str(request.url)
+    if not code:
+        log.warning("OAuth2 tool callback received without authorization code")
+        return HTMLResponse(
+            content="""
+            <html>
+                <head>
+                    <title>Authorization Error</title>
+                </head>
+                <body>
+                    <h2>Authorization Error</h2>
+                    <p>No authorization code received. Please close this window and try again.</p>
+                </body>
+            </html>
+            """,
+            status_code=400
+        )
+    log.info(f"OAuth2 tool callback received authorization code: {code}")
+    try:
+        from solace_agent_mesh_enterprise.auth.input_required import process_auth_grant_response
+        await process_auth_grant_response(component, code, state, url)
+    except ImportError:
+        pass
+    # Return simple HTML page instructing user to close the window
+    return HTMLResponse(
+        content="""
+        <html>
+            <head>
+                <title>Authorization Complete</title>
+                <style>
+                    body {
+                        font-family: Arial, sans-serif;
+                        text-align: center;
+                        padding: 50px;
+                        background-color: #f5f5f5;
+                    }
+                    .container {
+                        background: white;
+                        padding: 30px;
+                        border-radius: 8px;
+                        box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+                        display: inline-block;
+                    }
+                    h2 { color: #28a745; }
+                </style>
+            </head>
+            <body>
+                <div class="container">
+                    <h2>✓ Authorization Complete</h2>
+                    <p>You have successfully authorized the tool access.</p>
+                    <p><strong>Please close this window.</strong></p>
+                </div>
+            </body>
+        </html>
+        """
+    )

solace_agent_mesh/gateway/http_sse/routers/config.py CHANGED Viewed

@@ -69,7 +69,7 @@ async def get_app_config(
             "frontend_feature_enablement": feature_enablement,
             "persistence_enabled": api_config.get("persistence_enabled", False),
         }
-        log.info("%sReturning frontend configuration.", log_prefix)
+        log.debug("%sReturning frontend configuration.", log_prefix)
         return config_data
     except Exception as e:
         log.exception(

solace_agent_mesh/gateway/http_sse/routers/sessions.py CHANGED Viewed

@@ -129,7 +129,7 @@ async def save_task(
     Creates a new task or updates an existing one.
     """
     user_id = user.get("id")
-    log.info(
+    log.debug(
         "User %s attempting to save task %s for session %s",
         user_id,
         request.task_id,

solace_agent_mesh/gateway/http_sse/routers/sse.py CHANGED Viewed

@@ -13,6 +13,8 @@ from ....gateway.http_sse.sse_manager import SSEManager
 from ....gateway.http_sse.dependencies import get_sse_manager
 log = logging.getLogger(__name__)
+trace_logger = logging.getLogger("sam_trace")
 router = APIRouter()
@@ -27,12 +29,12 @@ async def subscribe_to_task_events(
     Establishes an SSE connection to receive real-time updates for a specific task.
     """
     log_prefix = "[GET /api/v1/sse/subscribe/%s] " % task_id
-    log.info("%sClient requesting SSE subscription.", log_prefix)
+    log.debug("%sClient requesting SSE subscription.", log_prefix)
     connection_queue: asyncio.Queue = None
     try:
         connection_queue = await sse_manager.create_sse_connection(task_id)
-        log.info("%sSSE connection queue created.", log_prefix)
+        log.debug("%sSSE connection queue created.", log_prefix)
         async def event_generator():
             nonlocal connection_queue
@@ -73,9 +75,17 @@ async def subscribe_to_task_events(
                                 log_prefix,
                             )
                             break
-                        log.debug(
-                            "%sYielding event_payload: %s", log_prefix, event_payload
-                        )
+                        if trace_logger.isEnabledFor(logging.DEBUG):
+                            trace_logger.debug(
+                                "%sYielding event_payload: %s",
+                                log_prefix, event_payload
+                            )
+                        else:
+                            log.debug(
+                                "%sYielding event: %s",
+                                log_prefix,
+                                event_payload.get("event") if event_payload else "unknown"
+                            )
                         yield event_payload
                         connection_queue.task_done()
                         log.debug(

solace_agent_mesh/gateway/http_sse/routers/tasks.py CHANGED Viewed

@@ -77,7 +77,7 @@ async def _submit_task(
                 status_code=status.HTTP_401_UNAUTHORIZED,
                 detail="User authentication failed or identity not found.",
             )
-        log.info(
+        log.debug(
             "%sAuthenticated user identity: %s",
             log_prefix,
             user_identity.get("id", "unknown"),
@@ -107,7 +107,7 @@ async def _submit_task(
         else:
             # Create new session when frontend doesn't provide one
             session_id = session_manager.create_new_session_id(request)
-            log.info(
+            log.debug(
                 "%sNo valid session ID from frontend, created new session: %s",
                 log_prefix,
                 session_id,
@@ -125,7 +125,7 @@ async def _submit_task(
                         session_id=session_id,
                     )
                     db.commit()
-                    log.info(
+                    log.debug(
                         "%sCreated session in database: %s", log_prefix, session_id
                     )
                 except Exception as e:

solace_agent_mesh/gateway/http_sse/routers/visualization.py CHANGED Viewed

@@ -14,6 +14,7 @@ from fastapi import (
     status,
 )
 from pydantic import BaseModel, Field
+import json
 from typing import List, Optional, Dict, Any, Set
@@ -31,6 +32,7 @@ if TYPE_CHECKING:
     from ....gateway.http_sse.component import WebUIBackendComponent
 log = logging.getLogger(__name__)
+trace_logger = logging.getLogger("sam_trace")
 router = APIRouter()
@@ -231,11 +233,77 @@ def _resolve_user_identity_for_authorization(
                 "%s No user_identity provided but authorization is enabled. This should not happen.",
                 log_id_prefix,
             )
-            raise ValueError("No user identity available when authorization is required")
+            raise ValueError(
+                "No user identity available when authorization is required"
+            )
     return user_identity
+def _include_for_visualization(event_payload: Dict[str, Any]) -> bool:
+    """
+    Check if an event should be included for visualization based on metadata.
+    Args:
+        event_payload: The event payload containing event data
+    Returns:
+        False if the event should be excluded from visualization (when visualization is False),
+        True otherwise (include by default)
+    """
+    try:
+        # Get the data field from the event payload
+        data_str = event_payload.get("data")
+        if not data_str:
+            return True  # Include by default if no data
+        # Parse the JSON data
+        try:
+            data = json.loads(data_str)
+        except (json.JSONDecodeError, TypeError):
+            return True
+        # Look for the full_payload in the data
+        full_payload = data.get("full_payload")
+        if not full_payload:
+            return True
+        # Check if full_payload has params
+        params = full_payload.get("params")
+        if not params:
+            return True
+        # Check if params has message
+        message = params.get("message")
+        if not message:
+            return True
+        # Check if message has metadata
+        metadata = message.get("metadata")
+        if not metadata:
+            return True
+        # Check the visualization setting in metadata
+        visualization_setting = metadata.get("visualization")
+        if visualization_setting is not None and (
+            (
+                isinstance(visualization_setting, str)
+                and visualization_setting.lower() == "false"
+            )
+            or (
+                isinstance(visualization_setting, bool)
+                and visualization_setting is False
+            )
+        ):
+            return False
+        return True
+    except Exception as e:
+        log.warning("Error checking visualization filter for event: %s", e)
+        return True
 @router.post(
     "/subscribe",
     response_model=VisualizationSubscribeResponse,
@@ -321,7 +389,7 @@ async def subscribe_to_visualization_stream(
     resolved_user_identity = _resolve_user_identity_for_authorization(
         component, user_id
     )
-    log.info(
+    log.debug(
         "%s Resolved user identity for authorization: '%s' (from raw user_id: '%s')",
         log_id_prefix,
         resolved_user_identity,
@@ -339,7 +407,7 @@ async def subscribe_to_visualization_stream(
         user_config = await config_resolver.resolve_user_config(
             resolved_user_identity, gateway_context, {}
         )
-        log.info(
+        log.debug(
             "%s Resolved user_config for resolved_user_identity '%s': %s",
             log_id_prefix,
             resolved_user_identity,
@@ -457,7 +525,7 @@ async def subscribe_to_visualization_stream(
                 )
                 firehose_topic = f"{component.namespace.strip('/')}/a2a/>"
-                log.info(
+                log.debug(
                     "%s Adding firehose subscription '%s' for my_a2a_messages stream.",
                     log_id_prefix,
                     firehose_topic,
@@ -780,13 +848,27 @@ async def get_visualization_stream_events(
                             stream_id,
                         )
                         break
-                    yield event_payload
+                    if _include_for_visualization(event_payload):
+                        if trace_logger.isEnabledFor(logging.DEBUG):
+                            trace_logger.debug(
+                                "%s Yielding event for stream %s: %s",
+                                log_id_prefix,
+                                stream_id,
+                                event_payload,
+                            )
+                        else:
+                            log.debug(
+                                "%s Yielding event for stream %s",
+                                log_id_prefix,
+                                stream_id,
+                            )
+                        yield event_payload
                     sse_queue.task_done()
                 except asyncio.TimeoutError:
                     yield {"comment": "keep-alive"}
                     continue
                 except asyncio.CancelledError:
-                    log.info(
+                    log.debug(
                         "%s SSE event generator for stream %s cancelled.",
                         log_id_prefix,
                         stream_id,
@@ -800,7 +882,7 @@ async def get_visualization_stream_events(
                 e,
             )
         finally:
-            log.info(
+            log.debug(
                 "%s SSE event generator for stream %s finished.",
                 log_id_prefix,
                 stream_id,
@@ -1135,4 +1217,4 @@ async def unsubscribe_from_visualization_stream(
     return Response(status_code=status.HTTP_204_NO_CONTENT)
-log.info("Router for A2A Message Visualization initialized.")
+log.info("Initialized Router for A2A Message Visualization.")

solace_agent_mesh/gateway/http_sse/services/session_service.py CHANGED Viewed

@@ -101,7 +101,7 @@ class SessionService:
         session_repository = self._get_repositories(db)
         created_session = session_repository.save(db, session)
-        log.info("Created new session %s for user %s", created_session.id, user_id)
+        log.debug("Created new session %s for user %s", created_session.id, user_id)
         if not created_session:
             raise ValueError(f"Failed to save session for {session_id}")

solace_agent_mesh/gateway/http_sse/session_manager.py CHANGED Viewed

@@ -35,10 +35,10 @@ class SessionManager:
         self.force_user_identity = app_config.get("force_user_identity")
         self.use_authorization = app_config.get("frontend_use_authorization", False)
         self._temp_code_cache = {}
-        log.info("[SessionManager] Initialized.")
+        log.info("Initialized SessionManager")
         if self.force_user_identity:
             log.warning(
-                f"[SessionManager] Forcing user identity to: {self.force_user_identity}"
+                f"Forcing user identity to: {self.force_user_identity}"
             )
     def _get_or_create_client_id(self, request: Request) -> str | None:
@@ -54,19 +54,19 @@ class SessionManager:
             user_id = request.state.user.get("id")
             if user_id:
                 log.debug(
-                    "[SessionManager] Using authenticated user ID from request.state: %s",
+                    "Using authenticated user ID from request.state: %s",
                     user_id,
                 )
                 return user_id
             else:
                 log.warning(
-                    "[SessionManager] request.state.user exists but has no 'id' field. Falling back to other methods."
+                    "request.state.user exists but has no 'id' field. Falling back to other methods."
                 )
         user_id = self.get_user_id(request)
         if user_id:
             log.debug(
-                "[SessionManager] Using authenticated user_id from session as A2A Client ID: %s",
+                "Using authenticated user_id from session as A2A Client ID: %s",
                 user_id,
             )
             return user_id
@@ -74,7 +74,7 @@ class SessionManager:
         client_id = request.session.get(SESSION_KEY_CLIENT_ID)
         if client_id:
             log.debug(
-                "[SessionManager] Using existing A2A Client ID: %s for web session.",
+                "Using existing A2A Client ID: %s for web session.",
                 client_id,
             )
             return client_id
@@ -82,14 +82,14 @@ class SessionManager:
         if not self.use_authorization:
             client_id = "sam_dev_user"
             log.info(
-                "[SessionManager] No authenticated user and auth is disabled, using client ID: %s for web session.",
+                "No authenticated user and auth is disabled, using client ID: %s for web session.",
                 client_id,
             )
             request.session[SESSION_KEY_CLIENT_ID] = client_id
             return client_id
         log.warning(
-            "[SessionManager] Could not determine client ID and authorization is enabled."
+            "Could not determine client ID and authorization is enabled."
         )
         return None
@@ -106,7 +106,7 @@ class SessionManager:
         Returns None if no session has been started for the current agent in this web session.
         """
         session_id = request.session.get(SESSION_KEY_SESSION_ID)
-        log.debug("[SessionManager] Retrieving A2A Session ID: %s", session_id)
+        log.debug("Retrieving A2A Session ID: %s", session_id)
         return session_id
     def start_new_a2a_session(self, request: Request) -> str:
@@ -123,7 +123,7 @@ class SessionManager:
         new_session_id = f"web-session-{uuid.uuid4().hex}"
         request.session[SESSION_KEY_SESSION_ID] = new_session_id
         log.info(
-            "[SessionManager] Started new A2A Session ID: %s for Client ID: %s",
+            "Started new A2A Session ID: %s for Client ID: %s",
             new_session_id,
             client_id,
         )
@@ -142,7 +142,7 @@ class SessionManager:
             )
         new_session_id = f"web-session-{uuid.uuid4().hex}"
         log.info(
-            "[SessionManager] Generated new A2A Session ID: %s for Client ID: %s (not stored in cookies)",
+            "Generated new A2A Session ID: %s for Client ID: %s (not stored in cookies)",
             new_session_id,
             client_id,
         )
@@ -157,7 +157,7 @@ class SessionManager:
         if not session_id:
             session_id = self.start_new_a2a_session(request)
             log.info(
-                "[SessionManager] No A2A Session ID found, created new one via ensure_a2a_session: %s",
+                "No A2A Session ID found, created new one via ensure_a2a_session: %s",
                 session_id,
             )
         return session_id
@@ -171,7 +171,7 @@ class SessionManager:
         request.session[SESSION_KEY_ACCESS_TOKEN] = access_token
         if refresh_token:
             request.session[SESSION_KEY_REFRESH_TOKEN] = refresh_token
-        log.info("[SessionManager] Stored auth tokens directly in session.")
+        log.info("Stored auth tokens directly in session.")
     def get_access_token(self, request: Request) -> str | None:
         """
@@ -191,14 +191,14 @@ class SessionManager:
         """
         request.session.pop(SESSION_KEY_ACCESS_TOKEN, None)
         request.session.pop(SESSION_KEY_REFRESH_TOKEN, None)
-        log.info("[SessionManager] Cleared auth tokens from session")
+        log.info("Cleared auth tokens from session")
     def store_user_id(self, request: Request, user_id: str) -> None:
         """
         Stores the user ID in the web session.
         """
         request.session[SESSION_KEY_USER_ID] = user_id
-        log.info("[SessionManager] Stored user ID in session: %s", user_id)
+        log.info("Stored user ID in session: %s", user_id)
     def get_user_id(self, request: Request) -> str | None:
         """

solace_agent_mesh/gateway/http_sse/shared/exception_handlers.py CHANGED Viewed

@@ -6,6 +6,8 @@ into appropriate HTTP responses with consistent formatting. These handlers
 can be used by any FastAPI application for uniform error handling.
 """
+import logging
 from fastapi import HTTPException, Request, status
 from fastapi.responses import JSONResponse
 from fastapi.exceptions import RequestValidationError
@@ -24,6 +26,8 @@ from .exceptions import (
 )
 from .error_dto import EventErrorDTO
+log = logging.getLogger(__name__)
 def create_error_response(
     status_code: int, message: str, validation_details: dict = None
@@ -105,7 +109,18 @@ async def webui_backend_exception_handler(
     request: Request, exc: WebUIBackendException
 ) -> JSONResponse:
     """Handle generic WebUI backend exceptions - 500 Internal Server Error."""
-    error_dto = EventErrorDTO.create("An unexpected server error occurred.")
+    log.error(
+        f"WebUIBackendException: {exc.message}",
+        extra={
+            "path": request.url.path,
+            "method": request.method,
+            "details": exc.details if hasattr(exc, 'details') else None
+        },
+        exc_info=True
+    )
+    message = exc.message if exc.message else "An unexpected server error occurred."
+    error_dto = EventErrorDTO.create(message)
     return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content=error_dto.model_dump())

solace_agent_mesh/gateway/http_sse/sse_manager.py CHANGED Viewed

@@ -13,6 +13,8 @@ import math
 from .sse_event_buffer import SSEEventBuffer
 log = logging.getLogger(__name__)
+trace_logger = logging.getLogger("sam_trace")
 class SSEManager:
     """
@@ -180,12 +182,19 @@ class SSEManager:
                 self._event_buffer.buffer_event(task_id, sse_payload)
                 return
-            log.debug(
-                "%s Prepared SSE payload for Task ID %s: %s",
-                self.log_identifier,
-                task_id,
-                sse_payload,
-            )
+            if trace_logger.isEnabledFor(logging.DEBUG):
+                trace_logger.debug(
+                    "%s Prepared SSE payload for Task ID %s: %s",
+                    self.log_identifier,
+                    task_id,
+                    sse_payload,
+                )
+            else:
+                log.debug(
+                    "%s Prepared SSE payload for Task ID %s",
+                    self.log_identifier,
+                    task_id,
+                )
             queues_to_remove = []
             for connection_queue in list(self._connections.get(task_id, [])):

solace_agent_mesh/templates/logging_config_template.ini CHANGED Viewed

@@ -38,8 +38,8 @@ args=(sys.stdout,)
 keys=simpleFormatter,jsonFormatter
 [formatter_simpleFormatter]
-format=%(asctime)s | %(levelname)-5s | %(name)s | %(message)s
+format=%(asctime)s | %(levelname)-5s | %(threadName)s | %(name)s | %(message)s
 [formatter_jsonFormatter]
 class=pythonjsonlogger.jsonlogger.JsonFormatter
-format=%(asctime)s %(levelname)s %(name)s %(message)s
+format=%(asctime)s %(levelname)s %(threadName)s %(name)s %(message)s

{solace_agent_mesh-1.6.1.dist-info → solace_agent_mesh-1.6.3.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: solace-agent-mesh
-Version: 1.6.1
+Version: 1.6.3
 Summary: Solace Agent Mesh is an open-source framework for building event-driven, multi-agent AI systems where specialized agents collaborate on complex tasks.
 Project-URL: Homepage, https://github.com/SolaceLabs/solace-agent-mesh
 Project-URL: Repository, https://github.com/SolaceLabs/solace-agent-mesh
@@ -249,7 +249,7 @@ Requires-Dist: python-jwt==4.1.0
 Requires-Dist: python-multipart==0.0.20
 Requires-Dist: pyyaml==6.0.2
 Requires-Dist: rouge==1.0.1
-Requires-Dist: solace-ai-connector==1.3.0
+Requires-Dist: solace-ai-connector==1.3.1
 Requires-Dist: sqlalchemy==2.0.40
 Requires-Dist: sse-starlette==3.0.2
 Requires-Dist: toml==0.10.2

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

Potentially problematic release.

solace-agent-mesh 1.6.1py3-none-any.whl → 1.6.3py3-none-any.whl