andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 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 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)