feat: add supports for volcengine (#1114)

pkg/core/migrations/m032_volcengine_config.py

@@ -0,0 +1,27 @@
+from __future__ import annotations
+
+from .. import migration
+
+
+@migration.migration_class("volcengine-requester-config", 32)
+class VolcengineRequesterConfigMigration(migration.Migration):
+    """迁移"""
+
+    async def need_migrate(self) -> bool:
+        """判断当前环境是否需要运行此迁移"""
+        
+        return 'volcengine-chat-completions' not in self.ap.provider_cfg.data['requester']
+
+    async def run(self):
+        """执行迁移"""
+        self.ap.provider_cfg.data['keys']['volcengine'] = [
+            "xxxxxxxx"
+        ]
+
+        self.ap.provider_cfg.data['requester']['volcengine-chat-completions'] = {
+            "base-url": "https://ark.cn-beijing.volces.com/api/v3",
+            "args": {},
+            "timeout": 120
+        }
+
+        await self.ap.provider_cfg.dump_config()

pkg/core/stages/migrate.py

@@ -10,7 +10,7 @@ from ..migrations import m010_ollama_requester_config, m011_command_prefix_confi
 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 m026_qqofficial_config, m027_wx_official_account_config, m028_aliyun_requester_config
-from ..migrations import m029_dashscope_app_api_config, m030_lark_config_cmpl, m031_dingtalk_config
+from ..migrations import m029_dashscope_app_api_config, m030_lark_config_cmpl, m031_dingtalk_config, m032_volcengine_config
 
 
 @stage.stage_class("MigrationStage")

pkg/provider/modelmgr/modelmgr.py

@@ -6,7 +6,7 @@ from . import entities, requester
 from ...core import app
 
 from . import token
-from .requesters import bailianchatcmpl, chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl, xaichatcmpl, zhipuaichatcmpl, lmstudiochatcmpl, siliconflowchatcmpl
+from .requesters import bailianchatcmpl, chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl, xaichatcmpl, zhipuaichatcmpl, lmstudiochatcmpl, siliconflowchatcmpl, volcenginechatcmpl
 
 FETCH_MODEL_LIST_URL = "https://api.qchatgpt.rockchin.top/api/v2/fetch/model_list"
 

pkg/provider/modelmgr/requesters/volcenginechatcmpl.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+import openai
+
+from . import chatcmpl
+from .. import requester
+from ....core import app
+
+
+@requester.requester_class("volcengine-chat-completions")
+class VolcengineChatCompletions(chatcmpl.OpenAIChatCompletions):
+    """火山方舟大模型平台 ChatCompletion API 请求器"""
+
+    client: openai.AsyncClient
+
+    requester_cfg: dict
+
+    def __init__(self, ap: app.Application):
+        self.ap = ap
+
+        self.requester_cfg = self.ap.provider_cfg.data['requester']['volcengine-chat-completions']

templates/provider.json

@@ -28,6 +28,9 @@
         ],
         "bailian": [
             "sk-xxxxxxx"
+        ],
+        "volcengine": [
+            "xxxxxxxx"
         ]
     },
     "requester": {
@@ -87,6 +90,11 @@
             "args": {},
             "base-url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
             "timeout": 120
+        },
+        "volcengine-chat-completions": {
+            "args": {},
+            "base-url": "https://ark.cn-beijing.volces.com/api/v3",
+            "timeout": 120
         }
     },
     "model": "gpt-4o",

templates/schema/provider.json

@@ -90,6 +90,14 @@
                         "type": "string"
                     },
                     "default": []
+                },
+                "volcengine": {
+                    "type": "array",
+                    "title": "火山引擎大模型平台 API 密钥",
+                    "items": {
+                        "type": "string"
+                    },
+                    "default": []
                 }
             }
         },
@@ -316,6 +324,26 @@
                             "default": 120
                         }
                     }
+                },
+                "volcengine-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
+                        }
+                    }
                 }
             }
         },