Update gen_dag.py

gen_dag.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_dag.txt")
         if os.path.isfile(success_flag):
-            print(f"✅ success.txt exists in {auto_path}, skipping...")
+            print(f"✅ success_dag.txt exists in {auto_path}, skipping...")
             continue
         
         print(f"\n🚀 Processing paper folder: {auto_path}")
@@ -122,7 +122,6 @@ def main():
         graph_json_path = os.path.join(auto_path, "graph.json")
 
         # === 清理 markdown ===                                                   去除无意义的段落，如relative work，reference，appendix等等
-        print(config)
         print("🧹 Cleaning markdown before splitting...")
         cleaned_md_path = clean_paper(md_path, clean_prompt, model="gemini-3-pro-preview", config=config)