Fix #2642: Add local file path handling to AddImageTool for Claude 3.7 Sonnet

Co-Authored-By: Joe Moura <joao@crewai.com>

tests/tools/agent_tools/test_add_image_tool.py

@@ -0,0 +1,57 @@
+import os
+import base64
+import pytest
+from unittest.mock import patch, MagicMock
+
+from crewai.tools.agent_tools.add_image_tool import AddImageTool
+
+
+class TestAddImageTool:
+    def setup_method(self):
+        self.tool = AddImageTool()
+        os.makedirs("tests/tools/agent_tools/test_files", exist_ok=True)
+    
+    def test_add_image_with_url(self):
+        result = self.tool._run(image_url="https://example.com/image.jpg")
+        assert result["role"] == "user"
+        assert len(result["content"]) == 2
+        assert result["content"][0]["type"] == "text"
+        assert result["content"][1]["type"] == "image_url"
+        assert result["content"][1]["image_url"]["url"] == "https://example.com/image.jpg"
+    
+    def test_add_image_with_local_file(self):
+        test_file_path = "tests/tools/agent_tools/test_files/test_image.jpg"
+        
+        with patch("builtins.open", MagicMock()), \
+             patch("base64.b64encode", return_value=b"test_encoded_content"), \
+             patch("os.path.exists", return_value=True):
+            
+            result = self.tool._run(image_url=test_file_path)
+            
+            assert result["role"] == "user"
+            assert len(result["content"]) == 2
+            assert result["content"][0]["type"] == "text"
+            assert result["content"][1]["type"] == "image_url"
+            assert result["content"][1]["image_url"]["url"].startswith("data:image/jpeg;base64,")
+    
+    def test_add_image_with_claude_3_7_model(self):
+        mock_llm = MagicMock()
+        mock_llm.model = "claude-3-7-sonnet-latest"
+        
+        with patch("os.path.exists", return_value=False):
+            result = self.tool._run(
+                image_url="https://example.com/image.jpg",
+                llm=mock_llm
+            )
+            
+            assert result["role"] == "user"
+            assert len(result["content"]) == 2
+            assert result["content"][0]["type"] == "text"
+            assert result["content"][1]["type"] == "image_url"
+            assert result["content"][1]["image_url"]["url"] == "https://example.com/image.jpg"
+    
+    def test_add_image_with_invalid_path(self):
+        with pytest.raises(ValueError):
+            with patch("os.path.exists", return_value=True), \
+                 patch("builtins.open", side_effect=FileNotFoundError()):
+                self.tool._run(image_url="/invalid/path/to/image.jpg")