Update llm_only_agent.py

llm_only_agent.py

@@ -12,7 +12,7 @@ class LLMOnlyAgent:
         )
 
         # Code Agent
-        agent = CodeAgent(
+        self.agent = CodeAgent(
             model=model,
             tools=[FinalAnswerTool()],
             max_steps=2
@@ -21,6 +21,6 @@ class LLMOnlyAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = agent.run(question)
+        answer = self.agent.run(question)
         print(f"Agent returning answer: {answer}")
         return answer