solace-agent-mesh 1.11.2__py3-none-any.whl → 1.12.0__py3-none-any.whl

solace_agent_mesh/gateway/http_sse/routers/artifacts.py

@@ -1,6 +1,7 @@
 """
 FastAPI router for managing session-specific artifacts via REST endpoints.
 """
+from __future__ import annotations
 
 import logging
 from collections.abc import Callable
@@ -319,20 +320,19 @@ async def upload_artifact_with_session(
             except ValueError:
                 log.warning("%s Invalid Content-Length header: %s", log_prefix, content_length)
         
-        # Read file content in chunks with size validation
+        # Validate file size by streaming through WITHOUT accumulating chunks in memory
         chunk_size = LOAD_FILE_CHUNK_SIZE
-        content_bytes = bytearray()
         total_bytes_read = 0
-        
+
         try:
+            # Step 1: Validate size by reading chunks (discard data, just count bytes)
             while True:
                 chunk = await upload_file.read(chunk_size)
                 if not chunk:
                     break  # End of file
-                
-                chunk_len = len(chunk)
-                total_bytes_read += chunk_len
-                
+
+                total_bytes_read += len(chunk)
+
                 # Validate size during reading (fail fast)
                 if total_bytes_read > max_upload_size:
                     error_msg = (
@@ -341,19 +341,20 @@ async def upload_artifact_with_session(
                         f"{total_bytes_read / (1024*1024):.2f} MB > {max_upload_size / (1024*1024):.2f} MB)"
                     )
                     log.warning("%s %s", log_prefix, error_msg)
-                    
+
                     raise HTTPException(
                         status_code=status.HTTP_413_REQUEST_ENTITY_TOO_LARGE,
                         detail=error_msg
                     )
-                
-                content_bytes.extend(chunk)
-            
-            # Convert to bytes for consistency with existing code
-            content_bytes = bytes(content_bytes)
-            
+
+            # Step 2: Size is valid - reset to beginning
+            await upload_file.seek(0)
+
+            # Step 3: Read all content at once
+            content_bytes = await upload_file.read()
+
             log.debug(
-                "%s File read successfully in chunks: %d bytes total",
+                "%s File validated (%d bytes) and loaded into memory",
                 log_prefix,
                 total_bytes_read
             )

solace_agent_mesh/gateway/http_sse/routers/auth.py

@@ -1,12 +1,11 @@
 """
 Router for handling authentication-related endpoints.
 """
+from __future__ import annotations
 
 import logging
 import secrets
-from urllib.parse import urlencode
 
-import httpx
 from fastapi import (
     APIRouter,
     Depends,
@@ -14,7 +13,7 @@ from fastapi import (
     Request as FastAPIRequest,
     Response,
 )
-from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.responses import HTMLResponse
 
 from ...http_sse.dependencies import get_api_config, get_sac_component
 
@@ -30,20 +29,16 @@ async def initiate_login(
     """
     Initiates the login flow by redirecting to the external authorization service.
     """
-    external_auth_url = config.get("external_auth_service_url", "http://localhost:8080")
-    callback_url = config.get(
-        "external_auth_callback_uri", "http://localhost:8000/api/v1/auth/callback"
-    )
-
-    params = {
-        "provider": config.get("external_auth_provider", "azure"),
-        "redirect_uri": callback_url,
-    }
-
-    login_url = f"{external_auth_url}/login?{urlencode(params)}"
-    log.info(f"Redirecting to external authorization service: {login_url}")
+    try:
+        from solace_agent_mesh_enterprise.gateway.auth import handle_login_initiation
 
-    return RedirectResponse(url=login_url)
+        return await handle_login_initiation(request, config)
+    except ImportError:
+        raise HTTPException(
+            status_code=501,
+            detail="OAuth authentication requires enterprise package. "
+            "Install: pip install solace-agent-mesh-enterprise",
+        )
 
 
 @router.get("/csrf-token")
@@ -76,86 +71,56 @@ async def auth_callback(
     """
     Handles the callback from the OIDC provider by calling an external exchange service.
     """
-    code = request.query_params.get("code")
-
-    if not code:
-        raise HTTPException(status_code=400, detail="Missing authorization code")
-
-    external_auth_url = config.get("external_auth_service_url", "http://localhost:8080")
-    exchange_url = f"{external_auth_url}/exchange-code"
-    redirect_uri = config.get(
-        "external_auth_callback_uri", "http://localhost:8000/api/v1/auth/callback"
-    )
-
-    async with httpx.AsyncClient() as client:
-        try:
-            response = await client.post(
-                exchange_url,
-                json={
-                    "code": code,
-                    "provider": config.get("external_auth_provider", "azure"),
-                    "redirect_uri": redirect_uri,
-                },
-                timeout=20.0,
-            )
-            response.raise_for_status()
-            token_data = response.json()
-        except httpx.HTTPStatusError as e:
-            log.error(f"Failed to exchange code: {e.response.text}")
-            raise HTTPException(
-                status_code=e.response.status_code,
-                detail=f"Failed to exchange code: {e.response.text}",
-            )
-        except Exception as e:
-            log.error(f"Error during code exchange: {e}")
-            raise HTTPException(status_code=500, detail="Error during code exchange")
-
-    access_token = token_data.get("access_token")
-    refresh_token = token_data.get("refresh_token")
+    try:
+        from solace_agent_mesh_enterprise.gateway.auth import handle_auth_callback
 
-    if not access_token:
+        return await handle_auth_callback(
+            request, config, component
+        )
+    except ImportError:
         raise HTTPException(
-            status_code=400, detail="Access token not in response from exchange service"
+            status_code=501,
+            detail="OAuth authentication requires enterprise package. "
+            "Install: pip install solace-agent-mesh-enterprise",
         )
 
-    request.session["access_token"] = access_token
-    if refresh_token:
-        request.session["refresh_token"] = refresh_token
-    log.debug("Tokens stored directly in session.")
 
+@router.post("/auth/logout")
+async def logout(request: FastAPIRequest):
+    """
+    Logout endpoint - clears server-side session and access token.
+    
+    This endpoint:
+    - Clears the access_token from the session storage
+    - Clears all session data
+    - Returns success even if already logged out (idempotent)
+    
+    The session cookie will be invalidated, requiring re-authentication
+    on the next request.
+    """
     try:
-        async with httpx.AsyncClient() as client:
-            user_info_response = await client.get(
-                f"{external_auth_url}/user_info",
-                params={"provider": config.get("external_auth_provider", "azure")},
-                headers={"Authorization": f"Bearer {access_token}"},
-            )
-            user_info_response.raise_for_status()
-            user_info = user_info_response.json()
-
-            user_id = user_info.get("email", "authenticated_user")
-            if user_id:
-                session_manager = component.get_session_manager()
-                session_manager.store_user_id(request, user_id)
-            else:
-                log.warning("Could not find 'email' in user info response.")
-
-    except httpx.HTTPStatusError as e:
-        log.error(f"Failed to get user info: {e.response.text}")
-
+        # Clear access token from session storage
+        if hasattr(request, 'session') and 'access_token' in request.session:
+            del request.session['access_token']
+            log.debug("Cleared access_token from session")
+        
+        # Clear refresh token if present
+        if hasattr(request, 'session') and 'refresh_token' in request.session:
+            del request.session['refresh_token']
+            log.debug("Cleared refresh_token from session")
+        
+        # Clear all other session data
+        if hasattr(request, 'session'):
+            request.session.clear()
+            log.debug("Cleared all session data")
+        
+        log.info("User logged out successfully")
+        return {"success": True, "message": "Logged out successfully"}
+    
     except Exception as e:
-        log.error(f"Error getting user info: {e}")
-
-    frontend_base_url = config.get("frontend_redirect_url", "http://localhost:3000")
-
-    hash_params = {"access_token": access_token}
-    if refresh_token:
-        hash_params["refresh_token"] = refresh_token
-
-    hash_fragment = urlencode(hash_params)
-
-    frontend_redirect_url = f"{frontend_base_url}/auth-callback.html#{hash_fragment}"
-    return RedirectResponse(url=frontend_redirect_url)
+        log.error(f"Error during logout: {e}")
+        # Still return success - logout should be idempotent
+        return {"success": True, "message": "Logged out successfully"}
 
 
 @router.post("/auth/refresh")
@@ -167,53 +132,17 @@ async def refresh_token(
     """
     Refreshes an access token using the external authorization service.
     """
-    data = await request.json()
-    refresh_token = data.get("refresh_token")
-    if not refresh_token:
-        raise HTTPException(status_code=400, detail="Missing refresh_token")
-
-    external_auth_url = config.get("external_auth_service_url", "http://localhost:8080")
-    refresh_url = f"{external_auth_url}/refresh_token"
-
-    async with httpx.AsyncClient() as client:
-        try:
-            response = await client.post(
-                refresh_url,
-                json={
-                    "refresh_token": refresh_token,
-                    "provider": config.get("external_auth_provider", "azure"),
-                },
-                timeout=20.0,
-            )
-            response.raise_for_status()
-            token_data = response.json()
-        except httpx.HTTPStatusError as e:
-            log.error(f"Failed to refresh token: {e.response.text}")
-            raise HTTPException(
-                status_code=e.response.status_code,
-                detail=f"Failed to refresh token: {e.response.text}",
-            )
-        except Exception as e:
-            log.error(f"Error during token refresh: {e}")
-            raise HTTPException(status_code=500, detail="Error during token refresh")
-
-    access_token = token_data.get("access_token")
-    new_refresh_token = token_data.get("refresh_token")
+    try:
+        from solace_agent_mesh_enterprise.gateway.auth import handle_token_refresh
 
-    if not access_token:
+        return await handle_token_refresh(request, config, component)
+    except ImportError:
         raise HTTPException(
-            status_code=400, detail="Access token not in response from refresh service"
+            status_code=501,
+            detail="OAuth authentication requires enterprise package. "
+            "Install: pip install solace-agent-mesh-enterprise",
         )
 
-    session_manager = component.get_session_manager()
-    session_manager.store_auth_tokens(request, access_token, new_refresh_token)
-    log.info("Successfully refreshed and updated tokens in session.")
-
-    return {
-        "access_token": access_token,
-        "refresh_token": new_refresh_token,
-    }
-
 
 @router.get("/auth/tool/callback")
 async def auth_tool_callback(

solace_agent_mesh/gateway/http_sse/routers/config.py

@@ -1,17 +1,19 @@
 """
 API Router for providing frontend configuration.
 """
+from __future__ import annotations
 
 import logging
-from fastapi import APIRouter, Depends, HTTPException, status
 from typing import Dict, Any
+from typing import TYPE_CHECKING
+
+from fastapi import APIRouter, Depends, HTTPException, status
 
+from ..routers.dto.requests.project_requests import CreateProjectRequest
 from ....gateway.http_sse.dependencies import get_sac_component, get_api_config
-from ..routers.dto.requests.project_requests import CreateProjectRequest, UpdateProjectRequest
-from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from gateway.http_sse.component import WebUIBackendComponent
+    from ..component import WebUIBackendComponent
 
 log = logging.getLogger(__name__)
 
@@ -337,14 +339,16 @@ async def get_app_config(
             "ttsProvider": tts_provider,
         }
 
+        platform_config = component.get_config("platform_service", {})
+        platform_service_url = platform_config.get("url", "")
+
         config_data = {
-            "frontend_server_url": "",
+            "frontend_server_url": component.frontend_server_url,
+            "frontend_platform_server_url": platform_service_url,
             "frontend_auth_login_url": component.get_config(
                 "frontend_auth_login_url", ""
             ),
-            "frontend_use_authorization": component.get_config(
-                "frontend_use_authorization", False
-            ),
+            "frontend_use_authorization": component.get_config("frontend_use_authorization", False),
             "frontend_welcome_message": component.get_config(
                 "frontend_welcome_message", ""
             ),

solace_agent_mesh/gateway/http_sse/routers/dto/requests/session_requests.py

@@ -5,7 +5,7 @@ Session-related request DTOs.
 from typing import Optional
 from pydantic import BaseModel, Field
 
-from ....shared.types import SessionId, UserId
+from solace_agent_mesh.shared.utils.types import SessionId, UserId
 
 
 class GetSessionRequest(BaseModel):

solace_agent_mesh/gateway/http_sse/routers/dto/responses/base_responses.py

@@ -6,7 +6,7 @@ from typing import Any
 
 from pydantic import BaseModel, ConfigDict
 
-from ....shared import epoch_ms_to_iso8601
+from solace_agent_mesh.shared.utils.timestamp_utils import epoch_ms_to_iso8601
 
 
 class BaseTimestampResponse(BaseModel):

solace_agent_mesh/gateway/http_sse/routers/dto/responses/session_responses.py

@@ -4,8 +4,8 @@ Session-related response DTOs.
 
 from pydantic import BaseModel, ConfigDict, Field
 
-from ....shared.pagination import PaginationMeta
-from ....shared.types import SessionId, UserId
+from solace_agent_mesh.shared.api.pagination import PaginationMeta
+from solace_agent_mesh.shared.utils.types import SessionId, UserId
 from .base_responses import BaseTimestampResponse
 
 

solace_agent_mesh/gateway/http_sse/routers/feedback.py

@@ -22,8 +22,8 @@ from ..repository.entities import Feedback
 from ..repository.feedback_repository import FeedbackRepository
 from ..repository.interfaces import ITaskRepository
 from ..services.feedback_service import FeedbackService
-from ..shared.pagination import PaginationParams
-from ..shared.types import UserId
+from solace_agent_mesh.shared.api.pagination import PaginationParams
+from solace_agent_mesh.shared.utils.types import UserId
 
 router = APIRouter()
 log = logging.getLogger(__name__)

solace_agent_mesh/gateway/http_sse/routers/people.py

@@ -9,8 +9,8 @@ from fastapi import APIRouter, Depends, Query
 
 from ..dependencies import get_people_service
 from ..services.people_service import PeopleService
-from ..shared.pagination import DataResponse
-from ..shared.response_utils import create_data_response
+from solace_agent_mesh.shared.api.pagination import DataResponse
+from solace_agent_mesh.shared.api.response_utils import create_data_response
 
 log = logging.getLogger(__name__)
 

solace_agent_mesh/gateway/http_sse/routers/projects.py

@@ -1,6 +1,7 @@
 """
 Project API controller using 3-tiered architecture.
 """
+from __future__ import annotations
 
 import json
 from typing import List, Optional, Dict, Any
@@ -20,7 +21,7 @@ from solace_ai_connector.common.log import log
 
 from ..dependencies import get_project_service, get_sac_component, get_api_config, get_db
 from ..services.project_service import ProjectService
-from ..shared.auth_utils import get_current_user
+from solace_agent_mesh.shared.api.auth_utils import get_current_user
 from ....common.a2a.types import ArtifactInfo
 from typing import TYPE_CHECKING
 

solace_agent_mesh/gateway/http_sse/routers/prompts.py

@@ -1,6 +1,7 @@
 """
 Prompts API router for prompt library feature.
 """
+from __future__ import annotations
 
 import uuid
 from typing import List, Optional, Dict, Any, Literal
@@ -27,7 +28,7 @@ from .dto.prompt_dto import (
     PromptImportRequest,
     PromptImportResponse,
 )
-from ..shared import now_epoch_ms
+from solace_agent_mesh.shared.utils.timestamp_utils import now_epoch_ms
 from solace_ai_connector.common.log import log
 
 from typing import TYPE_CHECKING

solace_agent_mesh/gateway/http_sse/routers/sessions.py

@@ -6,9 +6,9 @@ from sqlalchemy.orm import Session
 
 from ..dependencies import get_session_business_service, get_db
 from ..services.session_service import SessionService
-from ..shared.auth_utils import get_current_user
-from ..shared.pagination import DataResponse, PaginatedResponse, PaginationParams
-from ..shared.response_utils import create_data_response
+from solace_agent_mesh.shared.api.auth_utils import get_current_user
+from solace_agent_mesh.shared.api.pagination import DataResponse, PaginatedResponse, PaginationParams
+from solace_agent_mesh.shared.api.response_utils import create_data_response
 from .dto.requests.session_requests import (
     GetSessionRequest,
     UpdateSessionRequest,

solace_agent_mesh/gateway/http_sse/routers/speech.py

@@ -11,7 +11,7 @@ from solace_ai_connector.common.log import log
 
 from ..services.audio_service import AudioService
 from ..dependencies import get_audio_service
-from ..shared.auth_utils import get_current_user
+from solace_agent_mesh.shared.api.auth_utils import get_current_user
 
 
 router = APIRouter()

solace_agent_mesh/gateway/http_sse/routers/tasks.py

@@ -2,6 +2,7 @@
 API Router for submitting and managing tasks to agents.
 Includes background task status endpoints.
 """
+from __future__ import annotations
 
 import logging
 from typing import TYPE_CHECKING
@@ -43,8 +44,8 @@ from ....gateway.http_sse.repository.task_repository import TaskRepository
 from ....gateway.http_sse.services.session_service import SessionService
 from ....gateway.http_sse.services.task_service import TaskService
 from ....gateway.http_sse.session_manager import SessionManager
-from ....gateway.http_sse.shared.pagination import PaginationParams
-from ....gateway.http_sse.shared.types import UserId
+from solace_agent_mesh.shared.api.pagination import PaginationParams
+from solace_agent_mesh.shared.utils.types import UserId
 from ..utils.stim_utils import create_stim_from_task_data
 
 if TYPE_CHECKING:

solace_agent_mesh/gateway/http_sse/routers/users.py

@@ -9,7 +9,7 @@ from typing import Any
 from fastapi import APIRouter, Depends, Query
 
 from ..dependencies import get_config_resolver, get_user_config
-from ..shared.auth_utils import get_current_user
+from solace_agent_mesh.shared.api.auth_utils import get_current_user
 from ....common.middleware.config_resolver import ConfigResolver
 
 log = logging.getLogger(__name__)

solace_agent_mesh/gateway/http_sse/routers/visualization.py

@@ -1,6 +1,7 @@
 """
 API Router for managing A2A message visualization streams.
 """
+from __future__ import annotations
 
 import logging
 import asyncio
@@ -47,7 +48,7 @@ class SubscriptionTarget(BaseModel):
             "my_a2a_messages",
             "current_namespace_a2a_messages",
             "namespace_a2a_messages",
-            "agent_a2a_messages",
+            "agent_a2a_messages"
         ],
     )
     identifier: Optional[str] = Field(

solace_agent_mesh/gateway/http_sse/services/background_task_monitor.py

@@ -9,7 +9,7 @@ from sqlalchemy.orm import Session as DBSession
 
 from ....gateway.http_sse.repository.task_repository import TaskRepository
 from ....gateway.http_sse.services.task_service import TaskService
-from ....gateway.http_sse.shared import now_epoch_ms
+from ....shared.utils.timestamp_utils import now_epoch_ms
 
 log = logging.getLogger(__name__)
 

solace_agent_mesh/gateway/http_sse/services/data_retention_service.py

@@ -10,7 +10,7 @@ from sqlalchemy.orm import Session as DBSession
 
 from ..repository.feedback_repository import FeedbackRepository
 from ..repository.task_repository import TaskRepository
-from ..shared import now_epoch_ms
+from solace_agent_mesh.shared.utils.timestamp_utils import now_epoch_ms
 
 log = logging.getLogger(__name__)
 

solace_agent_mesh/gateway/http_sse/services/feedback_service.py

@@ -11,7 +11,7 @@ from sqlalchemy.orm import Session as DBSession
 
 from ..repository.entities import Feedback
 from ..repository.feedback_repository import FeedbackRepository
-from ..shared import now_epoch_ms
+from solace_agent_mesh.shared.utils.timestamp_utils import now_epoch_ms
 from ..utils.stim_utils import create_stim_from_task_data
 
 # The FeedbackPayload is defined in the router, this creates a forward reference

solace_agent_mesh/gateway/http_sse/services/session_service.py

@@ -11,10 +11,10 @@ from ..repository import (
 from ..repository.chat_task_repository import ChatTaskRepository
 from ..repository.task_repository import TaskRepository
 from ..repository.entities import ChatTask
-from ..shared.enums import SenderType
-from ..shared.types import SessionId, UserId
-from ..shared import now_epoch_ms
-from ..shared.pagination import PaginationParams, PaginatedResponse, get_pagination_or_default
+from solace_agent_mesh.shared.utils.enums import SenderType
+from solace_agent_mesh.shared.utils.types import SessionId, UserId
+from solace_agent_mesh.shared.utils.timestamp_utils import now_epoch_ms
+from solace_agent_mesh.shared.api.pagination import PaginationParams, PaginatedResponse, get_pagination_or_default
 
 log = logging.getLogger(__name__)
 

solace_agent_mesh/gateway/http_sse/services/task_logger_service.py

@@ -21,7 +21,7 @@ from sqlalchemy.orm import Session as DBSession
 from ....common import a2a
 from ..repository.entities import Task, TaskEvent
 from ..repository.task_repository import TaskRepository
-from ..shared import now_epoch_ms
+from solace_agent_mesh.shared.utils.timestamp_utils import now_epoch_ms
 
 log = logging.getLogger(__name__)
 

solace_agent_mesh/services/platform/__init__.py

@@ -1,18 +1,29 @@
 """
 Platform Service for Solace Agent Mesh.
 
-Provides REST API for platform configuration management:
-- Agents
-- Connectors
-- Toolsets
-- Deployments
-- AI Assistant
+ARCHITECTURE: Service vs Gateway
+--------------------------------
+SERVICES provide internal platform functionality (this component).
+GATEWAYS handle external communication channels (http_sse, slack, webhook, etc.).
 
-This service is NOT a gateway - it does not:
-- Manage chat sessions
-- Submit tasks to agents via A2A
-- Handle artifacts or embeds
-- Communicate with the agent mesh
+Platform Service Responsibilities:
+- REST API for platform configuration management
+- Agent Builder (CRUD operations on agents)
+- Connector management (CRUD operations on connectors)
+- Toolset discovery and management
+- Deployment orchestration (deploy/update/undeploy agents)
+- AI Assistant (AI-powered configuration help)
+- Deployer heartbeat monitoring (track deployer availability)
+- Background deployment status checking (verify agent deployments succeed)
 
-It only validates OAuth2 tokens and performs CRUD operations on platform data.
+Message Communication (Direct Messaging - Not A2A Protocol):
+- PUBLISHES: Deployment commands (direct publishing) to deployer ({namespace}/deployer/agent/...)
+- RECEIVES: Deployer heartbeats ({namespace}/deployer/heartbeat)
+- RECEIVES: Agent cards for deployment monitoring ({namespace}/a2a/agent-cards)
+
+What Platform Service is NOT:
+- NOT a chat interface (no user sessions)
+- NOT for task submission to agents (no orchestration of user tasks)
+- NOT for artifact management (no chat artifacts or embeds)
+- NOT for end-user communication (admin/platform operations only)
 """