Update alz_companion/agent.py

alz_companion/agent.py

@@ -23,12 +23,16 @@ try:
 except Exception:  # pragma: no cover
     gTTS = None  # type: ignore
 
+
 from .prompts import (
     SYSTEM_TEMPLATE, ANSWER_TEMPLATE_CALM, ANSWER_TEMPLATE_ADQ,
-    SAFETY_GUARDRAILS, RISK_FOOTER, render_emotion_guidelines, CLASSIFICATION_PROMPT
+    SAFETY_GUARDRAILS, RISK_FOOTER, render_emotion_guidelines, CLASSIFICATION_PROMPT,
+    # Add the new templates to the import list
+    ROUTER_PROMPT, 
+    ANSWER_TEMPLATE_FACTUAL, 
+    ANSWER_TEMPLATE_GENERAL
 )
 
-
 # -----------------------------
 # Multimodal Processing Functions
 # -----------------------------