create basic_agent.py

basic_agent.py

@@ -0,0 +1,21 @@
+import os
+from smolagents import (
+    CodeAgent, 
+    LiteLLMModel
+)
+
+class BasicAgent:
+    def __init__(self):
+        model = LiteLLMModel(
+            model_id=os.getenv("MODEL_ID"), 
+            api_key=os.getenv("MODEL_API_KEY")
+        )
+        self.agent = CodeAgent(
+            tools=[], 
+            model=model,
+        )
+        
+    def __call__(self, question: str) -> str:
+        fixed_answer = self.agent.run(task=question)
+        print(f"Agent returning fixed answer: {fixed_answer}")
+        return fixed_answer