Update agent.py

agent.py

@@ -192,10 +192,13 @@ def wikipedia_search(query: str) -> str:
             results = data.get("query", {}).get("search", [])
             if not results:
                 return f"No Wikipedia results found for '{query}'."
-            return "\n\n---\n\n".join(
-                f"Title: {item['title']}\n"
-                f"Snippet: {item.get('snippet', '')}"
-                for item in results
+            MAX_CONTENT_LENGTH = 3000
+            print(result[:100])
+
+            return "\n---\n".join(
+                f"Title: {doc.metadata.get('title', 'Unknown')}\n"
+                f"Content: {doc.page_content[:MAX_CONTENT_LENGTH]}"
+                for doc in results
             )
 
         except requests.exceptions.RequestException as e:
@@ -219,29 +222,6 @@ def wikipedia_search(query: str) -> str:
         f"Wikipedia search temporarily failed for '{query}'. "
         "Please use another search source."
     )
-# @tool
-# def wikipedia_search(query: str) -> str:
-#     """
-#     Search wikipedia for a query and return results.
-#     Takes a string query as the keywords to search
-#     Args:
-#         query (str): Keywords you want to search.
-#     """
-#     try:
-#         search_results = WikipediaLoader(query=query, load_max_docs=3).load()
-
-#         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:
-#         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