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

Merge pull request #185 from MahlerTom/fix/code-interpreter

Browse Source 
Fix/code interpreter
This commit is contained in:
João Moura
2025-01-19 02:11:10 -03:00
committed by GitHub
parent ef611a9dcd bee8fda006
commit b03d74abe5
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4,6 +4,8 @@ from typing import List, Optional, Type
from crewai.tools import BaseTool
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 docker.models.containers import Container
from pydantic import BaseModel, Field
@@ -46,7 +48,7 @@ class CodeInterpreterTool(BaseTool):
client = (
docker_from_env()
if self.user_docker_base_url == None
else docker.DockerClient(base_url=self.user_docker_base_url)
else DockerClient(base_url=self.user_docker_base_url)
)
try:
@@ -137,4 +139,4 @@ class CodeInterpreterTool(BaseTool):
exec(code, {}, exec_locals)
return exec_locals.get("result", "No result variable found.")
except Exception as e:
return f"An error occurred: {str(e)}"
return f"An error occurred: {str(e)}"