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 cohere (#1849)

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

_Briefly describe what this PR aims to solve. Include background context
that will help reviewers understand the purpose of the PR._

### 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-07 18:40:51 +08:00
committed by GitHub
parent 60428c4ad2
commit e34817c2a9
10 changed files with 260 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rag/llm/__init__.py
View File

@ -37,7 +37,8 @@ EmbeddingModel = {
"Gemini": GeminiEmbed,
"NVIDIA": NvidiaEmbed,
"LM-Studio": LmStudioEmbed,
"OpenAI-API-Compatible": OpenAI_APIEmbed
"OpenAI-API-Compatible": OpenAI_APIEmbed,
"cohere": CoHereEmbed
}
@ -81,7 +82,8 @@ ChatModel = {
"StepFun": StepFunChat,
"NVIDIA": NvidiaChat,
"LM-Studio": LmStudioChat,
"OpenAI-API-Compatible": OpenAI_APIChat
"OpenAI-API-Compatible": OpenAI_APIChat,
"cohere": CoHereChat
}
@ -92,7 +94,8 @@ RerankModel = {
"Xinference": XInferenceRerank,
"NVIDIA": NvidiaRerank,
"LM-Studio": LmStudioRerank,
"OpenAI-API-Compatible": OpenAI_APIRerank
"OpenAI-API-Compatible": OpenAI_APIRerank,
"cohere": CoHereRerank
}