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

solace_agent_mesh/gateway/http_sse/alembic/versions/20251024_add_projects_table.py

@@ -0,0 +1,135 @@
+"""Add projects table and project_id to sessions
+
+Revision ID: add_projects_table_001
+Revises: 20251015_session_idx
+Create Date: 2025-10-24 10:00:00.000000
+
+"""
+from typing import Sequence, Union
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy import inspect
+
+# revision identifiers, used by Alembic.
+revision: str = 'add_projects_table_001'
+down_revision: Union[str, Sequence[str], None] = '20251015_session_idx'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Create projects table and add project_id to sessions."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    existing_tables = inspector.get_table_names()
+    dialect_name = bind.dialect.name
+
+    # Create projects table if it doesn't exist
+    # Note: This table is created without is_global and created_by_user_id columns
+    # as those were removed in the squashed migration
+    if 'projects' not in existing_tables:
+        op.create_table('projects',
+            sa.Column('id', sa.String(), nullable=False),
+            sa.Column('name', sa.String(), nullable=False),
+            sa.Column('user_id', sa.String(), nullable=False),
+            sa.Column('description', sa.Text(), nullable=True),
+            sa.Column('system_prompt', sa.Text(), nullable=True),
+            sa.Column('created_at', sa.BigInteger(), nullable=False),
+            sa.Column('updated_at', sa.BigInteger(), nullable=True),
+            sa.PrimaryKeyConstraint('id')
+        )
+
+    # Add project_id column to sessions if it doesn't exist
+    sessions_columns = [col['name'] for col in inspector.get_columns('sessions')]
+    if 'project_id' not in sessions_columns:
+        if dialect_name == 'sqlite':
+            # SQLite doesn't support ALTER TABLE ADD CONSTRAINT, recreate table
+            op.create_table(
+                'sessions_new',
+                sa.Column('id', sa.String(), nullable=False),
+                sa.Column('name', sa.String(), nullable=True),
+                sa.Column('user_id', sa.String(), nullable=False),
+                sa.Column('agent_id', sa.String(), nullable=True),
+                sa.Column('created_time', sa.BigInteger(), nullable=False),
+                sa.Column('updated_time', sa.BigInteger(), nullable=False),
+                sa.Column('project_id', sa.String(), nullable=True),
+                sa.ForeignKeyConstraint(['project_id'], ['projects.id']),
+                sa.PrimaryKeyConstraint('id')
+            )
+
+            # Copy data from old table
+            op.execute("""
+                INSERT INTO sessions_new (id, name, user_id, agent_id, created_time, updated_time)
+                SELECT id, name, user_id, agent_id, created_time, updated_time
+                FROM sessions
+            """)
+
+            # Drop old table
+            op.drop_table('sessions')
+
+            # Rename new table
+            op.rename_table('sessions_new', 'sessions')
+
+            # Recreate indexes
+            op.create_index('ix_sessions_user_id', 'sessions', ['user_id'])
+            op.create_index('ix_sessions_project_id', 'sessions', ['project_id'])
+        else:
+            # PostgreSQL, MySQL - standard ALTER TABLE
+            op.add_column('sessions', sa.Column('project_id', sa.String(), nullable=True))
+            op.create_index('ix_sessions_project_id', 'sessions', ['project_id'])
+            op.create_foreign_key(
+                'fk_sessions_project_id',
+                'sessions',
+                'projects',
+                ['project_id'],
+                ['id']
+            )
+
+
+def downgrade() -> None:
+    """Downgrade schema - removes project-related changes."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    dialect_name = bind.dialect.name
+
+    # Drop project_id column from sessions if it exists
+    sessions_columns = [col['name'] for col in inspector.get_columns('sessions')]
+    if 'project_id' in sessions_columns:
+        if dialect_name == 'sqlite':
+            # SQLite doesn't support ALTER TABLE DROP CONSTRAINT, recreate table
+            op.create_table(
+                'sessions_old',
+                sa.Column('id', sa.String(), nullable=False),
+                sa.Column('name', sa.String(), nullable=True),
+                sa.Column('user_id', sa.String(), nullable=False),
+                sa.Column('agent_id', sa.String(), nullable=True),
+                sa.Column('created_time', sa.BigInteger(), nullable=False),
+                sa.Column('updated_time', sa.BigInteger(), nullable=False),
+                sa.PrimaryKeyConstraint('id')
+            )
+
+            # Copy data from current table (excluding project_id)
+            op.execute("""
+                INSERT INTO sessions_old (id, name, user_id, agent_id, created_time, updated_time)
+                SELECT id, name, user_id, agent_id, created_time, updated_time
+                FROM sessions
+            """)
+
+            # Drop current table
+            op.drop_table('sessions')
+
+            # Rename old table back
+            op.rename_table('sessions_old', 'sessions')
+
+            # Recreate index on user_id
+            op.create_index('ix_sessions_user_id', 'sessions', ['user_id'])
+        else:
+            # PostgreSQL, MySQL - standard ALTER TABLE
+            op.drop_constraint('fk_sessions_project_id', 'sessions', type_='foreignkey')
+            op.drop_index('ix_sessions_project_id', 'sessions')
+            op.drop_column('sessions', 'project_id')
+
+    # Drop projects table if it exists
+    existing_tables = inspector.get_table_names()
+    if 'projects' in existing_tables:
+        op.drop_table('projects')

solace_agent_mesh/gateway/http_sse/alembic/versions/versions_llm.txt

@@ -1,14 +1,14 @@
 # DEVELOPER GUIDE: versions
 
 ## Quick Summary
-This directory contains Alembic database migration files for the HTTP SSE gateway. These migrations handle the evolution of the database schema, including initial table creation, performance optimization through indexing, timestamp format standardization, and task management features with token usage tracking.
+This directory contains Alembic database migration files for the HTTP SSE gateway. These migrations handle the evolution of the database schema, including initial table creation, performance optimization through indexing, timestamp format standardization, and task management features.
 
 ## Files Overview
 - `20250910_d5b3f8f2e9a0_create_initial_database.py` - Creates the initial database schema with sessions and chat_messages tables
 - `20250911_b1c2d3e4f5g6_add_database_indexes.py` - Adds performance indexes for common query patterns
 - `20250916_f6e7d8c9b0a1_convert_timestamps_to_epoch_and_align_columns.py` - Converts datetime columns to epoch milliseconds and standardizes column names
-- `20250928_079e06e9b448_add_tasks_task_events_and_feedback_.py` - Adds task management tables (tasks, task_events, feedback)
-- `20250930_add_token_usage_to_tasks.py` - Adds token usage tracking columns to tasks table
+- `20251006_98882922fa59_add_tasks_events_feedback_chat_tasks.py` - Adds task management tables and replaces chat_messages with chat_tasks
+- `20251015_add_session_performance_indexes.py` - Adds optimized composite indexes for better query performance
 - `versions_llm.txt` - LLM-generated documentation for the versions directory
 
 ## Developer API Reference
@@ -85,46 +85,46 @@ alembic upgrade f6e7d8c9b0a1
 alembic downgrade b1c2d3e4f5g6
 ```
 
-### 20250928_079e06e9b448_add_tasks_task_events_and_feedback_.py
-**Purpose:** Adds task management functionality with tables for tracking tasks, events, and user feedback
+### 20251006_98882922fa59_add_tasks_events_feedback_chat_tasks.py
+**Purpose:** Major schema update that adds task management functionality and replaces chat_messages with chat_tasks
 **Import:** This is an Alembic migration file - not directly imported
 
 **Functions:**
-- `upgrade() -> None` - Creates tasks, task_events, and feedback tables with appropriate indexes and foreign key relationships
-- `downgrade() -> None` - Drops all task-related tables and recreates original session/message indexes
+- `upgrade() -> None` - Creates tasks, task_events, feedback, and chat_tasks tables; drops old chat_messages table and indexes
+- `downgrade() -> None` - Recreates chat_messages table and drops all task-related tables
 
 **Constants/Variables:**
-- `revision: str` - Migration identifier "079e06e9b448"
+- `revision: str` - Migration identifier "98882922fa59"
 - `down_revision: Union[str, Sequence[str], None]` - Previous migration "f6e7d8c9b0a1"
 
 **Usage Examples:**
 ```bash
 # Run this migration
-alembic upgrade 079e06e9b448
+alembic upgrade 98882922fa59
 
 # Rollback this migration
 alembic downgrade f6e7d8c9b0a1
 ```
 
-### 20250930_add_token_usage_to_tasks.py
-**Purpose:** Adds token usage tracking columns to the tasks table for monitoring AI model consumption
+### 20251015_add_session_performance_indexes.py
+**Purpose:** Performance optimization migration that adds composite indexes for optimal query patterns
 **Import:** This is an Alembic migration file - not directly imported
 
 **Functions:**
-- `upgrade() -> None` - Adds token usage columns (total_input_tokens, total_output_tokens, total_cached_input_tokens, token_usage_details) to tasks table
-- `downgrade() -> None` - Removes all token usage columns from tasks table
+- `upgrade() -> None` - Creates optimized composite indexes for user sessions, chat tasks, and task events; removes unused text index
+- `downgrade() -> None` - Removes performance indexes and recreates the original text index
 
 **Constants/Variables:**
-- `revision: str` - Migration identifier "20250930_token_usage"
-- `down_revision: Union[str, None]` - Previous migration "079e06e9b448"
+- `revision: str` - Migration identifier "20251015_session_idx"
+- `down_revision: str | Sequence[str] | None` - Previous migration "98882922fa59"
 
 **Usage Examples:**
 ```bash
 # Run this migration
-alembic upgrade 20250930_token_usage
+alembic upgrade 20251015_session_idx
 
 # Rollback this migration
-alembic downgrade 079e06e9b448
+alembic downgrade 98882922fa59
 
 # Run all migrations to latest
 alembic upgrade head
@@ -139,17 +139,23 @@ alembic history
 alembic revision --autogenerate -m "description"
 ```
 
-**Database Schema After All Migrations:**
+**Final Database Schema After All Migrations:**
 ```sql
 -- Core tables with epoch millisecond timestamps:
 -- sessions: id, name, user_id, agent_id, created_time, updated_time
--- chat_messages: id, session_id, message, sender_type, sender_name, created_time
+-- chat_tasks: id, session_id, user_id, user_message, message_bubbles, task_metadata, created_time, updated_time
 
 -- Task management tables:
 -- tasks: id, user_id, start_time, end_time, status, initial_request_text,
 --        total_input_tokens, total_output_tokens, total_cached_input_tokens, token_usage_details
 -- task_events: id, task_id, user_id, created_time, topic, direction, payload
 -- feedback: id, session_id, task_id, user_id, rating, comment, created_time
+
+-- Performance indexes:
+-- ix_sessions_user_id, ix_sessions_user_updated
+-- ix_chat_tasks_session_user_created
+-- ix_tasks_user_start_time, ix_task_events_task_created
+-- ix_feedback_task_id, ix_feedback_user_id, ix_feedback_created_time
 ```
 
-# content_hash: 93bf894b6e5451feedc0b65a723765d7e6798c6912fb3fc34eaf9548d27e4195
+# content_hash: c1d3d7f6855eba76bdabdb5cc0c7253756faf1583ef349dc503418c611a23bd2

solace_agent_mesh/gateway/http_sse/app.py

@@ -75,20 +75,6 @@ class WebUIBackendApp(BaseGatewayApp):
             "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": "model",
             "required": False,

solace_agent_mesh/gateway/http_sse/component.py

@@ -95,7 +95,16 @@ class WebUIBackendComponent(BaseGatewayComponent):
         app_config = component_config.get("app_config", {})
         resolve_uris = app_config.get("resolve_artifact_uris_in_gateway", True)
 
-        super().__init__(resolve_artifact_uris_in_gateway=resolve_uris, **kwargs)
+        # HTTP SSE gateway configuration:
+        # - supports_inline_artifact_resolution=True: Artifacts are converted to FileParts
+        #   during embed resolution and rendered inline in the web UI
+        # - filter_tool_data_parts=False: Web UI displays all parts including tool execution details
+        super().__init__(
+            resolve_artifact_uris_in_gateway=resolve_uris,
+            supports_inline_artifact_resolution=True,
+            filter_tool_data_parts=False,
+            **kwargs
+        )
         log.info("%s Initializing Web UI Backend Component...", self.log_identifier)
 
         try:
@@ -258,10 +267,8 @@ class WebUIBackendComponent(BaseGatewayComponent):
             if self.database_url:
                 # SessionLocal will be initialized later in setup_dependencies
                 # We'll pass a lambda that returns SessionLocal when called
-                session_factory = (
-                    lambda: dependencies.SessionLocal()
-                    if dependencies.SessionLocal
-                    else None
+                session_factory = lambda: (
+                    dependencies.SessionLocal() if dependencies.SessionLocal else None
                 )
 
             self.data_retention_service = DataRetentionService(
@@ -1711,9 +1718,9 @@ class WebUIBackendComponent(BaseGatewayComponent):
                 (summary_str[:100] + "...") if len(summary_str) > 100 else summary_str
             )
         except Exception:
-            details["payload_summary"]["params_preview"] = (
-                "[Could not serialize payload]"
-            )
+            details["payload_summary"][
+                "params_preview"
+            ] = "[Could not serialize payload]"
 
         return details
 
@@ -1935,52 +1942,6 @@ class WebUIBackendComponent(BaseGatewayComponent):
         """Returns the instance of the ConfigResolver."""
         return self._config_resolver
 
-    async def _resolve_embeds_for_persistence(
-        self, message_content: str, session_id: str, user_id: str, log_identifier: str
-    ) -> str:
-        """
-        Resolves embeds in a message for database storage.
-        Returns the resolved text.
-
-        Args:
-            message_content: The message text that may contain embeds
-            session_id: The A2A session ID
-            user_id: The user ID
-            log_identifier: Logging identifier
-
-        Returns:
-            The message with embeds resolved (or original if resolution fails)
-        """
-        try:
-            embed_context = {
-                "artifact_service": self.shared_artifact_service,
-                "session_context": {
-                    "app_name": self.gateway_id,
-                    "user_id": user_id,
-                    "session_id": session_id,
-                },
-                "config": self.get_embed_config(),
-            }
-
-            resolved_text, _, _ = await resolve_embeds_in_string(
-                text=message_content,
-                context=embed_context,
-                resolver_func=evaluate_embed,
-                types_to_resolve=EARLY_EMBED_TYPES,
-                log_identifier=log_identifier,
-                config=embed_context["config"],
-            )
-
-            return resolved_text
-
-        except Exception as e:
-            log.warning(
-                "%s Error resolving embeds for storage: %s. Using original message.",
-                log_identifier,
-                e,
-            )
-            return message_content
-
     def _start_listener(self) -> None:
         """
         GDK Hook: Starts the FastAPI/Uvicorn server.
