fix:优化图片识别,优化webui配置和排版,优化聊天流监控,新增mcp显示,新增prompt修改面板,优化插件状态显示,优化长期记忆控制台,
This commit is contained in:
@@ -1,33 +1,22 @@
|
||||
from typing import Any
|
||||
|
||||
import pytest
|
||||
|
||||
from src.config import config as config_module
|
||||
from src.config.config import Config, ConfigManager, ModelConfig
|
||||
|
||||
|
||||
class _StartupUpgradeExit(Exception):
|
||||
pass
|
||||
|
||||
|
||||
def test_initialize_upgrades_bot_and_model_config_before_exit(monkeypatch):
|
||||
def test_initialize_upgrades_bot_and_model_config_without_exit(monkeypatch):
|
||||
manager = ConfigManager()
|
||||
loaded_config_classes: list[type[Any]] = []
|
||||
exit_codes: list[int | None] = []
|
||||
warnings: list[Any] = []
|
||||
|
||||
def fake_load_config_from_file(config_class, config_path, new_ver, override_repr=False):
|
||||
loaded_config_classes.append(config_class)
|
||||
return object(), True
|
||||
|
||||
def fake_exit(code: int | None = None):
|
||||
exit_codes.append(code)
|
||||
raise _StartupUpgradeExit
|
||||
|
||||
monkeypatch.setattr(config_module, "load_config_from_file", fake_load_config_from_file)
|
||||
monkeypatch.setattr(config_module.sys, "exit", fake_exit)
|
||||
monkeypatch.setattr(ConfigManager, "_warn_if_vlm_not_configured", lambda self, model_config: warnings.append(model_config))
|
||||
|
||||
with pytest.raises(_StartupUpgradeExit):
|
||||
manager.initialize()
|
||||
manager.initialize()
|
||||
|
||||
assert loaded_config_classes == [Config, ModelConfig]
|
||||
assert exit_codes == [0]
|
||||
assert warnings == [manager.model_config]
|
||||
|
||||
Reference in New Issue
Block a user