Update app.py

app.py

@@ -37,14 +37,11 @@ class MyAgent:
         )
 
     def __call__(self, question: str) -> str:
-        messages = [
-            {"role": "system", "content": SYSTEM_PROMPT},
-            {"role": "user", "content": question}
-        ]
+        # Combine system prompt and user question as a single input
+        full_prompt = f"{SYSTEM_PROMPT}\n\nUser question: {question}"
         try:
-            response = self.model.chat(messages)
-            raw_output = response["content"]
-            return extract_final_answer(raw_output)
+            output = self.model.run(full_prompt)
+            return extract_final_answer(output)
         except Exception as e:
             import traceback
             traceback.print_exc()