Update src/DAG2ppt.py

src/DAG2ppt.py

@@ -177,8 +177,14 @@ def outline_initialize(dag_json_path, outline_initialize_prompt, model, config):
         # --- Gemini Call (google-genai SDK) ---
         api_key = config['api_keys'].get('gemini_api_key')
 
+        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
+
         # 配置 Client        
-        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)
 
         # 构造 user 消息内容
         user_content = f"{outline_initialize_prompt}\n\nData Context:\n{first_node_text}"
@@ -312,7 +318,13 @@ def generate_complete_outline(
     if is_gemini:
         api_key = config['api_keys'].get('gemini_api_key')
 
-        client = genai.Client(api_key=api_key)
+        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
+
+        client = genai.Client(api_key=api_key, http_options={'base_url': base_url} if base_url else None)
     else:
         api_key = config['api_keys'].get('openai_api_key')
         client = OpenAI(api_key=api_key)
@@ -427,7 +439,13 @@ def arrange_template(
     if is_gemini:
         api_key = config['api_keys'].get('gemini_api_key')
 
-        client = genai.Client(api_key=api_key)
+        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
+
+        client = genai.Client(api_key=api_key, http_options={'base_url': base_url} if base_url else None)
     else:
         api_key = config['api_keys'].get('openai_api_key')
 
@@ -647,8 +665,14 @@ def generate_ppt(
 
     if is_gemini:
         api_key = config['api_keys'].get('gemini_api_key')
+        
+        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
 
-        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:
         api_key = config['api_keys'].get('openai_api_key')