first commit, smolagent basic tools

README.md

@@ -1,5 +1,5 @@
 ---
-title: Template Final Assignment
+title: Antonio Russo (Nioi) - Agents Course Final Assignment
 emoji: 🕵🏻‍♂️
 colorFrom: indigo
 colorTo: indigo

agent.py

@@ -0,0 +1,95 @@
+'''
+TODO tools
+- web_search
+
+Facultative
+- wiki_search
+- arxiv_search
+'''
+
+from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool
+
+
+@tool
+def add(a:int, b:int) -> int:
+  """
+  This tool returns the sum of two numbers.
+
+  Args:
+    a: first number
+    b: second number
+  """
+
+  return a+b
+
+
+@tool
+def subtract(a:int, b:int) -> int:
+  """
+  This tool returns the difference between two numbers.
+
+  Args:
+    a: first number
+    b: second number
+  """
+
+  return a-b
+
+
+@tool
+def multiply(a:int, b:int) -> int:
+  """
+  This tool multiplies two numbers.
+
+  Args:
+    a: first number
+    b: second number
+  """
+
+  return a*b
+
+
+@tool
+def divide(a:int, b:int) -> float:
+  """
+  This tool divides two numbers.
+
+  Args:
+    a: first number
+    b: second number
+  """
+
+  if b==0: raise ValueError('Cannot divide by zero')
+  return a/b
+
+
+@tool
+def modulus(a:int, b:int) -> int:
+  """
+  This tool returns the modulus of two numbers.
+
+  Args:
+    a: first number
+    b: second number
+  """
+
+  return a%b
+
+
+@tool
+def rounder(a:float, n:int):
+  """
+  This tool rounds a number to a certain number of decimals.
+
+  Args:
+    a: number to be rounded
+    n: number of decimals to use when rounding the number
+  """
+
+  return round(a,n)
+
+
+def get_agent() -> CodeAgent:
+  search_tool = DuckDuckGoSearchTool()
+
+  return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool], model=HfApiModel())

app.py

@@ -1,8 +1,8 @@
 import os
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
+import agent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -12,12 +12,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
+        self.agent = agent.get_agent()
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        answer = self.agent.run(question)
+        print(f"Agent returning answer: {answer}")
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

system_prompt.txt

@@ -0,0 +1,5 @@
+You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]. 
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.