oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-03-18 03:14:39 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add tools refresh button and _tool_turn comment

Browse source
This commit is contained in:
oobabooga 2026-03-12 01:36:07 -03:00
parent cf9ad8eafe
commit 0d62038710
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/chat.py
View file

@ -1151,7 +1151,9 @@ def generate_chat_reply_wrapper(text, state, regenerate=False, _continue=False):
for _tool_turn in range(max_tool_turns):
history = state['history']
# Turn 0: use original flags; turns 2+: regenerate into the same entry
# Turn 0: use original flags; turns 2+: regenerate into the same entry.
# _tool_turn tells chatbot_wrapper to skip version creation/sync so
# that intermediate tool-loop regenerations don't pollute swipe history.
if _tool_turn > 0:
state['_tool_turn'] = True

1
modules/ui_chat.py
View file

@ -93,6 +93,7 @@ def create_ui():
from modules.tool_use import get_available_tools
shared.gradio['selected_tools'] = gr.CheckboxGroup(choices=get_available_tools(), value=[], label='Tools', info='Functions the model can call during generation.')
ui.create_refresh_button(shared.gradio['selected_tools'], lambda: None, lambda: {'choices': get_available_tools()}, 'refresh-button')
gr.HTML("<div class='sidebar-vertical-separator'></div>")