Create agent.py

agent.py

@@ -0,0 +1,31 @@
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+
+# Initialize a model (use your preferred model_id)
+model = HfApiModel(model_id="meta-llama/Llama-2-7b-chat-hf")  # Or another model
+
+# Define custom tools if needed (optional)
+from smolagents.tool import tool
+
+@tool
+def add(a: int, b: int) -> int:
+    """Add two numbers."""
+    return a + b
+
+@tool
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers."""
+    return a * b
+
+# Create the agent with web search and custom tools
+agent = CodeAgent(
+    tools=[DuckDuckGoSearchTool(), add, multiply],
+    model=model,
+)
+
+def answer_question(question: str) -> str:
+    """Run the agent on a question and return the answer."""
+    result = agent.run(question)
+    # Format as per system_prompt.txt
+    if "FINAL ANSWER:" not in result:
+        result = f"FINAL ANSWER: {result}"
+    return result