From d146e4a961812a5bae25279f2438fab918350eef Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Tue, 27 Aug 2024 09:01:09 -0300 Subject: [PATCH] feat: change the anme of user dockerfile path --- src/crewai_tools/tools/code_interpreter_tool/README.md | 2 +- .../tools/code_interpreter_tool/code_interpreter_tool.py | 8 +++----- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/src/crewai_tools/tools/code_interpreter_tool/README.md b/src/crewai_tools/tools/code_interpreter_tool/README.md index 035f910a4..bc73df7a4 100644 --- a/src/crewai_tools/tools/code_interpreter_tool/README.md +++ b/src/crewai_tools/tools/code_interpreter_tool/README.md @@ -35,6 +35,6 @@ from crewai_tools import CodeInterpreterTool Agent( ... - tools=[CodeInterpreterTool(user_docker_file_path="")], + tools=[CodeInterpreterTool(user_dockerfile_path="")], ) ``` diff --git a/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py b/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py index 2bde9a54b..6f427e440 100644 --- a/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py +++ b/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py @@ -27,7 +27,7 @@ class CodeInterpreterTool(BaseTool): args_schema: Type[BaseModel] = CodeInterpreterSchema default_image_tag: str = "code-interpreter:latest" code: Optional[str] = None - user_docker_file_path: Optional[str] = None + user_dockerfile_path: Optional[str] = None @staticmethod def _get_installed_package_path(): @@ -44,10 +44,8 @@ class CodeInterpreterTool(BaseTool): client.images.get(self.default_image_tag) except docker.errors.ImageNotFound: - if self.user_docker_file_path and os.path.exists( - self.user_docker_file_path - ): - dockerfile_path = self.user_docker_file_path + if self.user_dockerfile_path and os.path.exists(self.user_dockerfile_path): + dockerfile_path = self.user_dockerfile_path else: package_path = self._get_installed_package_path() dockerfile_path = os.path.join(