Added duckduckgo search.

app.py

@@ -1,8 +1,10 @@
 import os
 import gradio as gr
 import requests
+import json
 import pandas as pd
 from openai import OpenAI
+from bs4 import BeautifulSoup
 
 
 # (Keep Constants as is)
@@ -16,48 +18,162 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-
         self.client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=os.getenv("OR_TOKEN"))
 
+    def duckduckgo_search(self, query: str, num_results: int = 3) -> list:
+        """
+        Perform a search using DuckDuckGo and return the results.
+
+        Args:
+            query: The search query string
+            num_results: Maximum number of results to return (default: 5)
+
+        Returns:
+            List of dictionaries containing search results with title, url, and snippet
+        """
+        print(f"Performing DuckDuckGo search for: {query}")
+
+        try:
+            headers = {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+            }
+
+            # Format the query for the URL
+            formatted_query = query.replace(' ', '+')
+            url = f"https://html.duckduckgo.com/html/?q={formatted_query}"
+
+            # Send the request
+            response = requests.get(url, headers=headers, timeout=10)
+            response.raise_for_status()
+
+            # Parse the HTML response
+            soup = BeautifulSoup(response.text, 'html.parser')
+
+            # Extract search results
+            results = []
+            for result in soup.select('.result'):
+                title_elem = result.select_one('.result__title')
+                link_elem = result.select_one('.result__url')
+                snippet_elem = result.select_one('.result__snippet')
+
+                if title_elem and link_elem:
+                    title = title_elem.get_text(strip=True)
+                    url = link_elem.get('href') if link_elem.get('href') else link_elem.get_text(strip=True)
+                    snippet = snippet_elem.get_text(strip=True) if snippet_elem else ""
+
+                    results.append({
+                        "title": title,
+                        "url": url,
+                        "snippet": snippet
+                    })
+
+                    if len(results) >= num_results:
+                        break
+
+            print(f"Found {len(results)} results for query: {query}")
+            return results
+        except Exception as e:
+            print(f"Error during DuckDuckGo search: {e}")
+            return []
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print(f"Agent received question: {question}...")
 
         try:
-            # Generate response
-            print("Using Inference API for generation...")
-            completion = self.client.chat.completions.create(
-                extra_headers={
-                    "HTTP-Referer": "<YOUR_SITE_URL>",  # Optional. Site URL for rankings on openrouter.ai.
-                    "X-Title": "<YOUR_SITE_NAME>",  # Optional. Site title for rankings on openrouter.ai.
+            tools = [
+                {
+                    "type": "function",
+                    "function": {
+                            "name": "duckduckgo_search",
+                            "description": "Search the web using DuckDuckGo and return relevant results",
+                            "parameters": {
+                                "type": "object",
+                                "properties": {
+                                    "query": {
+                                        "type": "string",
+                                        "description": "The search query string"
+                                    },
+                                    "num_results": {
+                                        "type": "integer",
+                                        "description": "Maximum number of results to return",
+                                        "default": 5
+                                    }
+                                },
+                                "required": ["query"]
+                            }
+                    }
+                }
+            ]
+
+            tools_mapping = {
+                "duckduckgo_search": self.duckduckgo_search
+            }
+
+            messages = [
+                {
+                    "role": "system",
+                    "content": "You are a general AI assistant. I will ask you a question. Read the question carefully. Break down the question into multiple questions and use the tools available to you to answer the question. Do not report your thoughts, only give 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.",
                 },
-                extra_body={},
-                # model="meta-llama/llama-4-scout:free",
-                model="meta-llama/llama-4-maverick:free",
-                messages=[
-                    {
-                        "role": "system",
-                        "content": "You are a general AI assistant. I will ask you a question. Do not report your thoughts, only give 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.",
-                    },
-                    {
-                        "role": "user",
-                        "content": [
+                {
+                    "role": "user",
+                    "content": [
                             {
                                 "type": "text",
                                 "text": question
                             },
-                            # {
-                            #  "type": "image_url",
-                            #  "image_url": {
-                            #    "url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
-                            #  }
-                            # }
-                        ]
+                        # {
+                        #  "type": "image_url",
+                        #  "image_url": {
+                        #    "url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
+                        #  }
+                        # }
+                    ]
+                }
+            ]
+
+            # Execute once
+            for _ in range(3):
+                # Generate response
+                print("Using Inference API for generation...")
+                completion = self.client.chat.completions.create(
+                    extra_headers={
+                        "HTTP-Referer": "<YOUR_SITE_URL>",  # Optional. Site URL for rankings on openrouter.ai.
+                        "X-Title": "<YOUR_SITE_NAME>",  # Optional. Site title for rankings on openrouter.ai.
+                    },
+                    extra_body={},
+                    # model="meta-llama/llama-4-scout:free",
+                    model="meta-llama/llama-4-maverick:free",
+                    # model="google/gemini-2.0-flash-exp:free",
+                    # model="mistralai/mistral-small-3.1-24b-instruct:free",
+                    #tools=tools,
+                    messages=messages
+                )
+
+                messages.append(completion.choices[0].message)
+                print(f"Message after completion: {completion.choices[0].message}")
+
+                if completion.choices[0].message.tool_calls is None:
+                    answer = completion.choices[0].message.content
+                    print(f"Agent generated response: {answer}")
+                    return answer
+
+                for tool_call in completion.choices[0].message.tool_calls:
+                    """
+                    In this case we only provided one tool, so we know what function to call.
+                    When providing multiple tools, you can inspect `tool_call.function.name`
+                    to figure out what function you need to call locally.
+                    """
+                    tool_name = tool_call.function.name
+                    tool_args = json.loads(tool_call.function.arguments)
+                    tool_response = tools_mapping[tool_name](**tool_args)
+                    message = {
+                        "role": "tool",
+                        "tool_call_id": tool_call.id,
+                        "name": tool_name,
+                        "content": json.dumps(tool_response),
                     }
-                ]
-            )
-            answer = completion.choices[0].message.content
-            print(f"Agent generated response (first 50 chars): {answer[:50]}...")
-            return answer
+                    messages.append(message)
+                    print(f"Message after tools call: {message}")
         except Exception as e:
             print(f"Error generating response: {e}")
             fallback_answer = "I apologize, but I encountered an error when trying to answer your question."

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 requests
 huggingface_hub
-openai
+openai
+bs4