This commit is contained in:
miguelmanlyx 2025-12-15 19:56:39 +08:00 committed by GitHub
commit 149be6eb22
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 44 additions and 0 deletions

View file

@ -3859,6 +3859,29 @@
"status": "1",
"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",
"logo": "",

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 |
| --------------------- | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ |
| AI Badgr | :heavy_check_mark: | :heavy_check_mark: | | | | |
| Anthropic | :heavy_check_mark: | | | | | |
| Azure-OpenAI | :heavy_check_mark: | :heavy_check_mark: | | :heavy_check_mark: | :heavy_check_mark: | |
| BAAI | | :heavy_check_mark: | :heavy_check_mark: | | | |

View file

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

View file

@ -911,6 +911,15 @@ class GiteeEmbed(SILICONFLOWEmbed):
base_url = "https://ai.gitee.com/v1/embeddings"
super().__init__(key, model_name, base_url)
class AiBadgrEmbed(OpenAIEmbed):
_FACTORY_NAME = "AI Badgr"
def __init__(self, key, model_name, base_url="https://aibadgr.com/api/v1"):
if not base_url:
base_url = "https://aibadgr.com/api/v1"
super().__init__(key, model_name, base_url)
class DeepInfraEmbed(OpenAIEmbed):
_FACTORY_NAME = "DeepInfra"

View file

@ -0,0 +1,4 @@
<?xml version="1.0" encoding="UTF-8"?>
<svg version="1.1" xmlns="http://www.w3.org/2000/svg" width="48" height="48">
<path d="M0 0 C3.57426963 1.58348021 6.93089808 3.48126147 10.3125 5.4375 C11.23095703 5.95376953 12.14941406 6.47003906 13.09570312 7.00195312 C19.78304369 10.78304369 19.78304369 10.78304369 21 12 C21.41978793 15.9055194 21.46483625 19.82527993 21.5625 23.75 C21.62501953 24.84828125 21.68753906 25.9465625 21.75195312 27.078125 C21.9105491 34.99831199 21.9105491 34.99831199 19.41210938 37.74682617 C17.32997741 39.05815928 15.26334698 40.03982296 13 41 C11.87335937 41.66644531 10.74671875 42.33289062 9.5859375 43.01953125 C8.52632812 43.61121094 7.46671875 44.20289063 6.375 44.8125 C5.31023438 45.41707031 4.24546875 46.02164063 3.1484375 46.64453125 C-0.47709706 48.20540035 -2.09921716 48.64778343 -6 48 C-9.57656781 46.42203628 -12.93111774 44.51861146 -16.3125 42.5625 C-17.23095703 42.04623047 -18.14941406 41.52996094 -19.09570312 40.99804688 C-25.78304369 37.21695631 -25.78304369 37.21695631 -27 36 C-27.08726223 34.21928754 -27.10699041 32.4351686 -27.09765625 30.65234375 C-27.09443359 29.57275391 -27.09121094 28.49316406 -27.08789062 27.38085938 C-27.07951172 26.24455078 -27.07113281 25.10824219 -27.0625 23.9375 C-27.05798828 22.79732422 -27.05347656 21.65714844 -27.04882812 20.48242188 C-27.03699747 17.65489649 -27.0205141 14.8274736 -27 12 C-16.74609375 6.140625 -16.74609375 6.140625 -14.86669922 5.08007812 C-13.62333614 4.36066428 -12.39520285 3.61425683 -11.18408203 2.84179688 C-7.20754739 0.35933594 -4.65079291 -0.85538335 0 0 Z M-14.375 8.375 C-15.76525391 9.17550781 -15.76525391 9.17550781 -17.18359375 9.9921875 C-19.46627522 11.31002423 -21.73460518 12.65266387 -24 14 C-24 17.3 -24 20.6 -24 24 C-23.01 22.68 -22.02 21.36 -21 20 C-18.60009194 17.37510056 -17.41246434 16.13748811 -14 15 C-9.89266618 14.86076835 -6.05889069 14.94606098 -2 15.5625 C0.92192286 15.9979238 3.22731799 15.86339012 6.125 15.3125 C10.86268773 14.93042841 12.3479722 16.13054959 16 19 C16.66 19 17.32 19 18 19 C18 17.35 18 15.7 18 14 C15.46195322 12.53504663 12.91990283 11.07795678 10.375 9.625 C9.66214844 9.2125 8.94929688 8.8 8.21484375 8.375 C6.16320381 7.20801215 4.08806066 6.1001049 2 5 C1.05125 4.46375 0.1025 3.9275 -0.875 3.375 C-6.14674977 2.74239003 -9.95591655 5.72733232 -14.375 8.375 Z M-4 19 C-4 19.33 -4 19.66 -4 20 C-2.35 20 -0.7 20 1 20 C1 19.67 1 19.34 1 19 C-0.65 19 -2.3 19 -4 19 Z M17 21 C16.67 21.33 16.34 21.66 16 22 C14.845 22.144375 13.69 22.28875 12.5 22.4375 C10.7675 22.7159375 10.7675 22.7159375 9 23 C8.67 23.66 8.34 24.32 8 25 C9.25483449 28.42227589 11.3651481 30.54080489 14 33 C5.42857143 31.28571429 5.42857143 31.28571429 2 27 C1.67 27.99 1.34 28.98 1 30 C1.66 30 2.32 30 3 30 C3 30.99 3 31.98 3 33 C0.69 32.67 -1.62 32.34 -4 32 C-4.33 30.35 -4.66 28.7 -5 27 C-5.66 27 -6.32 27 -7 27 C-6.49396008 29.16874252 -6.00016187 30.99967627 -5 33 C-8.86648272 31.95902388 -12.49461443 30.96301592 -16 29 C-15.67 30.32 -15.34 31.64 -15 33 C-17.375 32.75 -17.375 32.75 -20 32 C-21.3125 29.9375 -21.3125 29.9375 -22 28 C-22.66 28 -23.32 28 -24 28 C-24 29.98 -24 31.96 -24 34 C-20.81042882 35.90837457 -17.5960446 37.77032342 -14.375 39.625 C-13.47136719 40.16640625 -12.56773438 40.7078125 -11.63671875 41.265625 C-6.56820701 44.29475409 -6.56820701 44.29475409 -0.875 44.625 C0.07375 44.08875 1.0225 43.5525 2 43 C2.64501465 42.6601709 3.2900293 42.3203418 3.95458984 41.97021484 C6.12293306 40.81624891 8.24908184 39.60518772 10.375 38.375 C11.47908203 37.74464844 11.47908203 37.74464844 12.60546875 37.1015625 C14.4063765 36.07247236 16.20357318 35.03689244 18 34 C18 29.71 18 25.42 18 21 C17.67 21 17.34 21 17 21 Z " fill="#0370FE" transform="translate(27,0)"/>
</svg>

