Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import DuckDuckGoSearchTool, HfApiModel, PythonInterpreterTool, VisitWebpageTool, CodeAgent
+from smolagents import DuckDuckGoSearchTool,GoogleSearchTool, HfApiModel, PythonInterpreterTool, VisitWebpageTool, CodeAgent
 import hashlib
 import json
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, TransformersEngine
@@ -21,6 +21,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 cache = {}
 
 web_search = DuckDuckGoSearchTool()
+google_search = GoogleSearchTool(provider="serper")
 python_interpreter = PythonInterpreterTool(authorized_imports = [
     # standard library
     'os',                      # For file path manipulation, checking existence, deletion
@@ -56,6 +57,13 @@ yt_transcript_fetcher = YoutubeTranscriptFetcher()
 math_model_querer = MathModelQuerer()
 code_model_querer = CodeModelQuerer()
 
+# batch of tools fromm Langchain. Credits DataDiva88
+lc_ddg_search = Tool.from_langchain(load_tools(["ddg-search"])[0])
+lc_wikipedia = Tool.from_langchain(load_tools(["wikipedia"])[0])
+lc_arxiv = Tool.from_langchain(load_tools(["arxiv"])[0])
+lc_pubmed = Tool.from_langchain(load_tools(["pubmed"])[0])
+lc_stackechange = Tool.from_langchain(load_tools(["stackexchange"])[0])
+
 
 def load_cached_answer(question_id: str) -> str:
     if question_id in cache.keys():
@@ -106,6 +114,10 @@ model = HfApiModel()
 #     max_new_tokens=256)
 
 # model = HfApiModel()
+lc_ddg_search = Tool.from_langchain(load_tools(["ddg-search"])[0])
+lc_wikipedia = Tool.from_langchain(load_tools(["wikipedia"])[0])
+lc_arxiv = Tool.from_langchain(load_tools(["arxiv"])[0])
+lc_pubmed = Tool.from_langchain(load_tools(["pubmed"])[0])
 
 
 class BasicAgent:
@@ -113,7 +125,7 @@ class BasicAgent:
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
             model=model,
-            tools=[web_search, python_interpreter, visit_webpage_tool, wiki_tool, code_model_querer, math_model_querer],
+            tools=[google_search,web_search, python_interpreter, visit_webpage_tool, wiki_tool, code_model_querer, math_model_querer,lc_wikipedia,lc_arxiv,lc_pubmed,lc_stackechange],
             max_steps=5,
             verbosity_level=1,
             grammar=None,