Upload folder using huggingface_hub

server/response_generator.py

@@ -85,12 +85,12 @@ def call_llm(prompt: str) -> str:
 # =========================
 # MAIN FUNCTION
 # =========================
-def generate_response(action_type: str, message: str, state, disorder_info) -> str:
+def generate_response(action_type: str, message: str, state, ) -> str:
     """
     Generate patient response using LLM
     """
 
-    prompt = build_prompt(action_type, message, state, disorder_info)
+    prompt = build_prompt(action_type, message, state, state.disorder)
 
     response = call_llm(prompt)