mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-03 14:22:13 +08:00
3b4e993531
2. use tavily from`langchain-tavily` rather than the older one from `langchain-community` Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import logging
|
|
import os
|
|
from typing import Annotated, Optional
|
|
|
|
from langchain_core.tools import tool
|
|
from langchain_experimental.utilities import PythonREPL
|
|
|
|
from .decorators import log_io
|
|
|
|
|
|
def _is_python_repl_enabled() -> bool:
|
|
"""Check if Python REPL tool is enabled from configuration."""
|
|
# Check environment variable first
|
|
env_enabled = os.getenv("ENABLE_PYTHON_REPL", "false").lower()
|
|
if env_enabled in ("true", "1", "yes", "on"):
|
|
return True
|
|
return False
|
|
|
|
|
|
# Initialize REPL and logger
|
|
repl: Optional[PythonREPL] = PythonREPL() if _is_python_repl_enabled() else None
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@tool
|
|
@log_io
|
|
def python_repl_tool(
|
|
code: Annotated[
|
|
str, "The python code to execute to do further analysis or calculation."
|
|
],
|
|
):
|
|
"""Use this to execute python code and do data analysis or calculation. If you want to see the output of a value,
|
|
you should print it out with `print(...)`. This is visible to the user."""
|
|
|
|
# Check if the tool is enabled
|
|
if not _is_python_repl_enabled():
|
|
error_msg = "Python REPL tool is disabled. Please enable it in environment configuration."
|
|
logger.warning(error_msg)
|
|
return f"Tool disabled: {error_msg}"
|
|
|
|
if not isinstance(code, str):
|
|
error_msg = f"Invalid input: code must be a string, got {type(code)}"
|
|
logger.error(error_msg)
|
|
return f"Error executing code:\n```python\n{code}\n```\nError: {error_msg}"
|
|
|
|
logger.info("Executing Python code")
|
|
try:
|
|
result = repl.run(code)
|
|
# Check if the result is an error message by looking for typical error patterns
|
|
if isinstance(result, str) and ("Error" in result or "Exception" in result):
|
|
logger.error(result)
|
|
return f"Error executing code:\n```python\n{code}\n```\nError: {result}"
|
|
logger.info("Code execution successful")
|
|
except BaseException as e:
|
|
error_msg = repr(e)
|
|
logger.error(error_msg)
|
|
return f"Error executing code:\n```python\n{code}\n```\nError: {error_msg}"
|
|
|
|
result_str = f"Successfully executed:\n```python\n{code}\n```\nStdout: {result}"
|
|
return result_str
|