BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

回退utils_model.py中的更改

pull/1112/head
UnCLAS-Prommer 2025-07-17 00:15:58 +08:00
parent 1aa2734d62
commit a83f8948e9
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/llm_models/utils_model.py
View File

@ -255,11 +255,12 @@ class LLMRequest:
if self.temp != 0.7:
payload["temperature"] = self.temp
# 添加enable_thinking参数（仅在启用时添加）
if self.enable_thinking:
payload["enable_thinking"] = True
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
# 添加enable_thinking参数（如果不是默认值False）
if not self.enable_thinking:
payload["enable_thinking"] = False
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
if self.max_tokens:
payload["max_tokens"] = self.max_tokens
@ -669,11 +670,12 @@ class LLMRequest:
if self.temp != 0.7:
payload["temperature"] = self.temp
# 添加enable_thinking参数（仅在启用时添加）
if self.enable_thinking:
payload["enable_thinking"] = True
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
# 添加enable_thinking参数（如果不是默认值False）
if not self.enable_thinking:
payload["enable_thinking"] = False
if self.thinking_budget != 4096:
payload["thinking_budget"] = self.thinking_budget
if self.max_tokens:
payload["max_tokens"] = self.max_tokens