feat(agent-runtime): 接入多服务商安全配置
This commit is contained in:
bruce
@@ -0,0 +1,286 @@
|
||||
"""模型服务商预设目录。
|
||||
|
||||
预设只保存公开的协议与端点，不保存用户凭据。需要云 IAM、部署名称或
|
||||
业务空间域名的服务商仍展示在目录中，但不会误导用户只靠 API Key 即可连接。
|
||||
"""
|
||||
|
||||
from dataclasses import dataclass
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class ModelProvider:
|
||||
"""描述一个可展示、可验证的模型服务商预设。"""
|
||||
|
||||
code: str
|
||||
name: str
|
||||
vendor: str
|
||||
category: str
|
||||
protocol: str
|
||||
base_url: str = ""
|
||||
models_path: str = "models"
|
||||
default_model: str = ""
|
||||
key_only: bool = True
|
||||
note: str = ""
|
||||
mark: str = "AI"
|
||||
|
||||
@property
|
||||
def protocol_label(self) -> str:
|
||||
labels = {
|
||||
"openai_responses": "Responses API",
|
||||
"openai_compatible": "OpenAI 兼容",
|
||||
"anthropic_messages": "Messages API",
|
||||
"gemini": "Gemini API",
|
||||
"cloud": "云平台接口",
|
||||
}
|
||||
return labels.get(self.protocol, self.protocol)
|
||||
|
||||
|
||||
PROVIDERS = (
|
||||
ModelProvider(
|
||||
"openai",
|
||||
"OpenAI",
|
||||
"OpenAI",
|
||||
"official",
|
||||
"openai_responses",
|
||||
"https://api.openai.com/v1",
|
||||
default_model="gpt-5.6-terra",
|
||||
mark="OA",
|
||||
),
|
||||
ModelProvider(
|
||||
"anthropic",
|
||||
"Anthropic Claude",
|
||||
"Anthropic",
|
||||
"official",
|
||||
"anthropic_messages",
|
||||
"https://api.anthropic.com/v1",
|
||||
default_model="claude-sonnet-4-6",
|
||||
mark="CL",
|
||||
),
|
||||
ModelProvider(
|
||||
"deepseek",
|
||||
"DeepSeek",
|
||||
"深度求索",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
"https://api.deepseek.com",
|
||||
default_model="deepseek-chat",
|
||||
mark="DS",
|
||||
),
|
||||
ModelProvider(
|
||||
"gemini",
|
||||
"Google Gemini",
|
||||
"Google",
|
||||
"official",
|
||||
"gemini",
|
||||
"https://generativelanguage.googleapis.com/v1beta",
|
||||
models_path="models",
|
||||
default_model="gemini-2.5-flash",
|
||||
mark="GE",
|
||||
),
|
||||
ModelProvider(
|
||||
"aliyun",
|
||||
"阿里云百炼 / Qwen",
|
||||
"阿里云",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
"https://dashscope.aliyuncs.com/compatible-mode/v1",
|
||||
default_model="qwen-plus",
|
||||
mark="QW",
|
||||
),
|
||||
ModelProvider(
|
||||
"zhipu",
|
||||
"智谱 AI / Z.AI",
|
||||
"智谱",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
"https://open.bigmodel.cn/api/paas/v4",
|
||||
default_model="glm-4.5",
|
||||
mark="GL",
|
||||
),
|
||||
ModelProvider(
|
||||
"xiaomi_mimo",
|
||||
"小米 MiMo",
|
||||
"小米",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
key_only=False,
|
||||
note="官方端点及账户区域确认后开放",
|
||||
mark="MI",
|
||||
),
|
||||
ModelProvider(
|
||||
"siliconflow",
|
||||
"硅基流动",
|
||||
"SiliconFlow",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
"https://api.siliconflow.cn/v1",
|
||||
default_model="deepseek-ai/DeepSeek-V3",
|
||||
mark="SF",
|
||||
),
|
||||
ModelProvider(
|
||||
"moonshot",
|
||||
"月之暗面 Kimi",
|
||||
"Moonshot AI",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
"https://api.moonshot.cn/v1",
|
||||
default_model="moonshot-v1-8k",
|
||||
mark="KM",
|
||||
),
|
||||
ModelProvider(
|
||||
"minimax",
|
||||
"MiniMax",
|
||||
"MiniMax",
|
||||
"domestic",
|
||||
"openai_compatible",
|
||||
key_only=False,
|
||||
note="部分账户还需要 Group ID",
|
||||
mark="MM",
|
||||
),
|
||||
ModelProvider(
|
||||
"volcengine",
|
||||
"火山引擎方舟 / 豆包",
|
||||
"火山引擎",
|
||||
"domestic",
|
||||
"cloud",
|
||||
key_only=False,
|
||||
note="需要接入点或推理端点配置",
|
||||
mark="DB",
|
||||
),
|
||||
ModelProvider(
|
||||
"baidu",
|
||||
"百度智能云千帆 / 文心",
|
||||
"百度智能云",
|
||||
"domestic",
|
||||
"cloud",
|
||||
key_only=False,
|
||||
note="需要应用或云鉴权配置",
|
||||
mark="BD",
|
||||
),
|
||||
ModelProvider(
|
||||
"tencent",
|
||||
"腾讯混元",
|
||||
"腾讯云",
|
||||
"domestic",
|
||||
"cloud",
|
||||
key_only=False,
|
||||
note="需要腾讯云鉴权配置",
|
||||
mark="HY",
|
||||
),
|
||||
ModelProvider(
|
||||
"huawei",
|
||||
"华为云 ModelArts / 盘古",
|
||||
"华为云",
|
||||
"domestic",
|
||||
"cloud",
|
||||
key_only=False,
|
||||
note="需要项目、区域及云鉴权配置",
|
||||
mark="HW",
|
||||
),
|
||||
ModelProvider(
|
||||
"openrouter",
|
||||
"OpenRouter",
|
||||
"OpenRouter",
|
||||
"gateway",
|
||||
"openai_compatible",
|
||||
"https://openrouter.ai/api/v1",
|
||||
default_model="openai/gpt-4o-mini",
|
||||
mark="OR",
|
||||
),
|
||||
ModelProvider(
|
||||
"groq",
|
||||
"Groq",
|
||||
"Groq",
|
||||
"gateway",
|
||||
"openai_compatible",
|
||||
"https://api.groq.com/openai/v1",
|
||||
default_model="llama-3.3-70b-versatile",
|
||||
mark="GQ",
|
||||
),
|
||||
ModelProvider(
|
||||
"together",
|
||||
"Together AI",
|
||||
"Together AI",
|
||||
"gateway",
|
||||
"openai_compatible",
|
||||
"https://api.together.xyz/v1",
|
||||
mark="TO",
|
||||
),
|
||||
ModelProvider(
|
||||
"mistral",
|
||||
"Mistral AI",
|
||||
"Mistral",
|
||||
"official",
|
||||
"openai_compatible",
|
||||
"https://api.mistral.ai/v1",
|
||||
default_model="mistral-large-latest",
|
||||
mark="MS",
|
||||
),
|
||||
ModelProvider(
|
||||
"cohere",
|
||||
"Cohere",
|
||||
"Cohere",
|
||||
"official",
|
||||
"openai_compatible",
|
||||
key_only=False,
|
||||
note="原生协议适配待接入",
|
||||
mark="CO",
|
||||
),
|
||||
ModelProvider(
|
||||
"xai",
|
||||
"xAI Grok",
|
||||
"xAI",
|
||||
"official",
|
||||
"openai_compatible",
|
||||
"https://api.x.ai/v1",
|
||||
default_model="grok-4",
|
||||
mark="XA",
|
||||
),
|
||||
ModelProvider(
|
||||
"azure_openai",
|
||||
"Azure OpenAI",
|
||||
"Microsoft Azure",
|
||||
"cloud",
|
||||
"cloud",
|
||||
key_only=False,
|
||||
note="需要资源端点和部署名称",
|
||||
mark="AZ",
|
||||
),
|
||||
ModelProvider(
|
||||
"aws_bedrock",
|
||||
"AWS Bedrock",
|
||||
"Amazon Web Services",
|
||||
"cloud",
|
||||
"cloud",
|
||||
key_only=False,
|
||||
note="需要区域和 AWS IAM 凭据",
|
||||
mark="AW",
|
||||
),
|
||||
ModelProvider(
|
||||
"custom",
|
||||
"自定义 OpenAI 兼容服务",
|
||||
"自定义",
|
||||
"gateway",
|
||||
"openai_compatible",
|
||||
key_only=False,
|
||||
note="需要 Base URL 和模型 ID",
|
||||
mark="+",
|
||||
),
|
||||
)
|
||||
|
||||
PROVIDER_BY_CODE = {provider.code: provider for provider in PROVIDERS}
|
||||
|
||||
CATEGORY_LABELS = {
|
||||
"all": "全部",
|
||||
"official": "官方直连",
|
||||
"domestic": "国内平台",
|
||||
"gateway": "云与聚合",
|
||||
"cloud": "云平台",
|
||||
"connected": "已连接",
|
||||
}
|
||||
|
||||
|
||||
def get_provider(code: str) -> ModelProvider | None:
|
||||
"""按稳定编码读取预设，未知编码返回空值。"""
|
||||
|
||||
return PROVIDER_BY_CODE.get(code)
|
||||
Reference in New Issue
Block a user