bug fix and minor changes

Webserch_tool.py

@@ -1,8 +1,8 @@
-from langchain_community.tools import DuckDuckGoSearchRun
+# from langchain_community.tools import DuckDuckGoSearchRun
 
-search_tool = DuckDuckGoSearchRun()
-results = search_tool.invoke("Who's the current President of France?")
-print(results)
+# search_tool = DuckDuckGoSearchRun()
+# results = search_tool.invoke("Who's the current President of France?")
+# print(results)
 
 from langchain.tools import Tool
 import random
@@ -11,7 +11,7 @@ import os
 from dotenv import load_dotenv
 load_dotenv()
 weather_API_key = os.getenv("OPENWEATHER_KEY")
-print(f'weather_API_key: {weather_API_key}')
+#print(f'weather_API_key: {weather_API_key}')
 def get_weather_info(location: str) -> str:
     """Fetches weather information for a given location."""
     try:

other_tools.py

@@ -34,15 +34,30 @@ def square_root(a: float) -> float | complex:
     return cmath.sqrt(a)
 
 from langchain_community.document_loaders import WikipediaLoader
-@tool(description="Search Wikipedia for a query and return maximum 2 results.")
-def wiki_search(query: str) -> str:
-    search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
-    formatted_search_docs = "\n\n---\n\n".join(
-        [
-            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>{doc.page_content}</Document>'
-            for doc in search_docs
-        ])
-    return {"wiki_results": formatted_search_docs}
+
+@tool(description="Search Wikipedia for a query and return up to 2 results.")
+def wiki_search(query: str) -> dict:
+    print(f"Searching Wikipedia for: {query}")
+    
+    try:
+        # Load up to 2 documents from Wikipedia
+        search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+
+        # Format the output for each document
+        formatted_search_docs = "\n\n---\n\n".join(
+            [
+                f'<Document source="{doc.metadata.get("source", "")}" page="{doc.metadata.get("page", "")}">\n{doc.page_content}\n</Document>'
+                for doc in search_docs
+            ]
+        )
+
+        print(f"Here are the outputs:\n{formatted_search_docs}")
+        return {"wiki_results": formatted_search_docs}
+
+    except Exception as e:
+        print(f"Error occurred: {e}")
+        return {"wiki_results": f"None found for {query}, use the results from the other tools or similar questions obtained earlier."}
+
 
 from langchain_community.document_loaders import ArxivLoader
 @tool(description="Search Arxiv for a query and return maximum 3 result.")