oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-17 04:23:42 +00:00
Code Issues Projects Releases Packages Wiki Activity

Make web_search not download the page contents, use fetch_webpage instead

Browse source
This commit is contained in:
oobabooga 2026-03-13 12:09:08 -07:00
parent d4c22ced83
commit 5362bbb413
3 changed files with 28 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

10
modules/ui_chat.py
View file

@ -96,6 +96,16 @@ def create_ui():
shared.gradio['tools_refresh'] = gr.Button('Refresh list', elem_id='tools-refresh-btn', visible=False)
shared.gradio['tools_refresh'].click(fn=lambda: gr.update(choices=get_available_tools()), inputs=[], outputs=[shared.gradio['selected_tools']])
def sync_web_tools(selected):
if 'web_search' in selected and 'fetch_webpage' not in selected:
selected.append('fetch_webpage')
elif 'web_search' not in selected and 'fetch_webpage' in selected:
selected.remove('fetch_webpage')
return gr.update(value=selected)
shared.gradio['selected_tools'].change(fn=sync_web_tools, inputs=[shared.gradio['selected_tools']], outputs=[shared.gradio['selected_tools']], show_progress=False)
gr.HTML("<div class='sidebar-vertical-separator'></div>")
with gr.Row():