Fix type error and test issues

Co-Authored-By: Joe Moura <joao@crewai.com>
2026-05-01 23:32:39 +00:00 · 2025-04-25 21:03:36 +00:00
parent 50059c7120
commit 5623e2c851
2 changed files with 7 additions and 1 deletions
--- a/tests/utilities/test_embedding_functions.py
+++ b/tests/utilities/test_embedding_functions.py
@@ -45,8 +45,13 @@ class TestFixedGoogleVertexEmbeddingFunction:
    def test_embedding_call(self, embedding_function):
        function, mock_post = embedding_function
        
+        mock_response = MagicMock()
+        mock_response.json.return_value = {"predictions": [[0.1, 0.2, 0.3]]}
+        mock_post.return_value = mock_response
+        
        embeddings = function(["test text"])
        
        mock_post.assert_called_once()
        
        assert isinstance(embeddings, list)
+        assert len(embeddings) > 0