Update agent.py

agent.py

@@ -16,6 +16,9 @@ from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingF
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.tools.retriever import create_retriever_tool
 from typing import TypedDict, Annotated, List
+from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun, ArxivQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper, ArxivAPIWrapper
+from langchain.tools import Tool
 
 # Load environment variables from .env
 load_dotenv()
@@ -311,47 +314,53 @@ def weather_tool(location: str) -> str:
     """
     return get_weather(location, search_tool)
 
+
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_community.tools.ddg_search import DuckDuckGoSearchRun
+from langchain_community.tools.wikipedia.tool import WikipediaQueryRun
+from langchain_community.utilities.wikipedia import WikipediaAPIWrapper
+from langchain_community.tools.arxiv.tool import ArxivQueryRun
+from langchain_community.utilities.arxiv import ArxivAPIWrapper
+from langchain.tools import tool
+
+# 1. Tavily Web Search Tool (already in correct format)
 @tool
 def web_search(query: str) -> str:
-    """Search the web for a given query and return the summary.
-    Args:
-        query (str): The search query.
-    """
-    
+    """Search the web for a given query and return the summary."""
     search_tool = TavilySearchResults()
     result = search_tool.run(query)
     return result[0]['content']
 
+# 2. DuckDuckGo Search Tool
 @tool
-def wiki_search(query : str) -> str:
-    """Search Wikipedia for a given query and return the summary.
-    Args:
-        query (str): The search query.
-    """
-
-    search_docs = WikipediaLoader(query=query, load_max_docs=1).load()
-    formatted_search_docs = "\n\n----\n\n".join(
-        [
-            f'<Document Source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}">\n{doc.page_content}\n</Document>'
-            for doc in search_docs
-        ]
-    )
-    return formatted_search_docs
-
-# @tool
-# def recommendation_tool(weather_condition: str) -> str:
-#     """
-#     Provides recommendations based on weather conditions.
+def duckduckgo_search(query: str) -> str:
+    """Search the web using DuckDuckGo for a given query and return the result."""
+    search_tool = DuckDuckGoSearchRun(verbose=False)
+    return search_tool.run(query)
 
-#     Args:
-#         weather_condition (str): The weather condition.
+# 3. Wikipedia Search Tool
+@tool
+def wikipedia_search(query: str) -> str:
+    """Search Wikipedia for a given query and return the top 3 results."""
+    wrapper = WikipediaAPIWrapper(top_k_results=3)
+    wikipedia = WikipediaQueryRun(api_wrapper=wrapper, verbose=False)
+    return wikipedia.run(query)
 
-#     Returns:
-#         str: The recommendations.
-#     """
-#     return get_recommendation(weather_condition, recommendation_chain)
+# 4. Arxiv Search Tool
+@tool
+def arxiv_search(query: str) -> str:
+    """Search arXiv for academic papers based on a query and return the top 3 results."""
+    wrapper = ArxivAPIWrapper(
+        top_k_results=3,
+        ARXIV_MAX_QUERY_LENGTH=300,
+        load_max_docs=3,
+        load_all_available_meta=False,
+        doc_content_chars_max=40000
+    )
+    arxiv = ArxivQueryRun(api_wrapper=wrapper, verbose=False)
+    return arxiv.run(query)
 
-tools = [weather_tool, wiki_search, web_search,
+tools = [arxiv_search, duckduckgo_search, web_search,wikipedia_search,
             add, subtract, multiply, divide, square, cube, power, factorial, mean, standard_deviation]
 
 # === LLM with Tools ===
@@ -362,9 +371,9 @@ llm = ChatGroq(
     groq_api_key=os.getenv("GROQ_API_KEY")
 )
 
-tools = [weather_tool, wiki_search, web_search,
-         add, subtract, multiply, divide, square, cube,
-         power, factorial, mean, standard_deviation]
+# tools = [weather_tool, wiki_search, web_search,
+#          add, subtract, multiply, divide, square, cube,
+#          power, factorial, mean, standard_deviation, arxiv_tool,wikisearch_tool, search_tool ]
 
 llm_with_tools = llm.bind_tools(tools)