Update agent.py

agent.py

@@ -1,2 +1,19 @@
-def run_agent(llm, task):
-    return
+from smolagents import (
+    CodeAgent, 
+    OpenAIServerModel, 
+    DuckDuckGoSearchTool, 
+    WikipediaSearchTool
+)
+
+def run_agent(model_id, question):
+    agent = CodeAgent(
+            model=OpenAIServerModel(model_id=model_id),
+            tools=[DuckDuckGoSearchTool(), 
+                   WikipediaSearchTool()],
+            add_base_tools=True,
+            max_steps=5,
+            verbosity_level=2,
+            additional_authorized_imports=["pandas", "numpy"]
+        )
+    
+    return agent.run(question)