Minor bugfix

Cinnamon · Oct 17, 2024 · 192edba · 192edba
1 parent f663c9f
commit 192edba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/kotaemon/kotaemon/embeddings/voyageai.py b/libs/kotaemon/kotaemon/embeddings/voyageai.py
@@ -59,13 +59,13 @@ def invoke(
         self, text: str | list[str] | Document | list[Document], *args, **kwargs
     ) -> list[DocumentWithEmbedding]:
         texts = [t.content for t in self.prepare_input(text)]
-        embeddings = self._client.embed()
+        embeddings = self._client.embed(texts, model=self.model_name).embeddings
         return _format_output(texts, embeddings)
 
     async def ainvoke(
         self, text: str | list[str] | Document | list[Document], *args, **kwargs
     ) -> list[DocumentWithEmbedding]:
         texts = [t.content for t in self.prepare_input(text)]
-        embeddings = await self._aclient.embed(texts)
+        embeddings = await self._aclient.embed(texts, model=self.model_name).embeddings
         return _format_output(texts, embeddings)