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:
committed by
GitHub
parent
c33bedca9e
commit
5132288386
@@ -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:
|
||||
|
@@ -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"""
|
||||
|
Reference in New Issue
Block a user