Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import (CodeAgent, DuckDuckGoSearchTool, HfApiModel)
+from smolagents import (CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool)
 from tavily import TavilyClient
 
 # (Keep Constants as is)
@@ -14,7 +14,29 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
 #search_tool = DuckDuckGoSearchTool()
-search_tool = TavilyClient(api_key="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")
+
+
+@tool
+def web_search_tool(query: str) -> str:
+    """
+    This tool search for question as web serach and return back the answer
+
+    Args:
+        query: A search term for finding catering services.
+
+    Ex: 
+    response =  web_search_tool("where is Taj Mahal located ?")
+    
+    """
+
+    # do a seach query
+    client = TavilyClient(api_key="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")
+    
+    response = client.qna_search(query=query)
+ 
+    return response
+
+
 
 agent = CodeAgent(tools=[search_tool], model=HfApiModel())