solace-agent-mesh 1.4.2__py3-none-any.whl → 1.4.4__py3-none-any.whl

solace_agent_mesh/gateway/http_sse/app.py

@@ -3,16 +3,15 @@ Custom Solace AI Connector App class for the Web UI Backend.
 Defines configuration schema and programmatically creates the WebUIBackendComponent.
 """
 
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List
 import os
 from alembic import command
 from alembic.config import Config
-from pydantic import Field, ValidationError
 from solace_ai_connector.common.log import log
 
 from ...gateway.http_sse.component import WebUIBackendComponent
 
-from ...gateway.base.app import BaseGatewayApp, BaseGatewayAppConfig
+from ...gateway.base.app import BaseGatewayApp
 from ...gateway.base.component import BaseGatewayComponent
 
 
@@ -22,88 +21,6 @@ info = {
 }
 
 
-class WebUIBackendAppConfig(BaseGatewayAppConfig):
-    """Pydantic model for the Web UI Backend application configuration."""
-
-    session_secret_key: str = Field(
-        ..., description="Secret key for signing web user sessions."
-    )
-    fastapi_host: str = Field(
-        default="127.0.0.1", description="Host address for the embedded FastAPI server."
-    )
-    fastapi_port: int = Field(
-        default=8000, description="Port for the embedded FastAPI server."
-    )
-    fastapi_https_port: int = Field(
-        default=8443, description="Port for the embedded FastAPI server when SSL is enabled."
-    )
-    cors_allowed_origins: List[str] = Field(
-        default=["*"], description="List of allowed origins for CORS requests."
-    )
-    sse_max_queue_size: int = Field(
-        default=200,
-        description="Maximum size of the SSE connection queues. Adjust based on expected load.",
-    )
-    sse_buffer_max_age_seconds: int = Field(
-        default=600,  # 10 minutes
-        description="Maximum age in seconds for a pending event buffer before it is cleaned up.",
-    )
-    sse_buffer_cleanup_interval_seconds: int = Field(
-        default=300,  # 5 minutes
-        description="How often to run the cleanup task for stale pending event buffers.",
-    )
-    resolve_artifact_uris_in_gateway: bool = Field(
-        default=True,
-        description="If true, the gateway will resolve artifact:// URIs found in A2A messages and embed the content as bytes before sending to the UI. If false, URIs are passed through.",
-    )
-    system_purpose: str = Field(
-        default="",
-        description="Detailed description of the system's overall purpose, to be optionally used by agents.",
-    )
-    response_format: str = Field(
-        default="",
-        description="General guidelines on how agent responses should be structured, to be optionally used by agents.",
-    )
-    frontend_welcome_message: str = Field(
-        default="Hi! How can I help?",
-        description="Initial welcome message displayed in the chat.",
-    )
-    frontend_bot_name: str = Field(
-        default="A2A Agent", description="Name displayed for the bot/agent in the UI."
-    )
-    frontend_collect_feedback: bool = Field(
-        default=False, description="Enable/disable the feedback buttons in the UI."
-    )
-    frontend_auth_login_url: str = Field(
-        default="", description="URL for the external login page (if auth is enabled)."
-    )
-    frontend_use_authorization: bool = Field(
-        default=False, description="Tell frontend whether backend expects authorization."
-    )
-    frontend_redirect_url: str = Field(
-        default="", description="Redirect URL for OAuth flows (if auth is enabled)."
-    )
-    external_auth_callback_uri: str = Field(
-        default="", description="Redirect URI for the OIDC application."
-    )
-    external_auth_service_url: str = Field(
-        default="http://localhost:8080",
-        description="External authorization service URL for login initiation.",
-    )
-    external_auth_provider: str = Field(
-        default="", description="The external authentication provider."
-    )
-    ssl_keyfile: str = Field(
-        default="", description="The file path to the SSL private key."
-    )
-    ssl_certfile: str = Field(
-        default="", description="The file path to the SSL certificate."
-    )
-    ssl_keyfile_password: str = Field(
-        default="", description="The passphrase for the SSL private key."
-    )
-
-
 class WebUIBackendApp(BaseGatewayApp):
     """
     Custom App class for the A2A Web UI Backend.
