aggiunto tool per eseguire codice python

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
+from tools import get_search_tool, get_wikipedia_tool, 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,7 +35,7 @@ chat = ChatGoogleGenerativeAI(
 search_tool = get_search_tool()
 wikipedia_tool = get_wikipedia_tool()
 
-tools = [search_tool, wikipedia_tool]
+tools = [search_tool, wikipedia_tool, execute_python_code_from_file]
 
 # Bind tools to the model
 chat_with_tools = chat.bind_tools(tools)

tools.py

@@ -1,11 +1,16 @@
+import os, sys
+
 # DuckDuckGo
 from langchain_community.tools import DuckDuckGoSearchRun
 # Wikipedia tool components
 from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_community.tools import WikipediaQueryRun
 
+from langchain_core.tools import tool
 from langchain.tools import Tool
 
+# per gestire esecuzione di codice python
+import subprocess
 
 def get_search_tool():
 
@@ -13,6 +18,7 @@ def get_search_tool():
 
     return search_tool
 
+
 def get_wikipedia_tool():
 
     # creates an instance of the Wikipedia API wrapper. top_k_results=1 means it will only fetch the top result from Wikipedia
@@ -21,4 +27,48 @@ def get_wikipedia_tool():
     # converts the WikipediaAPIWrapper into a LangChain tool.
     wikipedia_tool = WikipediaQueryRun(api_wrapper=wikipedia_api_wrapper)
 
-    return wikipedia_tool
+    return wikipedia_tool
+
+
+@tool
+def execute_python_code_from_file(file_path: str) -> str:
+    """
+    Reads a Python file from the given path, executes its code, and returns the combined stdout and stderr.
+    WARNING: Executing arbitrary code from files is a significant security risk.
+    Only use this tool with trusted code in a controlled environment.
+    """
+    if not os.path.exists(file_path):
+        return f"Error: File not found at '{file_path}'."
+    
+    if not file_path.endswith(".py"):
+        return f"Error: Provided file '{file_path}' is not a Python (.py) file."
+
+    try:
+        # Use subprocess to run the Python file in a new process.
+        # This provides some isolation compared to 'exec()' but is still dangerous for untrusted code.
+        result = subprocess.run(
+            [sys.executable, file_path], # sys.executable ensures it uses the current Python interpreter
+            capture_output=True,         # Capture stdout and stderr
+            text=True,                   # Capture output as text (strings)
+            check=False                  # Do not raise an exception for non-zero exit codes (handle errors manually)
+        )
+
+        stdout_output = result.stdout.strip()
+        stderr_output = result.stderr.strip()
+        
+        output_lines = []
+        if stdout_output:
+            output_lines.append(f"STDOUT:\n{stdout_output}")
+        if stderr_output:
+            output_lines.append(f"STDERR:\n{stderr_output}")
+        
+        if result.returncode != 0:
+            output_lines.append(f"Process exited with code {result.returncode}. This usually indicates an error.")
+        
+        if not output_lines:
+            return "Execution completed with no output."
+
+        return "\n".join(output_lines)
+
+    except Exception as e:
+        return f"An unexpected error occurred during code execution: {e}"