simcop2387/text-generation-webui-mirror
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-06-08 14:46:14 -04:00
Code Issues Projects Releases Packages Wiki Activity

Make web search attachments clickable

Browse source
This commit is contained in:
oobabooga 2025-05-28 05:28:15 -07:00
parent 0aedb89921
commit 6c3590ba9a
2 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
modules/html_generator.py
View file

@ -370,10 +370,16 @@ def format_message_attachments(history, role, index):
attachments_html = '<div class="message-attachments">' attachments_html = '<div class="message-attachments">'
for attachment in attachments: for attachment in attachments:
name = html.escape(attachment["name"])
# Make clickable if URL exists
if "url" in attachment:
name = f'<a href="{html.escape(attachment["url"])}" target="_blank" rel="noopener noreferrer">{name}</a>'
attachments_html += ( attachments_html += (
f'<div class="attachment-box">' f'<div class="attachment-box">'
f'<div class="attachment-icon">{attachment_svg}</div>' f'<div class="attachment-icon">{attachment_svg}</div>'
f'<div class="attachment-name">{html.escape(attachment["name"])}</div>' f'<div class="attachment-name">{name}</div>'
f'</div>' f'</div>'
) )
attachments_html += '</div>' attachments_html += '</div>'

5
modules/web_search.py
View file

@ -113,9 +113,10 @@ def add_web_search_attachments(history, row_idx, user_message, state):
for result in search_results: for result in search_results:
attachment = { attachment = {
"name": f"{result['title']}", "name": result['title'],
"type": "text/html", "type": "text/html",
"content": f"URL: {result['url']}\n\n{result['content']}" "url": result['url'],
"content": result['content']
} }
history['metadata'][key]["attachments"].append(attachment) history['metadata'][key]["attachments"].append(attachment)