Update app.py

app.py

@@ -29,20 +29,16 @@ class MyAgent:
         )
 
     def __call__(self, question: str) -> str:
-        # Manually prepend system prompt in the messages for chat completion
-        messages = [
-            {"role": "system", "content": SYSTEM_PROMPT},
-            {"role": "user", "content": question}
-        ]
-        try:
-            # Assuming your model supports a method like `complete` or `run` that accepts messages
-            # If your agent.run expects just the question string, you must use model.complete yourself.
-            # So call model.complete(messages=messages) instead:
-            return self.model.complete(messages=messages)
-        except Exception as e:
-            import traceback
-            traceback.print_exc()
-            return f"AGENT ERROR: {e}"
+    messages = [
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {"role": "user", "content": question}
+    ]
+    try:
+        return self.model.run(messages)
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return f"AGENT ERROR: {e}"
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):