@@ -111,8 +28,154 @@ class WebUIBackendApp(BaseGatewayApp):
     - Defines WebUI-specific configuration parameters.
     """
 
-    # This is now a placeholder. Validation is handled by the Pydantic model.
-    SPECIFIC_APP_SCHEMA_PARAMS: List[Dict[str, Any]] = []
+    SPECIFIC_APP_SCHEMA_PARAMS: List[Dict[str, Any]] = [
+        {
+            "name": "session_secret_key",
+            "required": True,
+            "type": "string",
+            "description": "Secret key for signing web user sessions.",
+        },
+        {
+            "name": "fastapi_host",
+            "required": False,
+            "type": "string",
+            "default": "127.0.0.1",
+            "description": "Host address for the embedded FastAPI server.",
+        },
+        {
+            "name": "fastapi_port",
+            "required": False,
+            "type": "integer",
+            "default": 8000,
+            "description": "Port for the embedded FastAPI server.",
+        },
+        {
+            "name": "fastapi_https_port",
+            "required": False,
+            "type": "integer",
+            "default": 8443,
+            "description": "Port for the embedded FastAPI server when SSL is enabled.",
+        },
+        {
+            "name": "cors_allowed_origins",
+            "required": False,
+            "type": "list",
+            "default": ["*"],
+            "description": "List of allowed origins for CORS requests.",
+        },
+        {
+            "name": "sse_max_queue_size",
+            "required": False,
+            "type": "integer",
+            "default": 200,
+            "description": "Maximum size of the SSE connection queues. Adjust based on expected load.",
+        },
+        {
+            "name": "resolve_artifact_uris_in_gateway",
+            "required": False,
+            "type": "boolean",
+            "default": True,
+            "description": "If true, the gateway will resolve artifact:// URIs found in A2A messages and embed the content as bytes before sending to the UI. If false, URIs are passed through.",
+        },
+        {
+            "name": "system_purpose",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "Detailed description of the system's overall purpose, to be optionally used by agents.",
+        },
+        {
+            "name": "response_format",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "General guidelines on how agent responses should be structured, to be optionally used by agents.",
+        },
+        {
+            "name": "frontend_welcome_message",
+            "required": False,
+            "type": "string",
+            "default": "Hi! How can I help?",
+            "description": "Initial welcome message displayed in the chat.",
+        },
+        {
+            "name": "frontend_bot_name",
+            "required": False,
+            "type": "string",
+            "default": "A2A Agent",
+            "description": "Name displayed for the bot/agent in the UI.",
+        },
+        {
+            "name": "frontend_collect_feedback",
+            "required": False,
+            "type": "boolean",
+            "default": False,
+            "description": "Enable/disable the feedback buttons in the UI.",
+        },
+        {
+            "name": "frontend_auth_login_url",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "URL for the external login page (if auth is enabled).",
+        },
+        {
+            "name": "frontend_use_authorization",
+            "required": False,
+            "type": "boolean",
+            "default": False,
+            "description": "Tell frontend whether backend expects authorization.",
+        },
+        {
+            "name": "frontend_redirect_url",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "Redirect URL for OAuth flows (if auth is enabled).",
+        },
+        {
+            "name": "external_auth_callback_uri",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "Redirect URI for the OIDC application.",
+        },
+        {
+            "name": "external_auth_service_url",
+            "required": False,
+            "type": "string",
+            "default": "http://localhost:8080",
+            "description": "External authorization service URL for login initiation.",
+        },
+        {
+            "name": "external_auth_provider",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "The external authentication provider.",
+        },
+        {
+            "name": "ssl_keyfile",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "The file path to the SSL private key.",
+        },
+        {
+            "name": "ssl_certfile",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "The file path to the SSL certificate.",
+        },
+        {
+            "name": "ssl_keyfile_password",
+            "required": False,
+            "type": "string",
+            "default": "",
+            "description": "The passphrase for the SSL private key.",
+        },
+    ]
 
     def __init__(self, app_info: Dict[str, Any], **kwargs):
         """
