Fix CI

libs/kotaemon/kotaemon/embeddings/voyageai.py

@@ -2,15 +2,11 @@
 """
 
 import importlib
-from typing import TYPE_CHECKING
 
 from kotaemon.base import Document, DocumentWithEmbedding, Param
 
 from .base import BaseEmbeddings
 
-if TYPE_CHECKING:
-    from voyageai import Client
-    from voyageai import AsyncClient
 
 vo = None
 
@@ -36,8 +32,7 @@ def _format_output(texts: list[str], embeddings: list[list]):
 
 
 class VoyageAIEmbeddings(BaseEmbeddings):
-    """Voyage AI provides best-in-class embedding models and rerankers.
-    """
+    """Voyage AI provides best-in-class embedding models and rerankers."""
 
     api_key: str = Param(None, help="Voyage API key", required=False)
     model_name: str = Param(
@@ -68,4 +63,3 @@ async def ainvoke(
         texts = [t.content for t in self.prepare_input(text)]
         embeddings = await self._aclient.embed(texts, model=self.model_name).embeddings
         return _format_output(texts, embeddings)
-

libs/kotaemon/tests/conftest.py

@@ -72,7 +72,7 @@ def if_llama_cpp_not_installed():
 
 def if_voyageai_not_installed():
     try:
-        import voyageai
+        import voyageai  # noqa: F401
     except ImportError:
         return True
     else:

libs/kotaemon/tests/test_embedding_models.py

@@ -160,6 +160,14 @@ def test_fastembed_embeddings():
 
 
 @skip_when_voyageai_not_installed
+@patch(
+    "voyageai.Client.embed",
+    side_effect=lambda *args, **kwargs: [[1.0, 2.1, 3.2]],
+)
+@patch(
+    "voyageai.AsyncClient.embed",
+    side_effect=lambda *args, **kwargs: [[1.0, 2.1, 3.2]],
+)
 def test_voyageai_embeddings():
     model = VoyageAIEmbeddings()
     output = model("Hello, world!")