another example

Dockerfile

@@ -23,5 +23,6 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Install the package
 RUN pip install -e .
 
-# Command to run when container starts
-CMD ["python"]
+COPY server.py /app/server.py
+
+CMD ["python", "/app/server.py"]

examples/docker_example.py

@@ -1,12 +1,39 @@
 from agents.search import DuckDuckGoSearchTool
 from agents.docker_alternative import DockerPythonInterpreter
 
+
+test = """
+from agents.tools import Tool
+
+class DummyTool(Tool):
+    name = "echo"
+    description = '''Perform a web search based on your query (think a Google search) then returns the top search results as a list of dict elements.
+    Each result has keys 'title', 'href' and 'body'.'''
+    inputs = {
+        "cmd": {"type": "string", "description": "The search query to perform."}
+    }
+    output_type = "any"
+
+    def forward(self, cmd: str) -> str:
+       return cmd 
+
+"""
+
 container = DockerPythonInterpreter()
 
-tools = [DuckDuckGoSearchTool]
+output = container.execute("x = 5")
+print(f"first output: {output}")
+output = container.execute("print(x)")
+print(f"second output: {output}")
 
-output = container.execute("res = web_search(query='whats the capital of Cambodia?'); print(res)", tools=tools)
+breakpoint()
 
+print("---------")
+
+output = container.execute(test)
+print(output)
+
+output = container.execute("res = DummyTool(cmd='echo this'); print(res)")
 print(output)
 
 container.stop()

examples/dummytool.py

@@ -0,0 +1,14 @@
+from agents.tools import Tool
+
+
+class DummyTool(Tool):
+    name = "echo"
+    description = """Perform a web search based on your query (think a Google search) then returns the top search results as a list of dict elements.
+    Each result has keys 'title', 'href' and 'body'."""
+    inputs = {
+        "cmd": {"type": "string", "description": "The search query to perform."}
+    }
+    output_type = "any"
+
+    def forward(self, cmd: str) -> str:
+       return cmd 

server.py

@@ -0,0 +1,46 @@
+import socket
+import sys
+import traceback
+import io
+
+exec_globals = {}
+exec_locals = {}
+
+def execute_code(code):
+    stdout = io.StringIO()
+    stderr = io.StringIO()
+    sys.stdout = stdout
+    sys.stderr = stderr
+
+    try:
+        exec(code, exec_globals, exec_locals)
+    except Exception as e:
+        traceback.print_exc(file=stderr)
+    
+    output = stdout.getvalue()
+    error = stderr.getvalue()
+
+    # Restore original stdout and stderr
+    sys.stdout = sys.__stdout__
+    sys.stderr = sys.__stderr__
+
+    return output + error
+
+def start_server(host='0.0.0.0', port=65432):
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        s.bind((host, port))
+        s.listen()
+        print(f"Server listening on {host}:{port}")
+        while True:
+            conn, addr = s.accept()
+            with conn:
+                print(f"Connected by {addr}")
+                data = conn.recv(1024)
+                if not data:
+                    break
+                code = data.decode('utf-8')
+                output = execute_code(code)
+                conn.sendall(output.encode('utf-8'))
+
+if __name__ == "__main__":
+    start_server() 

src/agents/docker_alternative.py

@@ -1,6 +1,7 @@
 import docker
 from typing import List, Optional 
 import warnings
+import socket
 
 from agents.tools import Tool
 
@@ -18,7 +19,7 @@ class DockerPythonInterpreter:
         try:            
             self.container = self.client.containers.run(
                 "pyrunner:latest",
-                "tail -f /dev/null",
+                ports={'65432/tcp': 65432},
                 detach=True,
                 remove=True,
            )
@@ -38,6 +39,7 @@ class DockerPythonInterpreter:
         Execute Python code in the container and return stdout and stderr
         """
 
+        if tools != None:        
             tool_instance = tools[0]()
 
             import_code = f"""
@@ -50,13 +52,13 @@ module = importlib.import_module(module_path)
 web_search = getattr(module, class_name)()
 """
 
-        full_code = import_code + "\n" + code
+            code = import_code + "\n" + code
 
         try:
-            exec_command = self.container.exec_run(
-                cmd=["python", "-c", full_code],
-            )
-            output = exec_command.output
+            # Connect to the server running inside the container
+            with socket.create_connection(('localhost', 65432)) as sock:
+                sock.sendall(code.encode('utf-8'))
+                output = sock.recv(4096)
                 return output.decode('utf-8')
 
         except Exception as e: