trying new model and new search tool

agents.py

@@ -7,15 +7,18 @@ from smolagents import (
     WikipediaSearchTool,
     DuckDuckGoSearchTool,
     SpeechToTextTool,
+    PythonInterpreterTool,
 )
 import os
+import time
 from typing import Optional
+from tools import MySearchTool
 
 
 gemini_api_key = os.environ.get("GEMINI_API_KEY", None)
 open_router_api_key = os.environ.get("OPENROUTER_API_KEY", None)
 gemini_model = LiteLLMModel(
-    model_id="gemini/gemini-2.0-flash-lite",
+    model_id="gemini/gemini-2.0-flash",
     api_key=gemini_api_key,
 )
 # alternative model
@@ -29,6 +32,9 @@ model = OpenAIServerModel(
 search_tool = DuckDuckGoSearchTool()
 speech_tool = SpeechToTextTool()
 wiki_tool = WikipediaSearchTool()
+python_tool = PythonInterpreterTool()
+visit_webpage_tool = VisitWebpageTool()
+my_search_tool = MySearchTool()
 
 
 class SkynetMultiAgent:
@@ -58,7 +64,9 @@ class SkynetMultiAgent:
             print(f"Processing question with file: {file_path}")
         else:
             eval_question = question
-        fixed_answer = self.agent.run(eval_question, max_steps=10)
+
+        time.sleep(2)  # 2-second delay
+        fixed_answer = self.agent.run(eval_question, max_steps=5)
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
@@ -66,11 +74,15 @@ class SkynetMultiAgent:
 class SkynetSingleAgent:
     def __init__(self):
         self.agent = CodeAgent(
-            tools=[speech_tool, wiki_tool],
+            tools=[
+                speech_tool,
+                wiki_tool,
+                my_search_tool,
+                visit_webpage_tool,
+                python_tool,
+            ],
             model=gemini_model,
             name="skynet_agent",
-            add_base_tools=True,
-            planning_interval=5,
             max_steps=5,
             description="Agent to answer questions and run code",
             additional_authorized_imports=["time", "numpy", "pandas"],
@@ -84,6 +96,7 @@ class SkynetSingleAgent:
             print(f"Processing question with file: {file_path}")
         else:
             eval_question = question
+        time.sleep(2)  # 2-second delay
         fixed_answer = self.agent.run(eval_question)
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer

app.py

@@ -31,7 +31,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        multi_agent = SkynetMultiAgent()
+        # multi_agent = SkynetMultiAgent()
         agent = SkynetSingleAgent()
         print(f"Agent instantiated successfully: {agent}")
     except Exception as e:

tools.py

@@ -1,6 +1,11 @@
 from smolagents import Tool
 from langchain_community.retrievers import BM25Retriever
 from langchain.docstore.document import Document
+import requests
+import os
+import time
+from typing import List, Dict, Any, Optional
+from .utils import get_env_var, logger
 
 
 class DownloadFilesTool(Tool):
@@ -58,3 +63,75 @@ class PythonCodeAnalyzerTool(Tool):
         # Placeholder for actual code analysis logic
         # In a real implementation, you would use a library or service to analyze the code
         return f"Analyzed code:\n{code}\n\nNo issues found."
+
+
+class MySearchTool(Tool):
+    name = "Search tool to avoid rate limit"
+    description = "It receives a question and returns a list of search results from various sources. It uses DuckDuckGo as the primary source"
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "The question or topic to search in internet.",
+        },
+    }
+    output_type = "string"
+
+    def __init__(self):
+        # Primary: Free alternatives
+        self.duckduckgo_enabled = True
+
+    def forward(self, query: str, max_results: int = 5) -> List[Dict[str, Any]]:
+        """
+        Free search using DuckDuckGo Instant Answer API
+
+        Args:
+            query: Search query
+            max_results: Maximum number of results
+
+        Returns:
+            List of search results
+        """
+        try:
+            # DuckDuckGo Instant Answer API (free)
+            url = "https://api.duckduckgo.com/"
+            params = {
+                "q": query,
+                "format": "json",
+                "no_html": "1",
+                "skip_disambig": "1",
+            }
+
+            response = requests.get(url, params=params, timeout=10)
+            response.raise_for_status()
+            time.sleep(2)  # 2-second delay
+            data = response.json()
+            results = []
+
+            # Process abstract
+            if data.get("Abstract"):
+                results.append(
+                    {
+                        "title": data.get("Heading", "DuckDuckGo Result"),
+                        "url": data.get("AbstractURL", ""),
+                        "content": data.get("Abstract", ""),
+                        "source": "DuckDuckGo",
+                    }
+                )
+
+            # Process related topics
+            for topic in data.get("RelatedTopics", [])[: max_results - len(results)]:
+                if isinstance(topic, dict) and "Text" in topic:
+                    results.append(
+                        {
+                            "title": topic.get("Text", "")[:100],
+                            "url": topic.get("FirstURL", ""),
+                            "content": topic.get("Text", ""),
+                            "source": "DuckDuckGo",
+                        }
+                    )
+
+            return results[:max_results]
+
+        except Exception as e:
+            logger.error(f"DuckDuckGo search failed: {str(e)}")
+            return []