Spaces:
Sleeping
Sleeping
Update alz_companion/agent.py
Browse files- alz_companion/agent.py +3 -0
alz_companion/agent.py
CHANGED
|
@@ -819,6 +819,9 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
|
|
| 819 |
else: template = ANSWER_TEMPLATE_ADQ
|
| 820 |
|
| 821 |
emotions_context = render_emotion_guidelines(kwargs.get("emotion_tag"))
|
|
|
|
|
|
|
|
|
|
| 822 |
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"))
|
| 823 |
print("[DEBUG] Caregiving Scenario PROMPT")
|
| 824 |
# end
|
|
|
|
| 819 |
else: template = ANSWER_TEMPLATE_ADQ
|
| 820 |
|
| 821 |
emotions_context = render_emotion_guidelines(kwargs.get("emotion_tag"))
|
| 822 |
+
# NEXT evolution
|
| 823 |
+
# if emotion in ["confusion", "sadness", "anxiety", "orientation_check"]:
|
| 824 |
+
# template = ANSWER_TEMPLATE_CAL
|
| 825 |
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"))
|
| 826 |
print("[DEBUG] Caregiving Scenario PROMPT")
|
| 827 |
# end
|