From 6c74404250dc6d0b81562373a1153db20faf8657 Mon Sep 17 00:00:00 2001 From: xunbu Date: Tue, 5 Aug 2025 15:09:26 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0gemini=E6=B7=B1=E5=BA=A6?= =?UTF-8?q?=E6=80=9D=E8=80=83=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docutranslate/agents/agent.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/docutranslate/agents/agent.py b/docutranslate/agents/agent.py index fb11615..69eddf3 100644 --- a/docutranslate/agents/agent.py +++ b/docutranslate/agents/agent.py @@ -70,10 +70,14 @@ class Agent: _think_factory = { "open.bigmodel.cn": ("thinking", {"type": "enabled"}, {"type": "disabled"}), "dashscope.aliyuncs.com": ("enable_thinking ", True, False), - "ark.cn-beijing.volces.com":("thinking", {"type": "enabled"}, {"type": "disabled"}) + "ark.cn-beijing.volces.com": ("thinking", {"type": "enabled"}, {"type": "disabled"}), + "https://generativelanguage.googleapis.com/v1beta/openai/": ("generationConfig", + {"thinkingConfig": {"thinkingBudget": -1}}, + {"thinkingConfig": {"thinkingBudget": 0}}) } def __init__(self, config: AgentConfig): + self.baseurl = config.baseurl.strip() if self.baseurl.endswith("/"): self.baseurl = self.baseurl[:-1] @@ -89,6 +93,7 @@ class Agent: self.thinking = config.thinking self.logger = config.logger or global_logger self.total_error_counter = TotalErrorCounter(logger=self.logger) + def _add_thinking_mode(self, data: dict): if self.domain not in self._think_factory: return