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..9ae7b36f8 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: | :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/", } diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index 58cdd8e76..fb13516f7 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" diff --git a/web/src/assets/svg/llm/aibadgr.svg b/web/src/assets/svg/llm/aibadgr.svg new file mode 100644 index 000000000..630ae5ad5 --- /dev/null +++ b/web/src/assets/svg/llm/aibadgr.svg @@ -0,0 +1,4 @@ + + + + diff --git a/web/src/components/svg-icon.tsx b/web/src/components/svg-icon.tsx index 756336621..ec32ca754 100644 --- a/web/src/components/svg-icon.tsx +++ b/web/src/components/svg-icon.tsx @@ -90,6 +90,7 @@ export const LlmIcon = ({ LLMFactory.Gemini, LLMFactory.StepFun, LLMFactory.MinerU, + LLMFactory.AIBadgr, // LLMFactory.DeerAPI, ]; if (svgIcons.includes(name as LLMFactory)) { diff --git a/web/src/constants/llm.ts b/web/src/constants/llm.ts index 1ff5f5387..7f36fb946 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]: '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', };