尝试传入自定义base_url(实验性)

pull/1247/head
foxplaying 2025-09-21 20:56:35 +08:00 committed by GitHub
parent 0e0179fcd2
commit 74183ff079
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions

View File

@ -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,11 @@ class GeminiClient(BaseClient):
api_key=api_provider.api_key,
) # 这里和openai不一样gemini会自己决定自己是否需要retry
# 尝试传入自定义base_url(实验性必须为Gemini格式)
if hasattr(api_provider, "base_url") and api_provider.base_url:
self.base_url = api_provider.base_url
self.client._api_client._base_url = self.base_url
@staticmethod
def clamp_thinking_budget(tb: int, model_id: str) -> int:
"""