feat: DATAGO-116432 prompt library

src/solace_agent_mesh/gateway/http_sse/repository/models/prompt_model.py

@@ -1,6 +1,5 @@
 """
 Prompt SQLAlchemy models for prompt library feature.
-Adapted for SAM fork with epoch millisecond timestamps and String IDs.
 """
 
 from enum import Enum
@@ -19,7 +18,7 @@ class PromptGroupRole(str, Enum):
 
 
 class PromptGroupModel(Base):
-    """SQLAlchemy model for prompt groups - adapted for SAM fork."""
+    """SQLAlchemy model for prompt groups"""
 
     __tablename__ = "prompt_groups"
 
@@ -81,7 +80,7 @@ def __repr__(self):
 
 
 class PromptModel(Base):
-    """SQLAlchemy model for individual prompt versions - adapted for SAM fork."""
+    """SQLAlchemy model for individual prompt versions"""
 
     __tablename__ = "prompts"
 

src/solace_agent_mesh/gateway/http_sse/routers/config.py

@@ -125,15 +125,25 @@ async def get_app_config(
                 ai_assisted_enabled = ai_assisted_config.get("enabled", True)
 
                 if ai_assisted_enabled:
-                    # Verify LLM is configured
-                    llm_model = os.getenv("LLM_SERVICE_GENERAL_MODEL_NAME")
+                    # Verify LLM is configured through the model config
+                    model_config = component.get_config("model", {})
+
+                    # Handle both dict and direct model config structures
+                    llm_model = None
+                    if isinstance(model_config, dict):
+                        general_model = model_config.get("general", {})
+                        if isinstance(general_model, dict):
+                            llm_model = general_model.get("model")
+                        if not llm_model:
+                            llm_model = model_config.get("model")
+
                     if llm_model:
                         feature_enablement["promptAIAssisted"] = True
                         log.debug("%s promptAIAssisted feature flag is enabled.", log_prefix)
                     else:
                         feature_enablement["promptAIAssisted"] = False
                         log.warning(
-                            "%s AI-assisted prompts disabled: LLM_SERVICE_GENERAL_MODEL_NAME not configured",
+                            "%s AI-assisted prompts disabled: general_model not configured",
                             log_prefix
                         )
                 else:

src/solace_agent_mesh/gateway/http_sse/routers/dto/prompt_dto.py

@@ -1,6 +1,5 @@
 """
 Pydantic models for prompt library API.
-Adapted for SAM fork with epoch millisecond timestamps.
 """
 
 from typing import Any, Dict, List, Optional

src/solace_agent_mesh/gateway/http_sse/routers/prompts.py

@@ -1,6 +1,5 @@
 """
 Prompts API router for prompt library feature.
-Adapted for SAM fork.
 """
 
 import uuid
@@ -967,9 +966,11 @@ async def delete_prompt(
 @router.get("/chat/init")
 async def init_prompt_builder_chat(
     db: Session = Depends(get_db),
+    component: "WebUIBackendComponent" = Depends(get_sac_component),
 ):
     """Initialize the prompt template builder chat"""
-    assistant = PromptBuilderAssistant(db=db)
+    model_config = component.get_config("model", {})
+    assistant = PromptBuilderAssistant(db=db, model_config=model_config)
     greeting = assistant.get_initial_greeting()
     return {
         "message": greeting.message,
@@ -982,6 +983,7 @@ async def prompt_builder_chat(
     request: PromptBuilderChatRequest,
     db: Session = Depends(get_db),
     user_id: str = Depends(get_user_id),
+    component: "WebUIBackendComponent" = Depends(get_sac_component),
     _: None = Depends(check_prompts_enabled),
 ):
     """
@@ -995,8 +997,11 @@ async def prompt_builder_chat(
     5. Avoid command conflicts with existing prompts
     """
     try:
-        # Initialize the assistant with database session
-        assistant = PromptBuilderAssistant(db=db)
+        # Get model configuration from component
+        model_config = component.get_config("model", {})
+
+        # Initialize the assistant with database session and model config
+        assistant = PromptBuilderAssistant(db=db, model_config=model_config)
 
         # Process the message using real LLM with conflict checking
         response = await assistant.process_message(

src/solace_agent_mesh/gateway/http_sse/services/prompt_builder_assistant.py

@@ -123,15 +123,28 @@ class PromptBuilderAssistant:
 - Set ready_to_save to true when template is complete
 """
 
-    def __init__(self, db: Optional[Session] = None):
-        """Initialize the assistant."""
+    def __init__(self, db: Optional[Session] = None, model_config: Optional[Dict[str, Any]] = None):
+        """Initialize the assistant with model configuration from component config."""
         self.system_prompt = self.SYSTEM_PROMPT
         self.db = db
 
-        # Get LLM configuration from environment
-        self.model = os.getenv("LLM_SERVICE_GENERAL_MODEL_NAME", "gpt-4.1")
-        self.api_base = os.getenv("LLM_SERVICE_ENDPOINT")
-        self.api_key = os.getenv("LLM_SERVICE_API_KEY", "dummy")
+        # Get LLM configuration from provided config
+        if not model_config or not isinstance(model_config, dict):
+            raise ValueError("model_config is required and must be a dictionary")
+
+        # Try nested structure first (model.general)
+        general_model = model_config.get("general", {})
+        if isinstance(general_model, dict) and general_model.get("model"):
+            self.model = general_model.get("model")
+            self.api_base = general_model.get("api_base")
+            self.api_key = general_model.get("api_key", "dummy")
+        # Try direct structure (model.model)
+        elif model_config.get("model"):
+            self.model = model_config.get("model")
+            self.api_base = model_config.get("api_base")
+            self.api_key = model_config.get("api_key", "dummy")
+        else:
+            raise ValueError("model_config must contain either 'general.model' or 'model' key")
 
     def _get_existing_commands(self, user_id: str) -> List[str]:
         """Get list of existing command shortcuts to avoid conflicts."""

templates/webui.yaml

@@ -20,6 +20,8 @@ apps:
       namespace: ${NAMESPACE}
       session_secret_key: "${SESSION_SECRET_KEY}"
 
+      model:
+        general: *general_model
       artifact_service: *default_artifact_service
       session_service: __SESSION_SERVICE__
       gateway_id: ${WEBUI_GATEWAY_ID}