ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add enable thinking

Browse Source
This commit is contained in:
tom
2025-06-23 16:45:05 +08:00
parent 0edf564173
commit aa83574c99
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/blackbox/chat.py
View File

@ -63,6 +63,7 @@ class Chat(Blackbox):
user_prompt_template = settings.get('user_prompt_template') user_prompt_template = settings.get('user_prompt_template')
user_stream = settings.get('stream') user_stream = settings.get('stream')
user_websearch = settings.get('websearch') user_websearch = settings.get('websearch')
user_thinking = settings.get('thinking', False)
llm_model = "llm" llm_model = "llm"
@ -249,6 +250,7 @@ class Chat(Blackbox):
"presence_penalty": str(user_presence_penalty), "presence_penalty": str(user_presence_penalty),
"stop": str(user_stop), "stop": str(user_stop),
"stream": user_stream, "stream": user_stream,
"chat_template_kwargs": {"enable_thinking": user_thinking},
} }
else: else:
chat_inputs={ chat_inputs={
@ -267,6 +269,7 @@ class Chat(Blackbox):
"presence_penalty":float( user_presence_penalty), "presence_penalty":float( user_presence_penalty),
# "stop": user_stop, # "stop": user_stop,
"stream": user_stream, "stream": user_stream,
"chat_template_kwargs": {"enable_thinking": user_thinking},
} }
# # 获取当前时间戳 # # 获取当前时间戳