KeenWoo commited on
Commit
4595c1f
·
verified ·
1 Parent(s): 8b38ffe

Update alz_companion/agent.py

Browse files
Files changed (1) hide show
  1. 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