Update src/DAG2pr.py

src/DAG2pr.py

@@ -70,10 +70,15 @@ def extract_basic_information(
     is_gemini = "gemini" in model.lower()
 
     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)
         
         # Gemini Call
         resp = client.models.generate_content(
@@ -236,6 +241,11 @@ def generate_pr_from_dag(
     client = None
 
     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 Configuration (New SDK)
         if genai is None:
             raise ImportError("Google GenAI SDK not installed. Please run `pip install google-genai`.")
@@ -244,7 +254,7 @@ def generate_pr_from_dag(
         if not api_key:
             raise RuntimeError("API Key not found for Gemini.")
         
-        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:
         # OpenAI Configuration
@@ -679,6 +689,11 @@ def add_title_and_hashtag(pr_path: str, add_title_and_hashtag_prompt: str, model
 
     # >>> 分支判断逻辑 >>>
     if "gemini" in model.lower():
+        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
         # --- Google Gemini (New SDK) ---
         api_key = api_keys.get("gemini_api_key", "").strip()
         
@@ -692,7 +707,7 @@ def add_title_and_hashtag(pr_path: str, add_title_and_hashtag_prompt: str, model
             raise ImportError("google-genai package is required. Install with: pip install google-genai") from e
 
         # 配置 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)
 
         try:
             response = client.models.generate_content(