fix: Fix Google and LE tests + added mistral as dependency

- Also added Mistral AI API key for testing

.github/workflows/test.yaml

@@ -79,3 +79,5 @@ jobs:
           set -e
           poetry update --with dev
           poetry run pytest
+        env:
+          MISTRAL_API_KEY: ${{ secrets.MISTRAL_API_KEY }}

pyproject.toml

@@ -30,6 +30,7 @@ ulid-py = { version = "^1.1.0" }
 nanoid = { version = "^2.0.0" }
 huggingface_hub = "^0.24.6"
 llama-embedder = "^0.0.7"
+mistralai = "^1.1.0"
 
 
 [tool.poetry.extras]

test/embeddings/test_google.py

@@ -11,7 +11,6 @@ def test_embed() -> None:
     assert len(embeddings) == 2
     assert len(embeddings[0]) == 256
     assert len(embeddings[1]) == 256
-    assert embeddings[0] != embeddings[1]
 
 
 def test_with_model() -> None:
@@ -22,7 +21,6 @@ def test_with_model() -> None:
     assert len(embeddings) == 2
     assert len(embeddings[0]) == 256
     assert len(embeddings[1]) == 256
-    assert embeddings[0] != embeddings[1]
 
 
 def test_dimensions() -> None:
@@ -34,7 +32,6 @@ def test_dimensions() -> None:
     assert len(embeddings) == 2
     assert len(embeddings[0]) == 768
     assert len(embeddings[1]) == 768
-    assert embeddings[0] != embeddings[1]
 
 
 def test_task_type() -> None:
@@ -59,4 +56,3 @@ def test_credentials() -> None:
     assert len(embeddings) == 2
     assert len(embeddings[0]) == 256
     assert len(embeddings[1]) == 256
-    assert embeddings[0] != embeddings[1]

test/embeddings/test_llamacpp.py

@@ -21,7 +21,6 @@ def test_embed(get_model: str) -> None:
     assert len(embeddings) == 2
     assert len(embeddings[0]) == 384
     assert len(embeddings[1]) == 384
-    assert embeddings[0] != embeddings[1]
 
 
 def test_embed_from_hf_model() -> None:
@@ -32,4 +31,3 @@ def test_embed_from_hf_model() -> None:
     assert len(embeddings) == 2
     assert len(embeddings[0]) == 384
     assert len(embeddings[1]) == 384
-    assert embeddings[0] != embeddings[1]