Update alz_companion/agent.py

alz_companion/agent.py

@@ -32,6 +32,7 @@ from .prompts import (
 
     NLU_ROUTER_PROMPT, SPECIALIST_CLASSIFIER_PROMPT,
     ANSWER_TEMPLATE_CALM, 
+    ANSWER_TEMPLATE_PATIENT, ANSWER_TEMPLATE_PATIENT_MODERATE, ANSWER_TEMPLATE_PATIENT_ADVANCED,
     ANSWER_TEMPLATE_ADQ, ANSWER_TEMPLATE_ADQ_MODERATE, ANSWER_TEMPLATE_ADQ_ADVANCED,
 
     ANSWER_TEMPLATE_FACTUAL, ANSWER_TEMPLATE_FACTUAL_MULTI, ANSWER_TEMPLATE_SUMMARIZE,
@@ -808,9 +809,15 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
             user_prompt = template.format(personal_context=personal_context, general_context=general_context, question=query, language=language, patient_name=p_name, caregiver_name=c_name, context=personal_context, role=role)
             print("[DEBUG] Personal Route Factual / Sum / Multi PROMPT") 
         else: # caregiving_scenario
+            # NEXT evolution
+            # if settings.get("role") == "patient":
+            # use template = ANSWER_TEMPLATE_PATIENT
+            # else template = ANSWER_TEMPLATE_ADQ
+            
             if disease_stage == "Advanced Stage": template = ANSWER_TEMPLATE_ADQ_ADVANCED
             elif disease_stage == "Moderate Stage": template = ANSWER_TEMPLATE_ADQ_MODERATE
             else: template = ANSWER_TEMPLATE_ADQ
+                
             emotions_context = render_emotion_guidelines(kwargs.get("emotion_tag"))
             user_prompt = template.format(general_context=general_context, personal_context=personal_context, question=query, scenario_tag=kwargs.get("scenario_tag"), emotions_context=emotions_context, role=role, language=language, patient_name=p_name, caregiver_name=c_name, emotion_tag=kwargs.get("emotion_tag"))
             print("[DEBUG] Caregiving Scenario PROMPT")