andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 07:42:40 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

added missing import

Browse Source
This commit is contained in:
Tom Mahler
2025-01-14 21:06:42 +02:00
parent 71f3ed9ef9
commit 334beda181
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
View File

@@ -3,6 +3,7 @@ import os
from typing import List, Optional, Type from typing import List, Optional, Type
from docker import from_env as docker_from_env from docker import from_env as docker_from_env
from docker import DockerClient
from docker.models.containers import Container from docker.models.containers import Container
from docker.errors import ImageNotFound, NotFound from docker.errors import ImageNotFound, NotFound
from crewai.tools import BaseTool 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. 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: try:
client.images.get(self.default_image_tag) client.images.get(self.default_image_tag)