fix: add ShellTool

nobu007 · May 5, 2024 · 35a7bc5 · 35a7bc5
1 parent 20063d3
commit 35a7bc5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/codeinterpreterapi/tools/tools.py b/src/codeinterpreterapi/tools/tools.py
@@ -1,4 +1,4 @@
-from langchain_community.tools.shell.tool import ShellInput
+from langchain_community.tools.shell.tool import ShellTool
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.tools import BaseTool, StructuredTool
 
@@ -39,9 +39,8 @@ def get_python(additional_tools: list[BaseTool], run_handler_func, arun_handler_
 
     @staticmethod
     def get_shell(additional_tools: list[BaseTool]) -> list[BaseTool]:
-        # TODO: use ShellInput
-        # tools = [ShellInput()]
-        return additional_tools
+        tools = [ShellTool()]
+        return additional_tools + tools
 
     @staticmethod
     def get_web_search(additional_tools: list[BaseTool]) -> list[BaseTool]: