chat worked

backend/src/plugins/llm_analysis/service.py

@@ -20,6 +20,7 @@ from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_excep
 from .models import LLMProviderType
 from ...core.logger import belief_scope, logger
 from ...core.config_models import Environment
+from ...services.llm_prompt_templates import DEFAULT_LLM_PROMPTS, render_prompt
 
 # [DEF:ScreenshotService:Class]
 # @PURPOSE: Handles capturing screenshots of Superset dashboards.
@@ -548,7 +549,12 @@ class LLMClient:
     # @PRE: screenshot_path exists, logs is a list of strings.
     # @POST: Returns a structured analysis dictionary (status, summary, issues).
     # @SIDE_EFFECT: Reads screenshot file and calls external LLM API.
-    async def analyze_dashboard(self, screenshot_path: str, logs: List[str]) -> Dict[str, Any]:
+    async def analyze_dashboard(
+        self,
+        screenshot_path: str,
+        logs: List[str],
+        prompt_template: str = DEFAULT_LLM_PROMPTS["dashboard_validation_prompt"],
+    ) -> Dict[str, Any]:
         with belief_scope("analyze_dashboard"):
             # Optimize image to reduce token count (US1 / T023)
             # Gemini/Gemma models have limits on input tokens, and large images contribute significantly.
@@ -582,25 +588,7 @@ class LLMClient:
                     base_64_image = base64.b64encode(image_file.read()).decode('utf-8')
 
             log_text = "\n".join(logs)
-            prompt = f"""
-            Analyze the attached dashboard screenshot and the following execution logs for health and visual issues.
-            
-            Logs:
-            {log_text}
-            
-            Provide the analysis in JSON format with the following structure:
-            {{
-                "status": "PASS" | "WARN" | "FAIL",
-                "summary": "Short summary of findings",
-                "issues": [
-                    {{
-                        "severity": "WARN" | "FAIL",
-                        "message": "Description of the issue",
-                        "location": "Optional location info (e.g. chart name)"
-                    }}
-                ]
-            }}
-            """
+            prompt = render_prompt(prompt_template, {"logs": log_text})
             
             messages = [
                 {