From ebf420da1769d46ab24c8fca75a1567c0986b4e4 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Sun, 14 Dec 2025 05:47:35 +0000 Subject: [PATCH 1/7] Initial plan From 088d3a983c4a6695065adae854b71e1d1204e556 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Sun, 14 Dec 2025 05:52:35 +0000 Subject: [PATCH 2/7] Add AI Badgr as OpenAI-compatible provider with tier-based models Co-authored-by: miguelmanlyx <220451577+miguelmanlyx@users.noreply.github.com> --- conf/llm_factories.json | 23 +++++++++++++++++++++++ docs/references/supported_models.mdx | 1 + rag/llm/__init__.py | 3 +++ 3 files changed, 27 insertions(+) diff --git a/conf/llm_factories.json b/conf/llm_factories.json index 4474cc71e..6255b58b1 100644 --- a/conf/llm_factories.json +++ b/conf/llm_factories.json @@ -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": "", diff --git a/docs/references/supported_models.mdx b/docs/references/supported_models.mdx index fc88fe091..59df1298c 100644 --- a/docs/references/supported_models.mdx +++ b/docs/references/supported_models.mdx @@ -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: | | | | | | | 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: | | | | diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py index 4b36d8a05..0b62de94e 100644 --- a/rag/llm/__init__.py +++ b/rag/llm/__init__.py @@ -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/", } From e52c40baab06362a8216cbbcb92859984ee60454 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Mon, 15 Dec 2025 07:30:00 +0000 Subject: [PATCH 3/7] Initial plan From 9799231f880ae369e258d3a2f9c4c3743096aaf6 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Mon, 15 Dec 2025 07:35:17 +0000 Subject: [PATCH 4/7] Add AI Badgr to frontend LLM factory constants Co-authored-by: miguelmanlyx <220451577+miguelmanlyx@users.noreply.github.com> --- web/src/constants/llm.ts | 3 +++ 1 file changed, 3 insertions(+) diff --git a/web/src/constants/llm.ts b/web/src/constants/llm.ts index 1ff5f5387..2ab775657 100644 --- a/web/src/constants/llm.ts +++ b/web/src/constants/llm.ts @@ -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]: 'openai-api', }; 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', }; From 2762b970cadef1e7a2d71b7788d9cf979a1b440f Mon Sep 17 00:00:00 2001 From: michaelmanley <55236695+michaelbrinkworth@users.noreply.github.com> Date: Mon, 15 Dec 2025 18:07:11 +1000 Subject: [PATCH 5/7] add logo --- web/src/assets/svg/llm/aibadgr.svg | 3 +++ web/src/components/svg-icon.tsx | 1 + web/src/constants/llm.ts | 2 +- 3 files changed, 5 insertions(+), 1 deletion(-) create mode 100644 web/src/assets/svg/llm/aibadgr.svg diff --git a/web/src/assets/svg/llm/aibadgr.svg b/web/src/assets/svg/llm/aibadgr.svg new file mode 100644 index 000000000..e8bd5130d --- /dev/null +++ b/web/src/assets/svg/llm/aibadgr.svg @@ -0,0 +1,3 @@ + + - + + + From 526fd0785860a3345432ab04903abe16e768e361 Mon Sep 17 00:00:00 2001 From: michaelmanley <55236695+michaelbrinkworth@users.noreply.github.com> Date: Mon, 15 Dec 2025 19:34:36 +1000 Subject: [PATCH 7/7] add embeddings --- docs/references/supported_models.mdx | 2 +- rag/llm/embedding_model.py | 9 +++++++++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/docs/references/supported_models.mdx b/docs/references/supported_models.mdx index 59df1298c..9ae7b36f8 100644 --- a/docs/references/supported_models.mdx +++ b/docs/references/supported_models.mdx @@ -15,7 +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: | | | | | | +| 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: | | | | diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index 1890b68d0..9c640e941 100644 --- a/rag/llm/embedding_model.py +++ b/rag/llm/embedding_model.py @@ -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"