From b7a6042dd35d7619ce351c436f31bacb96d99040 Mon Sep 17 00:00:00 2001
From: foxplaying <166147707+foxplaying@users.noreply.github.com>
Date: Sun, 21 Sep 2025 20:39:12 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=9D=E8=AF=95=E4=BC=A0=E5=85=A5=E8=87=AA?=
 =?UTF-8?q?=E5=AE=9A=E4=B9=89base=5Furl(=E5=AE=9E=E9=AA=8C=E6=80=A7)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/llm_models/model_client/gemini_client.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/llm_models/model_client/gemini_client.py b/src/llm_models/model_client/gemini_client.py
index e58466d1..1023910b 100644
--- a/src/llm_models/model_client/gemini_client.py
+++ b/src/llm_models/model_client/gemini_client.py
@@ -4,6 +4,7 @@ import base64
 from typing import Callable, AsyncIterator, Optional, Coroutine, Any, List
 
 from google import genai
+from google.genai import _base_url as genai_base_url
 from google.genai.types import (
     Content,
     Part,
@@ -349,6 +350,12 @@ class GeminiClient(BaseClient):
             api_key=api_provider.api_key,
         )  # 这里和openai不一样，gemini会自己决定自己是否需要retry
 
+        if hasattr(api_provider, "base_url") and api_provider.base_url:
+            genai_base_url.set_default_base_urls(
+                gemini_url=api_provider.base_url,
+                vertex_url=None  # 不修改 Vertex URL
+            ) # 尝试传入自定义base_url(实验性，必须为gemini格式)
+    
     @staticmethod
     def clamp_thinking_budget(tb: int, model_id: str) -> int:
         """