mirror of https://github.com/Mai-with-u/MaiBot.git
commit
f075f28f02
|
|
@ -349,6 +349,18 @@ 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:
|
||||
base_url = api_provider.base_url.rstrip("/") # 去掉末尾 /
|
||||
self.client._api_client._http_options.base_url = base_url
|
||||
|
||||
# 如果 base_url 已经带了 /v1 或 /v1beta,就清掉 SDK 的 api_version
|
||||
if base_url.endswith("/v1") or base_url.endswith("/v1beta"):
|
||||
self.client._api_client._http_options.api_version = None
|
||||
|
||||
# 让 GeminiClient 内部也能访问底层 api_client
|
||||
self._api_client = self.client._api_client
|
||||
|
||||
@staticmethod
|
||||
def clamp_thinking_budget(tb: int, model_id: str) -> int:
|
||||
"""
|
||||
|
|
|
|||
|
|
@ -23,7 +23,7 @@ retry_interval = 5
|
|||
|
||||
[[api_providers]] # 特殊:Google的Gimini使用特殊API,与OpenAI格式不兼容,需要配置client为"gemini"
|
||||
name = "Google"
|
||||
base_url = "https://api.google.com/v1"
|
||||
base_url = "https://generativelanguage.googleapis.com/v1beta"
|
||||
api_key = "your-google-api-key-1"
|
||||
client_type = "gemini"
|
||||
max_retry = 2
|
||||
|
|
|
|||
Loading…
Reference in New Issue