Skip to content

Add Qiniu Cloud (七牛云) as a built-in model provider #331

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
HannahKking wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Add Qiniu Cloud (七牛云) as a built-in model provider #331

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 30 additions & 0 deletions app/shared/ai_models_available.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -325,6 +325,7 @@ var BuiltInModels = []*BaseModelConfigSchema{
Providers: []BaseModelUsesProvider{
{Provider: ModelProviderDeepSeek, ModelName: "deepseek/deepseek-chat"},
{Provider: ModelProviderOpenRouter, ModelName: "deepseek/deepseek-chat-v3"},
{Provider: ModelProviderQiniu, ModelName: "deepseek-v3"},
},
},
{
Expand All @@ -343,6 +344,7 @@ var BuiltInModels = []*BaseModelConfigSchema{
Providers: []BaseModelUsesProvider{
{Provider: ModelProviderDeepSeek, ModelName: "deepseek/deepseek-reasoner"},
{Provider: ModelProviderOpenRouter, ModelName: "deepseek/deepseek-r1-0528"},
{Provider: ModelProviderQiniu, ModelName: "deepseek-r1"},
},
},
{
Expand Down Expand Up @@ -583,6 +585,34 @@ var BuiltInModels = []*BaseModelConfigSchema{
{Provider: ModelProviderOpenRouter, ModelName: "perplexity/sonar-reasoning"},
},
},

// Qiniu Cloud (七牛云) models via AI Inference Hub (api.qnaigc.com)
{
ModelTag: "qiniu/kimi-k2",
Publisher: ModelPublisherQwen,
Description: "Kimi K2 (via Qiniu)",
BaseModelShared: BaseModelShared{
DefaultMaxConvoTokens: 15000, MaxTokens: 256000,
MaxOutputTokens: 16384, ReservedOutputTokens: 16384,
PreferredOutputFormat: ModelOutputFormatXml,
},
Providers: []BaseModelUsesProvider{
{Provider: ModelProviderQiniu, ModelName: "moonshotai/kimi-k2.5"},
},
},
{
ModelTag: "qiniu/glm-5",
Publisher: ModelPublisherQwen,
Description: "GLM-5 (via Qiniu)",
BaseModelShared: BaseModelShared{
DefaultMaxConvoTokens: 15000, MaxTokens: 128000,
MaxOutputTokens: 8192, ReservedOutputTokens: 8192,
PreferredOutputFormat: ModelOutputFormatXml,
},
Providers: []BaseModelUsesProvider{
{Provider: ModelProviderQiniu, ModelName: "z-ai/glm-5"},
},
},
}

var BuiltInBaseModelsById = map[ModelId]*BaseModelConfigSchema{}
Expand Down
8 changes: 8 additions & 0 deletions app/shared/ai_models_providers.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ const GoogleAIStudioApiKeyEnvVar = "GEMINI_API_KEY"
const AzureOpenAIEnvVar = "AZURE_OPENAI_API_KEY"
const DeepSeekApiKeyEnvVar = "DEEPSEEK_API_KEY"
const PerplexityApiKeyEnvVar = "PERPLEXITY_API_KEY"
const QiniuApiKeyEnvVar = "QINIU_API_KEY"

// not set directly via env vars, but used for auth var resolution
const AnthropicClaudeMaxTokenEnvVar = "ANTHROPIC_CLAUDE_MAX_TOKEN"
Expand Down Expand Up @@ -46,6 +47,7 @@ const (
ModelProviderAzureOpenAI ModelProvider = "azure-openai"
ModelProviderDeepSeek ModelProvider = "deepseek"
ModelProviderPerplexity ModelProvider = "perplexity"
ModelProviderQiniu ModelProvider = "qiniu"

ModelProviderAmazonBedrock ModelProvider = "aws-bedrock"

Expand All @@ -72,6 +74,7 @@ var AllModelProviders = []ModelProvider{
ModelProviderAmazonBedrock,
ModelProviderDeepSeek,
ModelProviderPerplexity,
ModelProviderQiniu,
ModelProviderOllama,
ModelProviderCustom,
}
Expand Down Expand Up @@ -216,6 +219,11 @@ var BuiltInModelProviderConfigs = map[ModelProvider]ModelProviderConfigSchema{
SkipAuth: true,
LocalOnly: true,
},
ModelProviderQiniu: {
Provider: ModelProviderQiniu,
BaseUrl: "https://api.qnaigc.com/v1",
ApiKeyEnvVar: QiniuApiKeyEnvVar,
},
}

var BuiltInModelProviderConfigsByComposite = map[string]ModelProviderConfigSchema{}
Expand Down