diff --git a/tests/test_qianfan_provider.py b/tests/test_qianfan_provider.py index d2c1d3e2..fa58dc38 100644 --- a/tests/test_qianfan_provider.py +++ b/tests/test_qianfan_provider.py @@ -59,6 +59,8 @@ class TestQianfanConstantsAndRouting(unittest.TestCase): 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") + parent_had_cow_cli = hasattr(plugins, "cow_cli") + old_parent_cow_cli = getattr(plugins, "cow_cli", None) module_names = ("plugins.cow_cli", "plugins.cow_cli.cow_cli") old_modules = { name: sys.modules[name] @@ -82,6 +84,10 @@ class TestQianfanConstantsAndRouting(unittest.TestCase): sys.modules[name] = old_modules[name] else: sys.modules.pop(name, None) + if parent_had_cow_cli: + plugins.cow_cli = old_parent_cow_cli + elif hasattr(plugins, "cow_cli"): + delattr(plugins, "cow_cli") self.assertEqual( cow_cli_plugin._resolve_bot_type_for_model("ernie-4.5-turbo-128k"),