Update app.py

app.py

@@ -34,12 +34,12 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         openaikey =  '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'
-        # self.system_message = (
-        #     "You are a question answering agent for scientific and encyclopedic questions. "
-        #     "Your goal is to provide concise, factually correct answers using only verified sources. "
-        #     "Always return a direct answer, no extra text or explanation. "
-        #     "If a question asks for a count, return only the number."
-        # )
+        self.system_message = (
+            "You are a question answering agent for scientific and encyclopedic questions. "
+            "Your goal is to provide concise, factually correct answers using only verified sources. "
+            "Always return a direct answer, no extra text or explanation. "
+            "If a question asks for a count, return only the number."
+        )
         
         model = InferenceClientModel(
             model_id="gpt-4o",
@@ -57,9 +57,9 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # Inject system prompt + question
-        # question_with_prompt = f"{self.system_message}\n\n{question.strip()}"
-        # answer = self.agent.run(question_with_prompt)
-        answer = self.agent.run(question)
+        question_with_prompt = f"{self.system_message}\n\n{question.strip()}"
+        answer = self.agent.run(question_with_prompt)
+        # answer = self.agent.run(question)
         print(f"Agent returning answer: {answer.strip()}")
         return answer.strip()