@@ -123,8 +186,7 @@ class WebUIBackendApp(BaseGatewayApp):
             "%s Initializing WebUIBackendApp...",
             app_info.get("name", "WebUIBackendApp"),
         )
-
-        super().__init__(app_info, gateway_app_config=WebUIBackendAppConfig, **kwargs)
+        super().__init__(app_info, **kwargs)
 
         try:
             
@@ -155,4 +217,4 @@ class WebUIBackendApp(BaseGatewayApp):
         log.debug("%s WebUIBackendApp initialization complete.", self.name)
 
     def _get_gateway_component_class(self) -> type[BaseGatewayComponent]:
-        return WebUIBackendComponent
+        return WebUIBackendComponent

solace_agent_mesh/gateway/http_sse/dependencies.py

@@ -22,12 +22,8 @@ from ...gateway.http_sse.services.people_service import PeopleService
 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 Message, MessageRepository, SessionRepository
 from .services.session_service import SessionService
-from .repository import (
-    Message,
-    MessageRepository,
-    SessionRepository,
-)
 
 try:
     from google.adk.artifacts import BaseArtifactService
@@ -369,7 +365,6 @@ def get_db() -> Generator[Session, None, None]:
         db.close()
 
 
-
 def get_session_business_service(
     db: Session = Depends(get_db),
     component: "WebUIBackendComponent" = Depends(get_sac_component),
@@ -391,20 +386,29 @@ def create_session_service_with_transaction():
     try:
         session_repository = SessionRepository(db)
         message_repository = MessageRepository(db)
-        
+
         # Create a simple data access object for transaction contexts
         # This provides the basic repository operations without business logic
         class SessionDataAccess:
             def __init__(self, session_repo, message_repo):
                 self.session_repository = session_repo
                 self.message_repository = message_repo
-            
-            def add_message_to_session(self, session_id, user_id, message, sender_type, sender_name, agent_id=None):
+
+            def add_message_to_session(
+                self,
+                session_id,
+                user_id,
+                message,
+                sender_type,
+                sender_name,
+                agent_id=None,
+            ):
                 # Simple data access - just save the message
                 from uuid import uuid4
-                from datetime import datetime, timezone
+
                 from .shared.enums import MessageType
-                
+                from .shared import now_epoch_ms
+
                 message_entity = Message(
                     id=str(uuid4()),
                     session_id=session_id,
@@ -412,36 +416,38 @@ def create_session_service_with_transaction():
                     sender_type=sender_type,
                     sender_name=sender_name,
                     message_type=MessageType.TEXT,
-                    created_at=datetime.now(timezone.utc),
+                    created_time=now_epoch_ms(),
                 )
                 return self.message_repository.save(message_entity)
-            
+
             def get_session(self, session_id, user_id):
                 # Use the session repository to find the session
                 return self.session_repository.find_user_session(session_id, user_id)
-            
-            def create_session(self, user_id, name=None, agent_id=None, session_id=None):
+
+            def create_session(
+                self, user_id, name=None, agent_id=None, session_id=None
+            ):
                 # Create a new session using the session repository
                 from uuid import uuid4
-                from datetime import datetime, timezone
+
                 from .repository.entities import Session
-                
+                from .shared import now_epoch_ms
+
                 if not session_id:
                     session_id = str(uuid4())
-                
+
                 # Leave name as None/empty - frontend will generate display name if needed
-                
-                now = datetime.now(timezone.utc)
+
+                now_ms = now_epoch_ms()
                 session = Session(
                     id=session_id,
                     user_id=user_id,
                     name=name,
                     agent_id=agent_id,
-                    created_at=now,
-                    updated_at=now,
-                    last_activity=now,
+                    created_time=now_ms,
+                    updated_time=now_ms,
                 )
-                
+
                 return self.session_repository.save(session)
 
         session_service = SessionDataAccess(session_repository, message_repository)

solace_agent_mesh/gateway/http_sse/main.py

@@ -463,6 +463,25 @@ def _setup_routers() -> None:
     app.include_router(auth.router, prefix=api_prefix, tags=["Auth"])
     log.info("Legacy routers mounted for endpoints not yet migrated")
 
+    try:
+        from solace_agent_mesh_enterprise.webui_backend.routers import get_enterprise_routers
+
+        enterprise_routers = get_enterprise_routers()
+        for router_config in enterprise_routers:
+            app.include_router(
+                router_config["router"],
+                prefix=router_config["prefix"],
+                tags=router_config["tags"]
+            )
+        log.info("Mounted %d enterprise routers", len(enterprise_routers))
+
+    except ImportError:
+        log.debug("No enterprise package detected - skipping enterprise routers")
+    except ModuleNotFoundError:
+        log.debug("Enterprise router module not found - skipping enterprise routers")
+    except Exception as e:
+        log.warning("Failed to load enterprise routers: %s", e)
+
 
 def _setup_static_files() -> None:
     current_dir = os.path.dirname(os.path.abspath(__file__))

solace_agent_mesh/gateway/http_sse/repository/entities/message.py

@@ -2,8 +2,6 @@
 Message domain entity.
 """
 
-from datetime import datetime
-
 from pydantic import BaseModel
 
 from ...shared.enums import MessageType, SenderType
@@ -12,14 +10,14 @@ from ...shared.types import MessageId, SessionId
 
 class Message(BaseModel):
     """Message domain entity with business logic."""
-    
+
     id: MessageId
     session_id: SessionId
     message: str
     sender_type: SenderType
     sender_name: str
     message_type: MessageType = MessageType.TEXT
-    created_at: datetime
+    created_time: int
 
     def validate_message_content(self) -> None:
         """Validate message content."""
@@ -38,4 +36,4 @@ class Message(BaseModel):
 
     def is_system_message(self) -> bool:
         """Check if message is a system message."""
-        return self.sender_type == SenderType.SYSTEM
+        return self.sender_type == SenderType.SYSTEM

solace_agent_mesh/gateway/http_sse/repository/entities/session.py

@@ -2,23 +2,21 @@
 Session domain entity.
 """
 
-from datetime import datetime, timezone
-
 from pydantic import BaseModel
 
+from ...shared import now_epoch_ms
 from ...shared.types import AgentId, SessionId, UserId
 
 
 class Session(BaseModel):
     """Session domain entity with business logic."""
-    
+
     id: SessionId
     user_id: UserId
     name: str | None = None
     agent_id: AgentId | None = None
-    created_at: datetime
-    updated_at: datetime | None = None
-    last_activity: datetime | None = None
+    created_time: int
+    updated_time: int | None = None
 
     def update_name(self, new_name: str) -> None:
         """Update session name with validation."""
@@ -28,13 +26,11 @@ class Session(BaseModel):
             raise ValueError("Session name cannot exceed 255 characters")
 
         self.name = new_name.strip()
-        self.updated_at = datetime.now(timezone.utc)
+        self.updated_time = now_epoch_ms()
 
     def mark_activity(self) -> None:
         """Mark session as having recent activity."""
-        self.last_activity = datetime.now(timezone.utc)
-        self.updated_at = datetime.now(timezone.utc)
-
+        self.updated_time = now_epoch_ms()
 
     def can_be_deleted_by_user(self, user_id: UserId) -> bool:
         """Check if user can delete this session."""
@@ -42,4 +38,4 @@ class Session(BaseModel):
 
     def can_be_accessed_by_user(self, user_id: UserId) -> bool:
         """Check if user can access this session."""
-        return self.user_id == user_id
+        return self.user_id == user_id

solace_agent_mesh/gateway/http_sse/repository/message_repository.py

@@ -13,7 +13,7 @@ from .models import MessageModel
 
 class MessageRepository(IMessageRepository):
     """SQLAlchemy implementation of message repository."""
-    
+
     def __init__(self, db: DBSession):
         self.db = db
 
@@ -21,18 +21,22 @@ class MessageRepository(IMessageRepository):
         self, session_id: SessionId, pagination: PaginationInfo | None = None
     ) -> list[Message]:
         """Find all messages in a session."""
