From 334beda1810201fcadab8dde8c50c04d9f968549 Mon Sep 17 00:00:00 2001 From: Tom Mahler Date: Tue, 14 Jan 2025 21:06:42 +0200 Subject: [PATCH] added missing import --- .../tools/code_interpreter_tool/code_interpreter_tool.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 fd0d39932..8924d52c0 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 @@ -3,6 +3,7 @@ import os from typing import List, Optional, Type from docker import from_env as docker_from_env +from docker import DockerClient from docker.models.containers import Container from docker.errors import ImageNotFound, NotFound from crewai.tools import BaseTool @@ -43,7 +44,7 @@ class CodeInterpreterTool(BaseTool): Verify if the Docker image is available. Optionally use a user-provided Dockerfile. """ - client = docker_from_env() if self.user_docker_base_url == None else docker.DockerClient(base_url=self.user_docker_base_url) + client = docker_from_env() if self.user_docker_base_url == None else DockerClient(base_url=self.user_docker_base_url) try: client.images.get(self.default_image_tag)