Update gen_pr.py

gen_pr.py

@@ -24,7 +24,7 @@ def load_config():
         "api_keys": {
             "gemini_api_key": api_key
         },
-        "api_base_url": api_base_url  # <--- ✅ 放在这里才对！
+        "api_base_url": api_base_url
     }
     
     if api_key:
@@ -93,9 +93,9 @@ def main():
             continue  # 只处理存在 auto 文件夹的目录
         
         # ✅ 如果 success.txt 已存在，跳过该目录
-        success_flag = os.path.join(auto_path, "success.txt")
+        success_flag = os.path.join(auto_path, "success_pr.txt")
         if os.path.isfile(success_flag):
-            print(f"✅ success.txt exists in {auto_path}, skipping...")
+            print(f"✅ success_pr.txt exists in {auto_path}, skipping...")
             continue
         
         print(f"\n🚀 Processing paper folder: {auto_path}")
@@ -259,7 +259,7 @@ def main():
 
         # =============================================================
         # 在auto目录下创建success.txt作为标志
-        success_file_path = os.path.join(auto_path, "success_dag.txt")
+        success_file_path = os.path.join(auto_path, "success_pr.txt")
         open(success_file_path, "w").close()