From a4592ffdfe8af52c27412b23443e1b331abbda23 Mon Sep 17 00:00:00 2001 From: jimmyzhuu Date: Wed, 29 Apr 2026 16:08:40 +0800 Subject: [PATCH] test: isolate cow cli plugin import --- plugins/cow_cli/cow_cli.py | 2 +- tests/test_qianfan_provider.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) diff --git a/plugins/cow_cli/cow_cli.py b/plugins/cow_cli/cow_cli.py index bca045c3..42d27330 100644 --- a/plugins/cow_cli/cow_cli.py +++ b/plugins/cow_cli/cow_cli.py @@ -428,7 +428,7 @@ class CowCliPlugin(Plugin): @staticmethod def _resolve_bot_type_for_model(model_name: str) -> str: - """Resolve bot_type from model name, reusing AgentBridge mapping.""" + """Resolve bot_type from model name, matching AgentBridge mapping.""" from common import const _EXACT = { "wenxin": const.BAIDU, "wenxin-4": const.BAIDU, diff --git a/tests/test_qianfan_provider.py b/tests/test_qianfan_provider.py index 3d3e47c5..d2c1d3e2 100644 --- a/tests/test_qianfan_provider.py +++ b/tests/test_qianfan_provider.py @@ -57,6 +57,14 @@ class TestQianfanConstantsAndRouting(unittest.TestCase): import plugins old_plugin_path = plugins.instance.current_plugin_path + cow_cli_was_registered = "COW_CLI" in plugins.instance.plugins + old_cow_cli_plugin = plugins.instance.plugins.get("COW_CLI") + module_names = ("plugins.cow_cli", "plugins.cow_cli.cow_cli") + old_modules = { + name: sys.modules[name] + for name in module_names + if name in sys.modules + } plugins.instance.current_plugin_path = os.path.join( os.path.dirname(__file__), "..", "plugins", "cow_cli" ) @@ -65,6 +73,15 @@ class TestQianfanConstantsAndRouting(unittest.TestCase): cow_cli_plugin = plugins.instance.plugins["COW_CLI"] finally: plugins.instance.current_plugin_path = old_plugin_path + if cow_cli_was_registered: + plugins.instance.plugins["COW_CLI"] = old_cow_cli_plugin + else: + plugins.instance.plugins.pop("COW_CLI", None) + for name in module_names: + if name in old_modules: + sys.modules[name] = old_modules[name] + else: + sys.modules.pop(name, None) self.assertEqual( cow_cli_plugin._resolve_bot_type_for_model("ernie-4.5-turbo-128k"),