Fix CI issues related to missing dependency (#3096)

tests/embeddings/test_gemini_emeddings.py

@@ -1,4 +1,4 @@
-from unittest.mock import patch
+from unittest.mock import patch, ANY
 
 import pytest
 
@@ -8,8 +8,10 @@ from mem0.embeddings.gemini import GoogleGenAIEmbedding
 
 @pytest.fixture
 def mock_genai():
-    with patch("mem0.embeddings.gemini.genai.embed_content") as mock_genai:
-        yield mock_genai
+    with patch("mem0.embeddings.gemini.genai.Client") as mock_client_class:
+        mock_client = mock_client_class.return_value
+        mock_client.models.embed_content.return_value = None
+        yield mock_client.models.embed_content
 
 
 @pytest.fixture
@@ -18,7 +20,9 @@ def config():
 
 
 def test_embed_query(mock_genai, config):
-    mock_embedding_response = {"embedding": [0.1, 0.2, 0.3, 0.4]}
+    mock_embedding_response = type('Response', (), {
+        'embeddings': [type('Embedding', (), {'values': [0.1, 0.2, 0.3, 0.4]})]
+    })()
     mock_genai.return_value = mock_embedding_response
 
     embedder = GoogleGenAIEmbedding(config)
@@ -27,10 +31,11 @@ def test_embed_query(mock_genai, config):
     embedding = embedder.embed(text)
 
     assert embedding == [0.1, 0.2, 0.3, 0.4]
-    mock_genai.assert_called_once_with(model="test_model", content="Hello, world!", output_dimensionality=786)
+    mock_genai.assert_called_once_with(model="test_model", contents="Hello, world!", config=ANY)
+
 
 def test_embed_returns_empty_list_if_none(mock_genai, config):
-    mock_genai.return_value = None
+    mock_genai.return_value = type('Response', (), {'embeddings': [type('Embedding', (), {'values': []})]})()
 
     embedder = GoogleGenAIEmbedding(config)
     result = embedder.embed("test")
@@ -47,10 +52,10 @@ def test_embed_raises_on_error(mock_genai, config):
     with pytest.raises(RuntimeError, match="Embedding failed"):
         embedder.embed("some input")
 
+
 def test_config_initialization(config):
     embedder = GoogleGenAIEmbedding(config)
 
     assert embedder.config.api_key == "dummy_api_key"
     assert embedder.config.model == "test_model"
     assert embedder.config.embedding_dims == 786
-