From 61cce93fd020fa4fce74cc80f64e07f2a777215d Mon Sep 17 00:00:00 2001
From: Eduardo Chiarotti <dudumelgaco@hotmail.com>
Date: Thu, 20 Jun 2024 21:41:12 -0300
Subject: [PATCH] feat: remove unused Makefile, update README and update code

---
 .../tools/code_interpreter_tool/Makefile      |  6 -----
 .../tools/code_interpreter_tool/README.md     | 27 +++++++++++++++++++
 .../code_interpreter_tool.py                  | 22 ++++++++++++++-
 3 files changed, 48 insertions(+), 7 deletions(-)
 delete mode 100644 src/crewai_tools/tools/code_interpreter_tool/Makefile

diff --git a/src/crewai_tools/tools/code_interpreter_tool/Makefile b/src/crewai_tools/tools/code_interpreter_tool/Makefile
deleted file mode 100644
index 5a514db0a..000000000
--- a/src/crewai_tools/tools/code_interpreter_tool/Makefile
+++ /dev/null
@@ -1,6 +0,0 @@
-# Makefile
-IMAGE_NAME=code-interpreter
-TAG=latest
-
-build:
-	docker build -t $(IMAGE_NAME):$(TAG) .
diff --git a/src/crewai_tools/tools/code_interpreter_tool/README.md b/src/crewai_tools/tools/code_interpreter_tool/README.md
index e69de29bb..672c86f21 100644
--- a/src/crewai_tools/tools/code_interpreter_tool/README.md
+++ b/src/crewai_tools/tools/code_interpreter_tool/README.md
@@ -0,0 +1,27 @@
+# CodeInterpreterTool
+
+## Description
+This tool is used to give the Agent the ability to run code (Python3) from the code generated by the Agent itself. The code is executed in a sandboxed environment, so it is safe to run any code.
+
+## Requirements
+
+- Docker
+
+## Installation
+Install the crewai_tools package
+```shell
+pip install 'crewai[tools]'
+```
+
+## Example
+
+Remember that when using this tool, the code must be generated by the Agent itself. The code must be a Python3 code. And it will take some time for the first time to run because it needs to build the Docker image.
+
+```python
+from crewai_tools import CodeInterpreterTool
+
+Agent(
+    ...
+    tools=[CodeInterpreterTool()],
+)
+```
diff --git a/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py b/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
index caafd44e3..06cb081f0 100644
--- a/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
+++ b/src/crewai_tools/tools/code_interpreter_tool/code_interpreter_tool.py
@@ -1,3 +1,4 @@
+import os
 from typing import Optional, Type
 
 import docker
@@ -30,9 +31,27 @@ class CodeInterpreterTool(BaseTool):
     args_schema: Type[BaseModel] = CodeInterpreterSchema
     code: Optional[str] = None
 
+    def _verify_docker_image(self) -> None:
+        """
+        Verify if the Docker image is available
+        """
+        image_tag = "code-interpreter:latest"
+
+        client = docker.from_env()
+        images = client.images.list()
+        all_tags = [tag for image in images for tag in image.tags]
+
+        if image_tag not in all_tags:
+            client.images.build(
+                path=os.path.dirname(os.path.abspath(__file__)),
+                tag=image_tag,
+                rm=True,
+            )
+
     def __init__(self, code: Optional[str] = None, **kwargs) -> None:
         super().__init__(**kwargs)
         if code is not None:
+            self._verify_docker_image()
             self.code = code
             self.description = "Interprets Python code in a Docker container. ALWAYS PRINT the final result and the output of the code"
             self.args_schema = FixedCodeInterpreterSchemaSchema
@@ -67,7 +86,8 @@ class CodeInterpreterTool(BaseTool):
         cmd_to_run = f'python3 -c "{code}"'
         exec_result = container.exec_run(cmd_to_run)
 
-        container.stop().remove()
+        container.stop()
+        container.remove()
 
         if exec_result.exit_code != 0:
             return f"Something went wrong while running the code: \n{exec_result.output.decode('utf-8')}"