Update app.py

app.py

@@ -24,6 +24,7 @@ from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.tools.ddg_search import DuckDuckGoSearchRun
 from langchain_community.utilities.wikipedia import WikipediaAPIWrapper
+from langchain_community.tools import WikipediaQueryRun
 from langchain_experimental.tools import PythonREPLTool # Use with caution
 # Custom tools will be defined below (or implicitly used)
 
@@ -354,8 +355,11 @@ class GaiaAgent:
         else:
             logging.warning("TAVILY_API_KEY not found, using DuckDuckGoSearchRun.")
             self.tools.append(DuckDuckGoSearchRun())
-        self.tools.append(WikipediaAPIWrapper())
-        logging.info("Using Wikipedia Search Tool.")
+        # Initialize the API wrapper needed by the tool
+        api_wrapper = WikipediaAPIWrapper(top_k_results=3, doc_content_chars_max=4000, lang='en', load_all_available_meta=False) # Re-create the wrapper explicitly
+        # Add the WikipediaQueryRun tool, passing the wrapper
+        self.tools.append(WikipediaQueryRun(api_wrapper=api_wrapper))
+        logging.info("Using Wikipedia Query Run Tool.") # Update log message slightly
         try:
             self.tools.append(PythonREPLTool())
             logging.info("Using Python REPL Tool.")