Update gen_dag.py

gen_dag.py

@@ -6,23 +6,14 @@ from src import *
 
 # ============= 从环境变量动态生成配置 ===============
 def load_config():
-    """
-    不再读取 config.yaml，而是从 Gradio 主程序注入的环境变量中获取专属参数，
-    并组装成原有代码期望的字典结构，实现多用户隔离。
-    """
     base_dir = os.path.dirname(os.path.abspath(__file__))
-    
-    # 获取由 app.py 注入的专属输出文件夹，如果没有则回退到本地默认文件夹（方便本地单独运行调试）
     user_root_folder = os.environ.get("USER_OUTPUT_DIR", os.path.join(base_dir, "mineru_outputs"))
     
-    # 获取由 app.py 注入的 API 凭证
     api_key = os.environ.get("GEMINI_API_KEY", "")
     api_base_url = os.environ.get("GEMINI_API_BASE_URL", "")
-    
-    # 可以在这里设置默认使用的模型，或者也改用环境变量传入
     generation_model = os.environ.get("GENERATION_MODEL", "gemini-3-pro-preview")
     
-    # 拼装成原版 config.yaml 的字典结构，保证后续代码兼容
+    # ✅ 修正字典结构：把 api_base_url 移回到外层，和原版 config.yaml 保持绝对一致
     config = {
         "model_settings": {
             "generation_model": generation_model
@@ -31,16 +22,17 @@ def load_config():
             "root_folder": user_root_folder
         },
         "api_keys": {
-            "gemini_api_key": api_key,
-            "api_base_url": api_base_url
-        }
+            "gemini_api_key": api_key
+        },
+        "api_base_url": api_base_url  # <--- ✅ 放在这里才对！
     }
     
-    # 如果你的底层 src 代码库也需要直接用到 os.environ，我们在这里显式注入一下
     if api_key:
         os.environ["GEMINI_API_KEY"] = api_key
+        os.environ["GOOGLE_API_KEY"] = api_key
     if api_base_url:
         os.environ["GEMINI_API_BASE_URL"] = api_base_url
+        os.environ["GOOGLE_API_BASE"] = api_base_url
         
     return config