Update src/DAG2poster.py

src/DAG2poster.py

@@ -157,10 +157,15 @@ def generate_poster_outline_txt(
         api_keys_config = config.get("api_keys", {}) if config else {}
         
         if is_gemini:
+            raw_url = config.get('api_base_url', '').strip().rstrip("/")
+            if raw_url.endswith("/v1"):
+                base_url = raw_url[:-3].rstrip("/") # 去掉最后的 /v1
+            else:
+                base_url = raw_url
             # Setup Google GenAI Client
             api_key = api_key or api_keys_config.get("gemini_api_key") or os.getenv("GOOGLE_API_KEY")
                       
-            client = genai.Client(api_key=api_key)
+            client = genai.Client(api_key=api_key, http_options={'base_url': base_url} if base_url else None)
         else:
             # Setup OpenAI Client
             api_key = api_key or api_keys_config.get("openai_api_key") or os.getenv("OPENAI_API_KEY")
@@ -781,10 +786,15 @@ def modified_poster_logic(
     out_text = ""
 
     if is_gemini:
+        raw_url = config.get('api_base_url', '').strip().rstrip("/")
+        if raw_url.endswith("/v1"):
+            base_url = raw_url[:-3].rstrip("/") # 去掉最后的 /v1
+        else:
+            base_url = raw_url
         # --- Gemini Client Setup ---
         api_key = api_keys_config.get("gemini_api_key") or os.getenv("GOOGLE_API_KEY")
         
-        client = genai.Client(api_key=api_key)
+        client = genai.Client(api_key=api_key, http_options={'base_url': base_url} if base_url else None)
         
         # Call Gemini
         # 将 system prompt 放入配置中，user content 放入 contents