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

fix: avoid cow cli import side effects

Browse Source
This commit is contained in:
jimmyzhuu
2026-04-29 16:04:48 +08:00
parent 9eeca70292
commit 7cd7bd1a48
2 changed files with 46 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_qianfan_provider.py
View File

@@ -54,14 +54,24 @@ class TestQianfanConstantsAndRouting(unittest.TestCase):
def test_cow_cli_routes_ernie_models_to_qianfan(self):
from common import const
from plugins.cow_cli.cow_cli import CowCli
import plugins
old_plugin_path = plugins.instance.current_plugin_path
plugins.instance.current_plugin_path = os.path.join(
os.path.dirname(__file__), "..", "plugins", "cow_cli"
)
try:
import plugins.cow_cli.cow_cli
cow_cli_plugin = plugins.instance.plugins["COW_CLI"]
finally:
plugins.instance.current_plugin_path = old_plugin_path
self.assertEqual(
CowCli._resolve_bot_type_for_model("ernie-4.5-turbo-128k"),
cow_cli_plugin._resolve_bot_type_for_model("ernie-4.5-turbo-128k"),
const.QIANFAN,
)
self.assertEqual(
CowCli._resolve_bot_type_for_model("qianfan"),
cow_cli_plugin._resolve_bot_type_for_model("qianfan"),
const.QIANFAN,
)