pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add support for Voyage AI (#2159)

Browse Source 
### What problem does this PR solve?

#1853  #2138 add support for Voyage AI

### Type of change
- [x] New Feature (non-breaking change which adds functionality)

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
This commit is contained in:
黄腾
2024-08-29 16:14:49 +08:00
committed by GitHub
parent 15b78bd894
commit 99993e5026
9 changed files with 141 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rag/llm/__init__.py
View File

@ -45,7 +45,8 @@ EmbeddingModel = {
"Upstage": UpstageEmbed,
"SILICONFLOW": SILICONFLOWEmbed,
"Replicate": ReplicateEmbed,
"BaiduYiyan": BaiduYiyanEmbed
"BaiduYiyan": BaiduYiyanEmbed,
"Voyage AI": VoyageEmbed
}
@ -105,7 +106,7 @@ ChatModel = {
"Tencent Hunyuan": HunyuanChat,
"XunFei Spark": SparkChat,
"BaiduYiyan": BaiduYiyanChat,
"Anthropic": AnthropicChat
"Anthropic": VoyageChat,
}
@ -120,7 +121,8 @@ RerankModel = {
"cohere": CoHereRerank,
"TogetherAI": TogetherAIRerank,
"SILICONFLOW": SILICONFLOWRerank,
"BaiduYiyan": BaiduYiyanRerank
"BaiduYiyan": BaiduYiyanRerank,
"Voyage AI": VoyageRerank
}