After

Width:  |  Height:  |  Size: 3.6 KiB

View file

@ -90,6 +90,7 @@ export const LlmIcon = ({
LLMFactory.Gemini,
LLMFactory.StepFun,
LLMFactory.MinerU,
LLMFactory.AIBadgr,
// LLMFactory.DeerAPI,
];
if (svgIcons.includes(name as LLMFactory)) {

View file

@ -61,6 +61,7 @@ export enum LLMFactory {
JiekouAI = 'Jiekou.AI',
Builtin = 'Builtin',
MinerU = 'MinerU',
AIBadgr = 'AI Badgr',
}
// Please lowercase the file name
@ -127,6 +128,7 @@ export const IconMap = {
[LLMFactory.JiekouAI]: 'jiekouai',
[LLMFactory.Builtin]: 'builtin',
[LLMFactory.MinerU]: 'mineru',
[LLMFactory.AIBadgr]: 'aibadgr',
};
export const APIMapUrl = {
@ -177,4 +179,5 @@ export const APIMapUrl = {
[LLMFactory.DeerAPI]: 'https://api.deerapi.com/token',
[LLMFactory.TokenPony]: 'https://www.tokenpony.cn/#/user/keys',
[LLMFactory.DeepInfra]: 'https://deepinfra.com/dash/api_keys',
[LLMFactory.AIBadgr]: 'https://aibadgr.com/api-keys',
};