phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding reranking for voyage

Browse source
This commit is contained in:
Prathamesh 2025-04-16 08:49:48 +05:30
parent f670ebeb2f
commit df7459030d
5 changed files with 90 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -6604,6 +6604,8 @@ class ProviderConfigManager:
return litellm.InfinityRerankConfig()
elif litellm.LlmProviders.JINA_AI == provider:
return litellm.JinaAIRerankConfig()
elif litellm.LlmProviders.VOYAGE == provider:
return litellm.VoyageRerankConfig()
return litellm.CohereRerankConfig()
@staticmethod