-        query = self.db.query(MessageModel).filter(MessageModel.session_id == session_id)
+        query = self.db.query(MessageModel).filter(
+            MessageModel.session_id == session_id
+        )
 
         if pagination:
             offset = (pagination.page - 1) * pagination.page_size
             query = query.offset(offset).limit(pagination.page_size)
 
-        models = query.order_by(MessageModel.created_at.asc()).all()
+        models = query.order_by(MessageModel.created_time.asc()).all()
         return [self._model_to_entity(model) for model in models]
 
     def save(self, message: Message) -> Message:
         """Save or update a message."""
-        model = self.db.query(MessageModel).filter(MessageModel.id == message.id).first()
+        model = (
+            self.db.query(MessageModel).filter(MessageModel.id == message.id).first()
+        )
 
         if model:
             # Update existing
@@ -47,7 +51,7 @@ class MessageRepository(IMessageRepository):
                 message=message.message,
                 sender_type=message.sender_type.value,
                 sender_name=message.sender_name,
-                created_at=message.created_at,
+                created_time=message.created_time,
             )
             self.db.add(model)
 
@@ -74,5 +78,5 @@ class MessageRepository(IMessageRepository):
             sender_type=SenderType(model.sender_type),
             sender_name=model.sender_name,
             message_type=MessageType.TEXT,  # Default for now