@@ -2190,7 +2151,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
             )
             return
 
-        log.debug(
+        log.info(
             "%s Sending final response for A2A Task ID %s to SSE Task ID %s.",
             log_id_prefix,
             a2a_task_id,
@@ -2206,7 +2167,7 @@ class WebUIBackendComponent(BaseGatewayComponent):
             await self.sse_manager.send_event(
                 task_id=sse_task_id, event_data=sse_payload, event_type="final_response"
             )
-            log.info(
+            log.debug(
                 "%s Successfully sent final_response via SSE for A2A Task ID %s.",
                 log_id_prefix,
                 a2a_task_id,

solace_agent_mesh/gateway/http_sse/components/components_llm.txt

@@ -102,4 +102,4 @@ if not visualization_queue.empty():
 # }
 ```
 
-# content_hash: 6add2167cb9fdee9ed3e46635d6b4e9391cf062dde9e14dba250a3f33c6c4f73
+# content_hash: bb6b96e24d423d56a25e3c84eb411d061ad160ea98fc2ff83da46e3f317d34fb

solace_agent_mesh/gateway/http_sse/dependencies.py

@@ -5,9 +5,11 @@ managed by the WebUIBackendComponent.
 
 import logging
 from collections.abc import Callable, Generator
-from typing import TYPE_CHECKING, Any
+from contextlib import contextmanager
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any, Optional
 
-from fastapi import Depends, HTTPException, Request, status
+from fastapi import Depends, HTTPException, Request, status, Path, Query
 from sqlalchemy import create_engine
 from sqlalchemy.orm import Session, sessionmaker
 
@@ -17,7 +19,7 @@ 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.data_retention_service import DataRetentionService
+from ...gateway.http_sse.services.project_service import ProjectService
 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
@@ -26,6 +28,7 @@ from ...gateway.http_sse.session_manager import SessionManager
 from ...gateway.http_sse.sse_manager import SSEManager
 from .repository import SessionRepository
 from .repository.interfaces import ITaskRepository
+from .repository.project_repository import ProjectRepository
 from .repository.task_repository import TaskRepository
 from .services.session_service import SessionService
 
@@ -587,6 +590,21 @@ def get_db_optional() -> Generator[Session | None, None, None]:
         finally:
             db.close()
 
+def get_project_service(
+    component: "WebUIBackendComponent" = Depends(get_sac_component),
+) -> ProjectService:
+    """Dependency factory for ProjectService."""
+    return ProjectService(component=component)
+
+
+def get_project_service_optional(
+    component: "WebUIBackendComponent" = Depends(get_sac_component),
+) -> ProjectService | None:
+    """Optional project service dependency that returns None if database is not configured."""
+    if SessionLocal is None:
+        log.debug("Database not configured, projects unavailable")
+        return None
+    return ProjectService(component=component)
 
 def get_session_business_service_optional(
     component: "WebUIBackendComponent" = Depends(get_sac_component),

solace_agent_mesh/gateway/http_sse/http_sse_llm.txt

@@ -155,8 +155,8 @@ from solace_agent_mesh.gateway.http_sse.components.task_logger_forwarder import
 **Import Examples:**
 ```python
 from solace_agent_mesh.gateway.http_sse.repository import (
-    ISessionRepository, IMessageRepository, SessionRepository, MessageRepository,
-    Session, Message, SessionHistory, Base, SessionModel, MessageModel
+    ISessionRepository, ITaskRepository, SessionRepository, TaskRepository,
+    Session, Task, SessionModel, TaskModel, Base
 )
 ```
 
@@ -275,7 +275,7 @@ async def my_custom_endpoint(
     return {"agents": len(agents), "user": user_id}
 ```
 
-### 3. Working with Sessions and Messages
+### 3. Working with Sessions and Tasks
 
 ```python
 from fastapi import Depends
@@ -286,14 +286,14 @@ from solace_agent_mesh.gateway.http_sse.dependencies import (
 from solace_agent_mesh.gateway.http_sse.services.session_service import SessionService
 from solace_agent_mesh.gateway.http_sse.shared.enums import SenderType
 
-@router.post("/sessions/{session_id}/messages")
-async def add_message_to_session(
+@router.post("/sessions/{session_id}/tasks")
+async def save_task_to_session(
     session_id: str,
-    message_text: str,
+    task_data: dict,
     user_id: str = Depends(get_user_id),
     db: Session = Depends(get_db),
     session_service: SessionService = Depends(get_session_business_service_optional)
 ):
-    if session_
+    if session_service and db
 
-# content_hash: 1ec069cbf74863d29b45cd3b1031e6ab820e1b30d7c8b631488a54630a936828
+# content_hash: 7da6d9233d06a0b9137a843efe5238a14d1762ee3bba437875b5d915634e0f63

solace_agent_mesh/gateway/http_sse/main.py

@@ -1,9 +1,14 @@
 import logging
 import os
 from pathlib import Path
-from typing import TYPE_CHECKING
 
 import httpx
+import sqlalchemy as sa
+from fastapi import FastAPI, HTTPException
+from fastapi import Request as FastAPIRequest
+from fastapi import status
+from typing import TYPE_CHECKING
+
 import sqlalchemy as sa
 from a2a.types import InternalError, JSONRPCError
 from a2a.types import JSONRPCResponse as A2AJSONRPCResponse
@@ -18,23 +23,35 @@ from fastapi.responses import JSONResponse
 from starlette.middleware.sessions import SessionMiddleware
 from starlette.staticfiles import StaticFiles
 
+from .routers.sessions import router as session_router
+from .routers.tasks import router as task_router
+from .routers.users import router as user_router
 from ...common import a2a
 from ...gateway.http_sse import dependencies
-from ...gateway.http_sse.routers import (
+from .routers import (
     agent_cards,
     artifacts,
     auth,
     config,
+    feedback,
     people,
     sse,
-    tasks,
     visualization,
-    feedback,
+    projects,
 )
 from .routers.sessions import router as session_router
 from .routers.tasks import router as task_router
 from .routers.users import router as user_router
 
+from alembic import command
+from alembic.config import Config
+
+from a2a.types import InternalError, InvalidRequestError, JSONRPCError
+from a2a.types import JSONRPCResponse as A2AJSONRPCResponse
+from ...common import a2a
+from ...gateway.http_sse import dependencies
+
+
 if TYPE_CHECKING:
     from gateway.http_sse.component import WebUIBackendComponent
 
@@ -587,7 +604,7 @@ def _setup_routers() -> None:
     app.include_router(user_router, prefix=f"{api_prefix}/users", tags=["Users"])
     app.include_router(config.router, prefix=api_prefix, tags=["Config"])
     app.include_router(agent_cards.router, prefix=api_prefix, tags=["Agent Cards"])
-    app.include_router(tasks.router, prefix=api_prefix, tags=["Tasks"])
+    app.include_router(task_router, prefix=api_prefix, tags=["Tasks"])
     app.include_router(sse.router, prefix=f"{api_prefix}/sse", tags=["SSE"])
     app.include_router(
         artifacts.router, prefix=f"{api_prefix}/artifacts", tags=["Artifacts"]
@@ -599,6 +616,7 @@ def _setup_routers() -> None:
     )
     app.include_router(people.router, prefix=api_prefix, tags=["People"])
     app.include_router(auth.router, prefix=api_prefix, tags=["Auth"])
+    app.include_router(projects.router, prefix=api_prefix, tags=["Projects"])
     app.include_router(feedback.router, prefix=api_prefix, tags=["Feedback"])
     log.info("Legacy routers mounted for endpoints not yet migrated")
 

solace_agent_mesh/gateway/http_sse/repository/__init__.py

@@ -3,10 +3,20 @@ Repository layer containing all data access logic organized by entity type.
 """
 
 # Interfaces
-from .interfaces import ISessionRepository
+from .interfaces import (
+    IChatTaskRepository,
+    IFeedbackRepository,
+    IProjectRepository,
+    ISessionRepository,
+    ITaskRepository,
+)
 
 # Implementations
+from .chat_task_repository import ChatTaskRepository
+from .feedback_repository import FeedbackRepository
+from .project_repository import ProjectRepository
 from .session_repository import SessionRepository
+from .task_repository import TaskRepository
 
 # Entities (re-exported for convenience)
 from .entities.session import Session
@@ -17,9 +27,17 @@ from .models.session_model import SessionModel
 
 __all__ = [
     # Interfaces
+    "IChatTaskRepository",
+    "IFeedbackRepository",
+    "IProjectRepository",
     "ISessionRepository",
+    "ITaskRepository",
     # Implementations
+    "ChatTaskRepository",
+    "FeedbackRepository",
+    "ProjectRepository",
     "SessionRepository",
+    "TaskRepository",
     # Entities
     "Session",
     # Models