Create agent.py

agent.py

@@ -0,0 +1,46 @@
+import json
+from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, tool
+
+#========Tools========#
+@tool
+def multiply(a: int, b: int) -> int:
+    return a * b
+@tool
+def add(a: int, b: int) -> int:
+    return a + b
+@tool
+def subtract(a: int, b: int) -> int:
+    return a - b
+@tool
+def divide(a: int, b: int) -> int:
+    if b == 0:
+        raise ValueError("Cannot divide by zero.")
+    return a / b
+@tool
+def modulus(a: int, b: int) -> int:
+    return a % b
+search_tool=DuckDuckGoSearchTool()
+
+#=======Agent========#
+model = InferenceClientModel()
+agent = CodeAgent(
+    tools=[
+    multiply,
+    add,
+    subtract,
+    divide,
+    modulus,
+    search_tool],
+    model = model,
+    add_base_tools=True,
+    planning_interval=3
+)
+with open("Q20.json", "r") as f:
+    tasks = json.load(f)
+
+for task in tasks:
+        task_id = task["task_id"]
+        question = task["question"]
+        file_name = task["file_name"]
+        answer = agent.run(question)
+        print(f"task_id: {task_id},submitted_answer: {answer}")