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/", }