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

This commit is contained in:
foxplaying
2025-09-21 20:56:35 +08:00
committed by GitHub
parent 0e0179fcd2
commit 74183ff079

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:
"""