definito wikipedia tool alternativo per accedere al contenuto delle pagine wiki

agent.py

@@ -12,7 +12,7 @@ from langgraph.graph import StateGraph, END
 from langchain_google_genai import ChatGoogleGenerativeAI
 
 # Local imports
-from tools import get_search_tool, get_wikipedia_tool, execute_python_code_from_file
+from tools import get_search_tool, get_wikipedia_tool, wikipedia_search, execute_python_code_from_file
 
 # Nota: per i test in locale si usa il .env
 #       su HuggingFace invece si usano le variabili definite in Settings/"Variables and secrets"
@@ -35,9 +35,9 @@ chat = ChatGoogleGenerativeAI(
 
 # Imposta i tool
 search_tool = get_search_tool()
-wikipedia_tool = get_wikipedia_tool()
+#wikipedia_tool = get_wikipedia_tool()
 
-tools = [search_tool, wikipedia_tool, execute_python_code_from_file]
+tools = [search_tool, wikipedia_search, execute_python_code_from_file]
 
 # Bind tools to the model
 chat_with_tools = chat.bind_tools(tools)
@@ -62,6 +62,11 @@ def call_tool(state: AgentState):
     for tool_call in state["messages"][-1].tool_calls:
         # Get the tool by name
         tool_result = tools_by_name[tool_call["name"]].invoke(tool_call["args"])
+
+        # print(f"\n--- DEBUG: Tool Raw Output (Length: {len(tool_result)} chars) ---")
+        # print(tool_result)
+        # print("------------------------------------------------------------------\n")
+
         outputs.append(
             ToolMessage(
                 content=tool_result,

tools.py

@@ -1,12 +1,10 @@
 import os, sys
 from dotenv import load_dotenv
 
-# DuckDuckGo
 from langchain_community.tools import DuckDuckGoSearchRun
-# Wikipedia tool components
 from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_community.tools import WikipediaQueryRun
-
+import wikipedia
 from langchain_core.tools import tool
 from langchain.tools import Tool
 
@@ -39,6 +37,35 @@ def get_wikipedia_tool():
     return wikipedia_tool
 
 
+@tool
+def wikipedia_search(query: str) -> str:
+    """
+    Search Wikipedia and return the full content of the most relevant article.
+    """
+    try:
+        results = wikipedia.search(query)
+        if not results:
+            return f"No results found for '{query}'."
+
+        page = wikipedia.page(results[0])
+        content = page.content
+
+        # Truncate content if it's too long
+        if len(content) > WIKIPEDIA_DOC_CONTENT_CHARS_MAX:
+            content = content[:WIKIPEDIA_DOC_CONTENT_CHARS_MAX] + "..."
+
+        return content
+
+    except wikipedia.exceptions.DisambiguationError as e:
+        return f"Ambiguous query. Possible options: {', '.join(e.options[:5])}..."
+    except wikipedia.exceptions.PageError:
+        return f"Page not found for '{query}'."
+    except Exception as e:
+        return f"Error occurred: {str(e)}"
+
+
+
+
 @tool
 def execute_python_code_from_file(file_path: str) -> str:
     """