kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: restore cow cli parent module attribute

Browse Source
This commit is contained in:
jimmyzhuu
2026-04-29 16:12:08 +08:00
parent a4592ffdfe
commit 6f68ed6bce
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_qianfan_provider.py
View File

@@ -59,6 +59,8 @@ class TestQianfanConstantsAndRouting(unittest.TestCase):
old_plugin_path = plugins.instance.current_plugin_path old_plugin_path = plugins.instance.current_plugin_path
cow_cli_was_registered = "COW_CLI" in plugins.instance.plugins cow_cli_was_registered = "COW_CLI" in plugins.instance.plugins
old_cow_cli_plugin = plugins.instance.plugins.get("COW_CLI") 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") module_names = ("plugins.cow_cli", "plugins.cow_cli.cow_cli")
old_modules = { old_modules = {
name: sys.modules[name] name: sys.modules[name]
@@ -82,6 +84,10 @@ class TestQianfanConstantsAndRouting(unittest.TestCase):
sys.modules[name] = old_modules[name] sys.modules[name] = old_modules[name]
else: else:
sys.modules.pop(name, None) 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( self.assertEqual(
cow_cli_plugin._resolve_bot_type_for_model("ernie-4.5-turbo-128k"), cow_cli_plugin._resolve_bot_type_for_model("ernie-4.5-turbo-128k"),