-            created_at=model.created_at,
-        )
+            created_time=model.created_time,
+        )

solace_agent_mesh/gateway/http_sse/repository/models/message_model.py

@@ -2,26 +2,26 @@
 Message SQLAlchemy model.
 """
 
-from sqlalchemy import Column, DateTime, ForeignKey, String, Text
+from sqlalchemy import BigInteger, Column, ForeignKey, String, Text
 from sqlalchemy.orm import relationship
-from sqlalchemy.sql import func
 
+from ...shared import now_epoch_ms
 from .base import Base
 
 
 class MessageModel(Base):
     """SQLAlchemy model for messages."""
-    
+
     __tablename__ = "chat_messages"
-    
+
     id = Column(String, primary_key=True)
     session_id = Column(
         String, ForeignKey("sessions.id", ondelete="CASCADE"), nullable=False
     )
     message = Column(Text, nullable=False)
-    created_at = Column(DateTime, default=func.now())
+    created_time = Column(BigInteger, nullable=False, default=now_epoch_ms)
     sender_type = Column(String(50))
     sender_name = Column(String(255))
-    
+
     # Relationship to session
-    session = relationship("SessionModel", back_populates="messages")
+    session = relationship("SessionModel", back_populates="messages")

solace_agent_mesh/gateway/http_sse/repository/models/session_model.py

@@ -2,26 +2,28 @@
 Session SQLAlchemy model.
 """
 
-from sqlalchemy import Column, DateTime, String
+from sqlalchemy import BigInteger, Column, String
 from sqlalchemy.orm import relationship
-from sqlalchemy.sql import func
 
+from ...shared import now_epoch_ms
 from .base import Base
 
 
 class SessionModel(Base):
     """SQLAlchemy model for sessions."""
-    
+
     __tablename__ = "sessions"
-    
+
     id = Column(String, primary_key=True)
     name = Column(String, nullable=True)
     user_id = Column(String, nullable=False)
     agent_id = Column(String, nullable=True)
-    created_at = Column(DateTime, default=func.now())
-    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
-    
+    created_time = Column(BigInteger, nullable=False, default=now_epoch_ms)
+    updated_time = Column(
+        BigInteger, nullable=False, default=now_epoch_ms, onupdate=now_epoch_ms
+    )
+
     # Relationship to messages
     messages = relationship(
         "MessageModel", back_populates="session", cascade="all, delete-orphan"
-    )
+    )