mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-24 07:38:14 +00:00
changed == None to is None
This commit is contained in:
@@ -31,7 +31,7 @@ class CodeInterpreterTool(BaseTool):
|
|||||||
default_image_tag: str = "code-interpreter:latest"
|
default_image_tag: str = "code-interpreter:latest"
|
||||||
code: Optional[str] = None
|
code: Optional[str] = None
|
||||||
user_dockerfile_path: Optional[str] = None
|
user_dockerfile_path: Optional[str] = None
|
||||||
user_docker_base_url: Optional[str] = None
|
user_docker_base_url: Optional[str] = None
|
||||||
unsafe_mode: bool = False
|
unsafe_mode: bool = False
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -44,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 DockerClient(base_url=self.user_docker_base_url)
|
client = docker_from_env() if self.user_docker_base_url is 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)
|
||||||
@@ -136,4 +136,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)}"
|
||||||
Reference in New Issue
Block a user