feat: 模型视觉多模态支持

2026-08-12 22:00:57 +00:00 · 2024-05-15 21:40:18 +08:00
parent 8807f02f36
commit d5b5d667a5
32 changed files with 596 additions and 72 deletions
@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+from .. import migration
+
+
+@migration.migration_class("vision-and-oss-config", 6)
+class VisionAndOSSConfigMigration(migration.Migration):
+    """迁移"""
+
+    async def need_migrate(self) -> bool:
+        """判断当前环境是否需要运行此迁移"""
+        return "enable-vision" not in self.ap.provider_cfg.data \
+            or "oss" not in self.ap.system_cfg.data
+
+    async def run(self):
+        """执行迁移"""
+        if "enable-vision" not in self.ap.provider_cfg.data:
+            self.ap.provider_cfg.data["enable-vision"] = False
+
+        if "oss" not in self.ap.system_cfg.data:
+            self.ap.system_cfg.data["oss"] = [
+                {
+                    "type": "aliyun",
+                    "endpoint": "https://oss-cn-hangzhou.aliyuncs.com",
+                    "public-read-base-url": "https://qchatgpt.oss-cn-hangzhou.aliyuncs.com",
+                    "access-key-id": "LTAI5tJ5Q5J8J6J5J5J5J5J5",
+                    "access-key-secret": "xxxxxx",
+                    "bucket": "qchatgpt",
+                    "prefix": "qchatgpt",
+                    "enable": False,
+                }
+            ]
+
+        await self.ap.provider_cfg.dump_config()
+        await self.ap.system_cfg.dump_config()