diff --git a/pkg/config/migrations/m010_ollama_requester_config.py b/pkg/config/migrations/m010_ollama_requester_config.py new file mode 100644 index 00000000..56e49663 --- /dev/null +++ b/pkg/config/migrations/m010_ollama_requester_config.py @@ -0,0 +1,23 @@ +from __future__ import annotations + +from .. import migration + + +@migration.migration_class("ollama-requester-config", 10) +class MsgTruncatorConfigMigration(migration.Migration): + """迁移""" + + async def need_migrate(self) -> bool: + """判断当前环境是否需要运行此迁移""" + return 'ollama-chat' not in self.ap.provider_cfg.data['requester'] + + async def run(self): + """执行迁移""" + + self.ap.provider_cfg.data['requester']['ollama-chat'] = { + "base-url": "http://127.0.0.1:11434", + "args": {}, + "timeout": 600 + } + + await self.ap.provider_cfg.dump_config() diff --git a/pkg/core/stages/migrate.py b/pkg/core/stages/migrate.py index 2ad1e974..862d90af 100644 --- a/pkg/core/stages/migrate.py +++ b/pkg/core/stages/migrate.py @@ -6,6 +6,7 @@ from .. import stage, app from ...config import migration from ...config.migrations import m001_sensitive_word_migration, m002_openai_config_migration, m003_anthropic_requester_cfg_completion, m004_moonshot_cfg_completion from ...config.migrations import m005_deepseek_cfg_completion, m006_vision_config, m007_qcg_center_url, m008_ad_fixwin_config_migrate, m009_msg_truncator_cfg +from ...config.migrations import m010_ollama_requester_config @stage.stage_class("MigrationStage") diff --git a/templates/provider.json b/templates/provider.json index 309fb827..32878fe9 100644 --- a/templates/provider.json +++ b/templates/provider.json @@ -37,6 +37,11 @@ "base-url": "https://api.deepseek.com", "args": {}, "timeout": 120 + }, + "ollama-chat": { + "base-url": "http://127.0.0.1:11434", + "args": {}, + "timeout": 600 } }, "model": "gpt-3.5-turbo",