feat: add VoyageAI's rerank and embeddings models (#733) #none

* Introducing VoyageAI's rerank and embeddings models

* fix: comfort CI

* fix: update test case

---------

Co-authored-by: fzowl <zoltan@voyageai.com>
This commit is contained in:
Tuan Anh Nguyen Dang (Tadashi_Cin)
2025-04-15 15:54:23 +07:00
committed by GitHub
parent c33bedca9e
commit 5132288386
11 changed files with 194 additions and 5 deletions

View File

@@ -62,6 +62,7 @@ class EmbeddingManager:
LCMistralEmbeddings,
OpenAIEmbeddings,
TeiEndpointEmbeddings,
VoyageAIEmbeddings,
)
self._vendors = [
@@ -73,6 +74,7 @@ class EmbeddingManager:
LCGoogleEmbeddings,
LCMistralEmbeddings,
TeiEndpointEmbeddings,
VoyageAIEmbeddings,
]
def __getitem__(self, key: str) -> BaseEmbeddings:

View File

@@ -52,9 +52,13 @@ class RerankingManager:
self._default = item.name
def load_vendors(self):
from kotaemon.rerankings import CohereReranking, TeiFastReranking
from kotaemon.rerankings import (
CohereReranking,
TeiFastReranking,
VoyageAIReranking,
)
self._vendors = [TeiFastReranking, CohereReranking]
self._vendors = [TeiFastReranking, CohereReranking, VoyageAIReranking]
def __getitem__(self, key: str) -> BaseReranking:
"""Get model by name"""