Update agent.py

agent.py

@@ -149,6 +149,18 @@ def build_gemini_llm():
 #     except Exception as e:
 #         return f"error downloading or reading file: {str(e)}"
 
+loader = WikipediaLoader(
+    query="Mercedes Sosa discography",
+    load_max_docs=3
+)
+
+docs = loader.load()
+
+print(docs)
+
+print(WikipediaLoader(query="Python programming language", load_max_docs=1).load())
+
+
 @tool
 def wikipedia_search(query: str) -> str:
     """
@@ -162,14 +174,16 @@ def wikipedia_search(query: str) -> str:
 
         if not search_results:
             return f"No Wikipedia results found for {query}. Consider another query or try a web search."
+        print("wiki result:")
+        print(search_results)
         return "---\n".join(
             f"Title: {doc.metadata.get('title', 'Unknown')}\n"
             f"Content: {doc.page_content}"
             for doc in search_results
         )
     except Exception as e:
-        print(f"Wikipedia search failed for {query}: {e}")
-        return f"Wikipedia search failed for {query} due to {e}."
+        traceback.print_exc()
+        return f"Wikipedia search failed: {type(e).__name__}: {e}"
 
 # fix wikipedia engine builds invalid URL for region="wt-wt"(default) issue from duckducksearchrun
 from langchain_community.utilities import DuckDuckGoSearchAPIWrapper
@@ -188,6 +202,8 @@ def search_web(query: str) -> str:
         result = search_ddgs.invoke(query)
         if not result:
             return f"No web results found for: {query}"
+        print("ddgs result:")
+        print(result)
         return result
     except Exception as e:
         print(f"DuckDuckGo search failed for {query}: {e}")