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,
|
api_key=api_provider.api_key,
|
||||||
) # 这里和openai不一样,gemini会自己决定自己是否需要retry
|
) # 这里和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
|
@staticmethod
|
||||||
def clamp_thinking_budget(tb: int, model_id: str) -> int:
|
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"
|
[[api_providers]] # 特殊:Google的Gimini使用特殊API,与OpenAI格式不兼容,需要配置client为"gemini"
|
||||||
name = "Google"
|
name = "Google"
|
||||||
base_url = "https://api.google.com/v1"
|
base_url = "https://generativelanguage.googleapis.com/v1beta"
|
||||||
api_key = "your-google-api-key-1"
|
api_key = "your-google-api-key-1"
|
||||||
client_type = "gemini"
|
client_type = "gemini"
|
||||||
max_retry = 2
|
max_retry = 2
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue