update

AlphaGenerator.py

@@ -35,6 +35,14 @@ def read_prompt():
         return None
     return prompt
 
+def read_operator():
+    with open("wqb_operator.txt", 'r', encoding='utf-8') as f:
+        operator = f.read().strip()
+    if not operator:
+        print("operator.txt是空的，请填入操作符")
+        return None
+    return operator
+
 
 def create_result_folder():
     # 修改这里：创建 generated_alpha 文件夹
@@ -96,7 +104,7 @@ def get_ai_config(config):
 
 def get_user_info():
     # 获取用户信息
-    with open('config.json', 'r') as f:
+    with open('ai_config.json', 'r') as f:
         config = json.load(f)
         siliconflow = config['siliconflow']
         token = siliconflow['api_keys']
@@ -130,6 +138,11 @@ def main():
     if not config_result:
         return
 
+    operator = read_operator()
+
+    if operator:
+        prompt = prompt + "\n\n以下是我的账号有权限使用的操作符以及操作符的使用方法, 请严格按照操作符, 进行生成,组合因子\n\n" + operator
+
     service_name, service_config = config_result
 
     print(f"使用服务: {service_name}")