Initial implementation

agent.py

@@ -0,0 +1,27 @@
+from smolagents import DuckDuckGoSearchTool, CodeAgent, HfApiModel
+import yaml
+
+class BasicAgent:
+    def __init__(self):
+        print("BasicAgent initialized.")
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        
+        with open("prompts.yaml", 'r') as stream:
+            prompt_templates = yaml.safe_load(stream)
+
+        model = HfApiModel()
+        search_tool = DuckDuckGoSearchTool()
+
+        agent = CodeAgent(
+            tools=[search_tool],
+            model=model,
+            add_base_tools = True,
+            verbose=True,
+            prompt_templates=prompt_templates
+        )
+
+        fixed_answer = agent.run(question)
+        print(f"Agent returning fixed answer: {fixed_answer}")
+        return fixed_answer

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from agent import BasicAgent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -10,14 +11,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        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
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -91,6 +84,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
+    return
     # 4. Prepare Submission 
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."

prompts.yaml

@@ -0,0 +1,2 @@
+"system_prompt": |-
+  You are a general AI assistant. 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.

requirements.txt

@@ -1,2 +1,4 @@
 gradio
-requests
+requests
+smolagents
+yaml