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 Baidu yiyan (#2049)

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

add support for Baidu yiyan

### Type of change

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

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
This commit is contained in:
黄腾
2024-08-22 16:45:15 +08:00
committed by GitHub
parent 21f2c5838b
commit 733219cc3f
17 changed files with 307 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rag/llm/__init__.py
View File

@ -43,7 +43,8 @@ EmbeddingModel = {
"PerfXCloud": PerfXCloudEmbed,
"Upstage": UpstageEmbed,
"SILICONFLOW": SILICONFLOWEmbed,
"Replicate": ReplicateEmbed
"Replicate": ReplicateEmbed,
"BaiduYiyan": BaiduYiyanEmbed
}
@ -101,7 +102,8 @@ ChatModel = {
"01.AI": YiChat,
"Replicate": ReplicateChat,
"Tencent Hunyuan": HunyuanChat,
"XunFei Spark": SparkChat
"XunFei Spark": SparkChat,
"BaiduYiyan": BaiduYiyanChat
}
@ -115,7 +117,8 @@ RerankModel = {
"OpenAI-API-Compatible": OpenAI_APIRerank,
"cohere": CoHereRerank,
"TogetherAI": TogetherAIRerank,
"SILICONFLOW": SILICONFLOWRerank
"SILICONFLOW": SILICONFLOWRerank,
"BaiduYiyan": BaiduYiyanRerank
}