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 supprot for OpenAI-API-Compatible llm (#1787)

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

#1771  add supprot for OpenAI-API-Compatible 

### 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-06 16:20:21 +08:00
committed by GitHub
parent 66e4113e0b
commit b67484e77d
12 changed files with 74 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
rag/llm/__init__.py
View File

@ -36,7 +36,8 @@ EmbeddingModel = {
"Bedrock": BedrockEmbed,
"Gemini": GeminiEmbed,
"NVIDIA": NvidiaEmbed,
"LM-Studio": LmStudioEmbed
"LM-Studio": LmStudioEmbed,
"OpenAI-API-Compatible": OpenAI_APIEmbed
}
@ -53,7 +54,8 @@ CvModel = {
"LocalAI": LocalAICV,
"NVIDIA": NvidiaCV,
"LM-Studio": LmStudioCV,
"StepFun":StepFunCV
"StepFun":StepFunCV,
"OpenAI-API-Compatible": OpenAI_APICV
}
@ -78,7 +80,8 @@ ChatModel = {
"OpenRouter": OpenRouterChat,
"StepFun": StepFunChat,
"NVIDIA": NvidiaChat,
"LM-Studio": LmStudioChat
"LM-Studio": LmStudioChat,
"OpenAI-API-Compatible": OpenAI_APIChat
}
@ -88,7 +91,8 @@ RerankModel = {
"Youdao": YoudaoRerank,
"Xinference": XInferenceRerank,
"NVIDIA": NvidiaRerank,
"LM-Studio": LmStudioRerank
"LM-Studio": LmStudioRerank,
"OpenAI-API-Compatible": OpenAI_APIRerank
}