chore(bailian): related configuration

This commit is contained in:
Junyan Qin
2025-02-12 11:25:28 +08:00
parent e17da4e2ee
commit 191f8866ae
6 changed files with 64 additions and 9 deletions
@@ -0,0 +1,27 @@
from __future__ import annotations
from .. import migration
@migration.migration_class("bailian-requester-config", 28)
class BailianRequesterConfigMigration(migration.Migration):
"""迁移"""
async def need_migrate(self) -> bool:
"""判断当前环境是否需要运行此迁移"""
return 'bailian-chat-completions' not in self.ap.provider_cfg.data['requester']
async def run(self):
"""执行迁移"""
self.ap.provider_cfg.data['keys']['bailian'] = [
"sk-xxxxxxx"
]
self.ap.provider_cfg.data['requester']['bailian-chat-completions'] = {
"base-url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
"args": {},
"timeout": 120
}
await self.ap.provider_cfg.dump_config()
+1 -1
View File
@@ -9,7 +9,7 @@ from ..migrations import m005_deepseek_cfg_completion, m006_vision_config, m007_
from ..migrations import m010_ollama_requester_config, m011_command_prefix_config, m012_runner_config, m013_http_api_config, m014_force_delay_config from ..migrations import m010_ollama_requester_config, m011_command_prefix_config, m012_runner_config, m013_http_api_config, m014_force_delay_config
from ..migrations import m015_gitee_ai_config, m016_dify_service_api, m017_dify_api_timeout_params, m018_xai_config, m019_zhipuai_config from ..migrations import m015_gitee_ai_config, m016_dify_service_api, m017_dify_api_timeout_params, m018_xai_config, m019_zhipuai_config
from ..migrations import m020_wecom_config, m021_lark_config, m022_lmstudio_config, m023_siliconflow_config, m024_discord_config, m025_gewechat_config from ..migrations import m020_wecom_config, m021_lark_config, m022_lmstudio_config, m023_siliconflow_config, m024_discord_config, m025_gewechat_config
from ..migrations import m026_qqofficial_config from ..migrations import m026_qqofficial_config, m028_aliyun_requester_config
@stage.stage_class("MigrationStage") @stage.stage_class("MigrationStage")
class MigrationStage(stage.BootingStage): class MigrationStage(stage.BootingStage):
+1 -1
View File
@@ -6,7 +6,7 @@ from . import entities, requester
from ...core import app from ...core import app
from . import token from . import token
from .requesters import chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl, xaichatcmpl, zhipuaichatcmpl, lmstudiochatcmpl, siliconflowchatcmpl, aliyunchatcmpl from .requesters import bailianchatcmpl, chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl, xaichatcmpl, zhipuaichatcmpl, lmstudiochatcmpl, siliconflowchatcmpl
FETCH_MODEL_LIST_URL = "https://api.qchatgpt.rockchin.top/api/v2/fetch/model_list" FETCH_MODEL_LIST_URL = "https://api.qchatgpt.rockchin.top/api/v2/fetch/model_list"
@@ -7,9 +7,9 @@ from .. import requester
from ....core import app from ....core import app
@requester.requester_class("aliyun-chat-completions") @requester.requester_class("bailian-chat-completions")
class AliyunChatCompletions(chatcmpl.OpenAIChatCompletions): class BailianChatCompletions(chatcmpl.OpenAIChatCompletions):
"""Aliyun ChatCompletion API 请求器""" """阿里云百炼大模型平台 ChatCompletion API 请求器"""
client: openai.AsyncClient client: openai.AsyncClient
@@ -18,4 +18,4 @@ class AliyunChatCompletions(chatcmpl.OpenAIChatCompletions):
def __init__(self, ap: app.Application): def __init__(self, ap: app.Application):
self.ap = ap self.ap = ap
self.requester_cfg = self.ap.provider_cfg.data['requester']['aliyun-chat-completions'] self.requester_cfg = self.ap.provider_cfg.data['requester']['bailian-chat-completions']
+3 -3
View File
@@ -26,8 +26,8 @@
"siliconflow": [ "siliconflow": [
"xxxxxxx" "xxxxxxx"
], ],
"aliyun": [ "bailian": [
"sk-aliyun_token" "sk-xxxxxxx"
] ]
}, },
"requester": { "requester": {
@@ -83,7 +83,7 @@
"args": {}, "args": {},
"timeout": 120 "timeout": 120
}, },
"aliyun-chat-completions": { "bailian-chat-completions": {
"args": {}, "args": {},
"base-url": "https://dashscope.aliyuncs.com/compatible-mode/v1", "base-url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
"timeout": 120 "timeout": 120
+28
View File
@@ -82,6 +82,14 @@
"type": "string" "type": "string"
}, },
"default": [] "default": []
},
"bailian": {
"type": "array",
"title": "阿里云百炼大模型平台 API 密钥",
"items": {
"type": "string"
},
"default": []
} }
} }
}, },
@@ -288,6 +296,26 @@
"default": 120 "default": 120
} }
} }
},
"bailian-chat-completions": {
"type": "object",
"title": "阿里云百炼大模型平台 API 请求配置",
"description": "仅可编辑 URL 和 超时时间,额外请求参数不支持可视化编辑,请到编辑器编辑",
"properties": {
"base-url": {
"type": "string",
"title": "API URL"
},
"args": {
"type": "object",
"default": {}
},
"timeout": {
"type": "number",
"title": "API 请求超时时间",
"default": 120
}
}
} }
} }
}, },