gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1 from miguelmanlyx/copilot/add-ai-badgr-provider

Browse source 
Add AI Badgr as OpenAI-compatible provider
This commit is contained in:
miguelmanlyx 2025-12-14 16:03:11 +10:00 • committed by GitHub
commit eac090ccc5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 27 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
conf/llm_factories.json
View file

@ -3859,6 +3859,29 @@
"status": "1", "status": "1",
"llm": [] "llm": []
}, },
{
"name": "AI Badgr",
"logo": "",
"tags": "LLM",
"status": "1",
"llm": [
{
"llm_name": "basic",
"tags": "LLM,CHAT,Budget/Utility",
"model_type": "chat"
},
{
"llm_name": "normal",
"tags": "LLM,CHAT,Budget/Utility",
"model_type": "chat"
},
{
"llm_name": "premium",
"tags": "LLM,CHAT,Budget/Utility",
"model_type": "chat"
}
]
},
{ {
"name": "DeepInfra", "name": "DeepInfra",
"logo": "", "logo": "",

1
docs/references/supported_models.mdx
View file

@ -15,6 +15,7 @@ A complete list of models supported by RAGFlow, which will continue to expand.
| Provider | Chat | Embedding | Rerank | Img2txt | Speech2txt | TTS | | Provider | Chat | Embedding | Rerank | Img2txt | Speech2txt | TTS |
| --------------------- | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ | | --------------------- | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ |
| AI Badgr | :heavy_check_mark: | | | | | |
| Anthropic | :heavy_check_mark: | | | | | | | Anthropic | :heavy_check_mark: | | | | | |
| Azure-OpenAI | :heavy_check_mark: | :heavy_check_mark: | | :heavy_check_mark: | :heavy_check_mark: | | | Azure-OpenAI | :heavy_check_mark: | :heavy_check_mark: | | :heavy_check_mark: | :heavy_check_mark: | |
| BAAI | | :heavy_check_mark: | :heavy_check_mark: | | | | | BAAI | | :heavy_check_mark: | :heavy_check_mark: | | | |

3
rag/llm/__init__.py
View file

@ -54,6 +54,7 @@ class SupportedLiteLLMProvider(StrEnum):
MiniMax = "MiniMax" MiniMax = "MiniMax"
DeerAPI = "DeerAPI" DeerAPI = "DeerAPI"
GPUStack = "GPUStack" GPUStack = "GPUStack"
AI_Badgr = "AI Badgr"
OpenAI = "OpenAI" OpenAI = "OpenAI"
Azure_OpenAI = "Azure-OpenAI" Azure_OpenAI = "Azure-OpenAI"
@ -80,6 +81,7 @@ FACTORY_DEFAULT_BASE_URL = {
SupportedLiteLLMProvider.ZHIPU_AI: "https://open.bigmodel.cn/api/paas/v4", SupportedLiteLLMProvider.ZHIPU_AI: "https://open.bigmodel.cn/api/paas/v4",
SupportedLiteLLMProvider.MiniMax: "https://api.minimaxi.com/v1", SupportedLiteLLMProvider.MiniMax: "https://api.minimaxi.com/v1",
SupportedLiteLLMProvider.DeerAPI: "https://api.deerapi.com/v1", SupportedLiteLLMProvider.DeerAPI: "https://api.deerapi.com/v1",
SupportedLiteLLMProvider.AI_Badgr: "https://aibadgr.com/api/v1",
SupportedLiteLLMProvider.OpenAI: "https://api.openai.com/v1", SupportedLiteLLMProvider.OpenAI: "https://api.openai.com/v1",
} }
@ -116,6 +118,7 @@ LITELLM_PROVIDER_PREFIX = {
SupportedLiteLLMProvider.MiniMax: "openai/", SupportedLiteLLMProvider.MiniMax: "openai/",
SupportedLiteLLMProvider.DeerAPI: "openai/", SupportedLiteLLMProvider.DeerAPI: "openai/",
SupportedLiteLLMProvider.GPUStack: "openai/", SupportedLiteLLMProvider.GPUStack: "openai/",
SupportedLiteLLMProvider.AI_Badgr: "openai/",
SupportedLiteLLMProvider.OpenAI: "openai/", SupportedLiteLLMProvider.OpenAI: "openai/",
SupportedLiteLLMProvider.Azure_OpenAI: "azure/", SupportedLiteLLMProvider.Azure_OpenAI: "azure/",
} }