andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 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 crewai.tools import BaseTool
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.errors import ImageNotFound, NotFound from docker.errors import ImageNotFound, NotFound
from docker.models.containers import Container from docker.models.containers import Container
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
@@ -46,7 +48,7 @@ class CodeInterpreterTool(BaseTool):
client = ( client = (
docker_from_env() docker_from_env()
if self.user_docker_base_url == None 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: try:
@@ -137,4 +139,4 @@ class CodeInterpreterTool(BaseTool):
exec(code, {}, exec_locals) exec(code, {}, exec_locals)
return exec_locals.get("result", "No result variable found.") return exec_locals.get("result", "No result variable found.")
except Exception as e: except Exception as e:
return f"An error occurred: {str(e)}" return f"An error occurred: {str(e)}"