++

main.py

@@ -152,22 +152,21 @@ def manual_prompt(prompt):
     print(f"手动提示词保存到: {filepath}")
 
 
-def call_ai(prompt):
+def call_ai(prompt, model):
     balance = get_user_info()
 
     folder = create_result_folder()
 
-    for model in MODELS:
-        print(f"正在调用AI...{model}")
-        result = call_siliconflow(prompt, model)
+    print(f"正在调用AI...{model}")
+    result = call_siliconflow(prompt, model)
 
-        if result:
-            print(f"AI回复: {result[:200]}...")
-            save_result(result, folder)
-            used_balance = balance - get_user_info()
-            print(f'本次调用 api 使用额度 {used_balance}')
-        else:
-            print("AI调用失败")
+    if result:
+        print(f"AI回复: {result[:200]}...")
+        save_result(result, folder)
+        used_balance = balance - get_user_info()
+        print(f'本次调用 api 使用额度 {used_balance}')
+    else:
+        print("AI调用失败")
 
 
 def prepare_prompt():
@@ -220,13 +219,14 @@ def prepare_prompt():
 
 
 def main():
-    prompt = prepare_prompt()
+    for model in MODELS:
+        prompt = prepare_prompt()
 
-    # # 如果需要手动在页面段模型, 使用提示词, 打开这个, 将生成的提示词存到本地
-    manual_prompt(prompt)
+        # # 如果需要手动在页面段模型, 使用提示词, 打开这个, 将生成的提示词存到本地
+        manual_prompt(prompt)
 
-    # # 如果需要使用模型, 打开这个
-    call_ai(prompt)
+        # # 如果需要使用模型, 打开这个
+        call_ai(prompt, model)
 
 
 if __name__ == "__main__":