Merge pull request #90 from chg0901/patch-9

Update qwen_gen_data_NoBash.py 添加对qwen-max的异常处理
2024-03-17 13:16:01 +08:00 · 2024-03-17 13:16:01 +08:00 · 1c2f697d72
commit 1c2f697d72
parent 025061b0ab 24a16455ab
1 changed files with 8 additions and 1 deletions
--- a/generate_data/qwen_gen_data_NoBash.py
+++ b/generate_data/qwen_gen_data_NoBash.py
@ -23,6 +23,13 @@ def qwen_api(data, emo):
    病人：病人的咨询或陈述 
    医生：医生的安抚和建议
    '''
+    try:
+        response = dashscope.Generation.call(
+        model='qwen-max',
+        prompt=prompt,
+        history=[],
+    )
+    except:
        response = dashscope.Generation.call(
        model='qwen-max',
        prompt=prompt,