Update agent.py

agent.py

@@ -64,7 +64,6 @@ class BasicAgent:
         model_id="gpt-4o",
         api_base="https://api.openai.com/v1",
         api_key=os.environ["OPENAI_API_KEY"],
-        system_prompt=prompt
         )
         
         self.agent = CodeAgent(
@@ -73,6 +72,8 @@ class BasicAgent:
         )
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.agent.run(question)
+        # Prepend the prompt to the question with a clear separator
+        full_input = prompt + "\n\nQuestion: " + question
+        fixed_answer = self.agent.run(full_input)
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer