Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新gemini深度思考支持

Browse Source
This commit is contained in:
xunbu
2025-08-05 15:09:26 +08:00
parent 0891d4f0b2
commit 6c74404250
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docutranslate/agents/agent.py
View File

@@ -70,10 +70,14 @@ class Agent:
_think_factory = { _think_factory = {
"open.bigmodel.cn": ("thinking", {"type": "enabled"}, {"type": "disabled"}), "open.bigmodel.cn": ("thinking", {"type": "enabled"}, {"type": "disabled"}),
"dashscope.aliyuncs.com": ("enable_thinking ", True, False), "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): def __init__(self, config: AgentConfig):
self.baseurl = config.baseurl.strip() self.baseurl = config.baseurl.strip()
if self.baseurl.endswith("/"): if self.baseurl.endswith("/"):
self.baseurl = self.baseurl[:-1] self.baseurl = self.baseurl[:-1]
@@ -89,6 +93,7 @@ class Agent:
self.thinking = config.thinking self.thinking = config.thinking
self.logger = config.logger or global_logger self.logger = config.logger or global_logger
self.total_error_counter = TotalErrorCounter(logger=self.logger) self.total_error_counter = TotalErrorCounter(logger=self.logger)
def _add_thinking_mode(self, data: dict): def _add_thinking_mode(self, data: dict):
if self.domain not in self._think_factory: if self.domain not in self._think_factory:
return return