Update qwen_gen_data_NoBash.py 添加对qwen-max的异常处理

2024-03-17 08:53:50 +09:00 · 2024-03-17 08:53:50 +09:00 · 24a16455ab
commit 24a16455ab
parent 025061b0ab
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,7 +23,14 @@ def qwen_api(data, emo):
    病人：病人的咨询或陈述 
    医生：医生的安抚和建议
    '''
-    response = dashscope.Generation.call(
+    try:
+        response = dashscope.Generation.call(
+        model='qwen-max',
+        prompt=prompt,
+        history=[],
+    )
+    except:
+        response = dashscope.Generation.call(
        model='qwen-max',
        prompt=prompt,
        history=[],