changed == None to is None

This commit is contained in:
Tom Mahler
2025-01-14 21:07:08 +02:00
parent 334beda181
commit 1bd87f514e

View File

@@ -31,7 +31,7 @@ class CodeInterpreterTool(BaseTool):
default_image_tag: str = "code-interpreter:latest"
code: 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
@staticmethod
@@ -44,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 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:
client.images.get(self.default_image_tag)
@@ -136,4 +136,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)}"