Update llm_only_agent.py

llm_only_agent.py

@@ -1,16 +1,30 @@
 from smolagents import CodeAgent, HfApiModel, FinalAnswerTool
 
-# Basic inference model
-model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
-)
 
-# Code Agent
-agent = CodeAgent(
-    model=model,
-    tools=[FinalAnswerTool()],
-    max_steps=2
-)
+
+
+
+class BasicAgent:
+    def __init__(self):
+
+    # Basic inference model
+        model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+        custom_role_conversions=None,
+        )
+
+        # Code Agent
+        agent = CodeAgent(
+            model=model,
+            tools=[FinalAnswerTool()],
+            max_steps=2
+        )
+        
+        print("BasicAgent initialized.")
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        answer = agent.run(question)
+        print(f"Agent returning answer: {answer}")
+        return answer