Add execution_image parameter to Agent for custom Docker images

- Add execution_image field to Agent class for specifying custom Docker images
- Modify get_code_execution_tools() to pass custom image to CodeInterpreterTool
- Add comprehensive tests for the new functionality
- Update documentation with usage examples
- Fixes #2933: Allow customizing CrewAI code executor Docker image

Co-Authored-By: João <joao@crewai.com>

docs/tools/ai-ml/codeinterpretertool.mdx

@@ -92,6 +92,22 @@ programmer_agent = Agent(
 )
 ```
 
+You can also specify a custom Docker image for code execution:
+
+```python Code
+from crewai import Agent
+
+# Create an agent with code execution enabled and custom Docker image
+programmer_agent = Agent(
+    role="Python Programmer",
+    goal="Write and execute Python code to solve problems",
+    backstory="An expert Python programmer who can write efficient code to solve complex problems.",
+    allow_code_execution=True,  # This automatically adds the CodeInterpreterTool
+    execution_image="python:3.11-slim",  # Custom Docker image with specific Python version
+    verbose=True,
+)
+```
+
 ### Enabling `unsafe_mode`
 
 ```python Code

src/crewai/agent.py

@@ -127,6 +127,10 @@ class Agent(BaseAgent):
         default="safe",
         description="Mode for code execution: 'safe' (using Docker) or 'unsafe' (direct execution).",
     )
+    execution_image: Optional[str] = Field(
+        default=None,
+        description="Custom Docker image to use for code execution. If not specified, uses the default image.",
+    )
     reasoning: bool = Field(
         default=False,
         description="Whether the agent should reflect and create a plan before executing a task.",
@@ -564,7 +568,11 @@ class Agent(BaseAgent):
 
             # Set the unsafe_mode based on the code_execution_mode attribute
             unsafe_mode = self.code_execution_mode == "unsafe"
-            return [CodeInterpreterTool(unsafe_mode=unsafe_mode)]
+            
+            if self.execution_image:
+                return [CodeInterpreterTool(unsafe_mode=unsafe_mode, default_image_tag=self.execution_image)]
+            else:
+                return [CodeInterpreterTool(unsafe_mode=unsafe_mode)]
         except ModuleNotFoundError:
             self._logger.log(
                 "info", "Coding tools not available. Install crewai_tools. "

tests/test_agent_custom_execution_image.py

@@ -0,0 +1,82 @@
+import pytest
+from unittest.mock import patch, MagicMock
+from crewai import Agent
+
+
+def test_agent_with_custom_execution_image():
+    """Test that Agent can be created with custom execution image."""
+    agent = Agent(
+        role="Test Agent",
+        goal="Test goal",
+        backstory="Test backstory",
+        allow_code_execution=True,
+        execution_image="my-custom-image:latest"
+    )
+    
+    assert agent.execution_image == "my-custom-image:latest"
+
+
+def test_agent_without_custom_execution_image():
+    """Test that Agent works without custom execution image (default behavior)."""
+    agent = Agent(
+        role="Test Agent", 
+        goal="Test goal",
+        backstory="Test backstory",
+        allow_code_execution=True
+    )
+    
+    assert agent.execution_image is None
+
+
+@patch('crewai.agent.CodeInterpreterTool')
+def test_get_code_execution_tools_with_custom_image(mock_code_interpreter):
+    """Test that get_code_execution_tools passes custom image to CodeInterpreterTool."""
+    agent = Agent(
+        role="Test Agent",
+        goal="Test goal", 
+        backstory="Test backstory",
+        allow_code_execution=True,
+        execution_image="my-custom-image:latest"
+    )
+    
+    tools = agent.get_code_execution_tools()
+    
+    mock_code_interpreter.assert_called_once_with(
+        unsafe_mode=False,
+        default_image_tag="my-custom-image:latest"
+    )
+
+
+@patch('crewai.agent.CodeInterpreterTool')
+def test_get_code_execution_tools_without_custom_image(mock_code_interpreter):
+    """Test that get_code_execution_tools works without custom image."""
+    agent = Agent(
+        role="Test Agent",
+        goal="Test goal",
+        backstory="Test backstory", 
+        allow_code_execution=True
+    )
+    
+    tools = agent.get_code_execution_tools()
+    
+    mock_code_interpreter.assert_called_once_with(unsafe_mode=False)
+
+
+@patch('crewai.agent.CodeInterpreterTool')
+def test_get_code_execution_tools_with_unsafe_mode_and_custom_image(mock_code_interpreter):
+    """Test that both unsafe_mode and custom image work together."""
+    agent = Agent(
+        role="Test Agent",
+        goal="Test goal",
+        backstory="Test backstory",
+        allow_code_execution=True,
+        code_execution_mode="unsafe",
+        execution_image="my-custom-image:latest"
+    )
+    
+    tools = agent.get_code_execution_tools()
+    
+    mock_code_interpreter.assert_called_once_with(
+        unsafe_mode=True,
+        default_image_tag="my-custom-image:latest"
+    )

tests/test_custom_execution_image_integration.py

@@ -0,0 +1,36 @@
+import pytest
+from unittest.mock import patch, MagicMock
+from crewai import Agent, Task, Crew
+
+
+@patch('crewai_tools.CodeInterpreterTool')
+def test_crew_with_custom_execution_image_integration(mock_code_interpreter_class):
+    """Integration test for custom execution image in a Crew workflow."""
+    mock_tool_instance = MagicMock()
+    mock_code_interpreter_class.return_value = mock_tool_instance
+    
+    agent = Agent(
+        role="Python Developer",
+        goal="Execute Python code",
+        backstory="Expert in Python programming",
+        allow_code_execution=True,
+        execution_image="python:3.11-slim"
+    )
+    
+    task = Task(
+        description="Calculate 2 + 2",
+        expected_output="The result of 2 + 2",
+        agent=agent
+    )
+    
+    crew = Crew(
+        agents=[agent],
+        tasks=[task]
+    )
+    
+    tools = crew._prepare_tools(task, agent)
+    
+    mock_code_interpreter_class.assert_called_with(
+        unsafe_mode=False,
+        default_image_tag="python:3.11-slim"
+    )