++

main.py

@@ -28,7 +28,7 @@ RANDOM_DATA_SETS_COUNT = 0
 
 # 数据集筛选之后大于此数值, 则随机抽取x条数据
 MAX_DATA_COUNT = 800
-RANDOM_DATA_COUNT = 400
+RANDOM_DATA_COUNT = 500
 
 # 数据库搜索字段
 REGION = 'USA'
@@ -39,7 +39,11 @@ UNIVERSE = 'TOP3000'
 LOAD_OPERATOR = 1
 
 # 数据类别名
-CATEGORY_NAME_LIST = ['Analyst', 'Fundamental', 'Price Volume']
+CATEGORY_NAME_LIST = [
+        'Analyst',
+        'Fundamental',
+        'Price Volume'
+      ]
 
 # 基础提示词文件
 alpha_prompt_path = os.path.join(PREPARE_PROMPT, "alpha_prompt.txt")
@@ -47,16 +51,14 @@ alpha_prompt_path = os.path.join(PREPARE_PROMPT, "alpha_prompt.txt")
 SILICONFLOW_API_KEY = "sk-pvdiisdowmuwkrpnxsrlhxaovicqibmlljwrwwvbbdjaitdl"
 SILICONFLOW_BASE_URL = "https://api.siliconflow.cn/v1"
 MODELS = [
-    # 'Pro/deepseek-ai/DeepSeek-V3.1-Terminus',
-    'deepseek-ai/DeepSeek-V3.2-Exp',
+    'Pro/deepseek-ai/DeepSeek-V3.1-Terminus',
+    # 'deepseek-ai/DeepSeek-V3.2-Exp',
     # 'Qwen/Qwen3-VL-235B-A22B-Instruct',
-    'Pro/moonshotai/Kimi-K2-Thinking',
-    # 'MiniMaxAI/MiniMax-M2',
+    # 'Pro/moonshotai/Kimi-K2-Thinking',
+    'MiniMaxAI/MiniMax-M2',
     # 'zai-org/GLM-4.6',
     # 'inclusionAI/Ring-flash-2.0',
-    # 'zai-org/GLM-4.6',
     # 'inclusionAI/Ling-flash-2.0',
-    # 'inclusionAI/Ring-flash-2.0',
 ]
 
 def process_text(text):