++

main.py

@@ -7,6 +7,7 @@ import httpx
 import csv
 from datetime import datetime
 import jieba
+import time
 
 sys.path.append(os.path.join(os.path.abspath(__file__).split('AlphaGenerator')[0] + 'AlphaGenerator'))
 PROJECT_PATH = os.path.join(os.path.abspath(__file__).split('AlphaGenerator')[0] + 'AlphaGenerator')
@@ -16,7 +17,7 @@ KEYS_TEXT = os.path.join(PREPARE_PROMPT, 'keys_text.txt')
 
 TEMPERATURE = 0.1
 
-USE_AI = 1
+USE_AI = 0
 
 SILICONFLOW_API_KEY = "sk-pvdiisdowmuwkrpnxsrlhxaovicqibmlljwrwwvbbdjaitdl"
 SILICONFLOW_BASE_URL = "https://api.siliconflow.cn/v1"
@@ -304,11 +305,17 @@ def main():
     # # 如果需要手动在页面段模型, 使用提示词, 打开这个, 将生成的提示词存到本地
     manual_prompt(prompt)
 
+    if len(result_data_sets) <= 10:
+        print(f'从数据集中提取了 {len(result_data_sets)} 条数据, 数量太少, 程序退出')
+        exit(1)
+
     if USE_AI:
         for model in MODELS:
             # 如果需要使用模型, 打开这个
             call_ai(prompt, model)
 
+            time.sleep(5)
+
 
 if __name__ == "__main__":
     main()