add some tools

GaiaAgent.py

@@ -0,0 +1,66 @@
+from smolagents import InferenceClientModel, CodeAgent, GoogleSearchTool, VisitWebpageTool, WikipediaSearchTool
+
+from tool.SupadataTool import SupadataTool
+from tool.UnknownTextTool import UnknownTextTool, BasicMathTool
+
+
+class GaiaAgent:
+    def __init__(self):
+        manager_model = InferenceClientModel(
+            model_id='deepseek-ai/DeepSeek-R1',
+            provider='nebius',
+            max_tokens=8096,
+        )
+        agent_model = InferenceClientModel(
+            "Qwen/Qwen3-Coder-30B-A3B-Instruct", provider="together", max_tokens=8096
+        )
+
+        self.web_agent = CodeAgent(
+            model=agent_model,
+            tools=[
+                GoogleSearchTool(provider="serper"),
+                VisitWebpageTool()
+            ],
+            name="web_agent",
+            description="Browses the web to find information",
+            verbosity_level=0,
+            max_steps=10,
+        )
+        self.wikipedia_agent = CodeAgent(
+            model=agent_model,
+            tools=[WikipediaSearchTool()],
+            name="wikipedia_agent",
+            description="Searches Wikipedia and returns a summary or full text of the given topic, along with the page URL. If you need to get any encyclopedia information, use this tool.",
+            max_steps=5,
+        )
+        self.text_agent = CodeAgent(
+            model=agent_model,
+            tools=[UnknownTextTool(), BasicMathTool()],
+            name="text_agent",
+            description="This agent is perfect for solving specific tasks: * decoding unknown text (if you have unknown text, use this tool); * math and algorithmic problems add, subtract, multiply, divide, etc.",
+        )
+        self.youtube_transcript_agent = CodeAgent(
+            model=agent_model,
+            tools=[SupadataTool()],
+            name="youtube_transcript_agent",
+            description="This agent is perfect for generating transcripts of YouTube videos. If you need to know what the video is about, use this tool.",
+        )
+        self.manager_agent = CodeAgent(
+            tools=[],
+            model=manager_model,
+            managed_agents=[self.web_agent, self.wikipedia_agent, self.text_agent, self.youtube_transcript_agent],
+            additional_authorized_imports=[
+            ],
+            planning_interval=5,
+            verbosity_level=2,
+            max_steps=15,
+        )
+
+        print("GaiaAgent initialized.")
+
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        answer = self.manager_agent.run(question)
+        print(f"Agent returning fixed answer: {answer}")
+        return answer

app.py

@@ -4,24 +4,15 @@ import requests
 import inspect
 import pandas as pd
 
+from GaiaAgent import GaiaAgent
+
 # (Keep Constants as is)
 # --- Constants ---
 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):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
+    Fetches all questions, runs the GaiaAgent on them, submits all answers,
     and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
@@ -35,12 +26,12 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         return "Please Login to Hugging Face with the button.", None
 
     api_url = DEFAULT_API_URL
-    questions_url = f"{api_url}/questions"
+    questions_url = f"{api_url}/random-question"
     submit_url = f"{api_url}/submit"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = GaiaAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

requirements.txt

@@ -1,2 +1,4 @@
 gradio
-requests
+requests
+supadata
+wikipedia-api

tool/SupadataTool.py

@@ -0,0 +1,36 @@
+import os
+
+from smolagents import Tool
+from supadata import Supadata
+
+
+class SupadataTool(Tool):
+    name = "supadata_tool"
+    description = "Get the YouTube video URL and generate transcription describing the video content. If you need to know what the video is about, use this tool."
+    inputs = {
+        "url": {
+            "type": "string",
+            "description": "Full YouTube video URL",
+        }
+    }
+    output_type = "string"
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.supadata = Supadata(api_key=os.getenv("SUPADATA_API_KEY"))
+
+    def forward(self, url):
+        video_id = self._extract_video_id(url)
+        transcript = self.supadata.youtube.transcript(video_id=video_id, lang="en")
+        return transcript.content
+
+    @staticmethod
+    def _extract_video_id(url):
+        if not url.startswith('https://'):
+            return url
+
+        if 'youtube.com/watch?v=' in url:
+            return url.split('watch?v=')[1].split('&')[0]
+        elif 'youtu.be/' in url:
+            return url.split('youtu.be/')[1].split('?')[0]
+        return url

tool/UnknownTextTool.py

@@ -0,0 +1,57 @@
+from smolagents import Tool
+
+
+class UnknownTextTool(Tool):
+    name = "unknown_text_tool"
+    description = "Returns the reversed version of the text. If you receive some unknown text, that can't be recognized and analyzed, then you need to use this tool to make it clear."
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "Unknown text to be decoded"
+        }
+    }
+    output_type = "string"
+
+    def forward(self, query):
+        try:
+            print(f"Reversed text: {query[::-1]}")
+            return query[::-1]
+        except Exception as e:
+            return f"Can't reverse text: {e}"
+
+
+class BasicMathTool(Tool):
+    name = "basic_math_tool"
+    description = "Performs basic arithmetic operations (addition, subtraction, multiplication, division)"
+    inputs = {
+        "operation": {
+            "type": "string",
+            "description": "Mathematical operation to perform (+, -, *, /)"
+        },
+        "x": {
+            "type": "float",
+            "description": "First number"
+        },
+        "y": {
+            "type": "float",
+            "description": "Second number"
+        }
+    }
+    output_type = "float"
+
+    def forward(self, operation, x, y):
+        try:
+            if operation == "+":
+                return x + y
+            elif operation == "-":
+                return x - y
+            elif operation == "*":
+                return x * y
+            elif operation == "/":
+                if y == 0:
+                    raise ValueError("Division by zero")
+                return x / y
+            else:
+                raise ValueError(f"Unknown operation: {operation}")
+        except Exception as e:
+            return f"Math error: {e}"