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

Truncate web_search and fetch_webpage tools to 8192 tokens

Browse source
This commit is contained in:
oobabooga 2026-03-12 17:10:41 -07:00
parent 5c02b7f603
commit f8936ec47c
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
user_data/tools/web_search.py
View file

@ -1,4 +1,4 @@
from modules.web_search import perform_web_search
from modules.web_search import perform_web_search, truncate_content_by_tokens
tool = {
"type": "function",
@ -22,6 +22,6 @@ def execute(arguments):
output = []
for r in results:
if r and r["content"].strip():
output.append({"title": r["title"], "url": r["url"], "content": r["content"][:4000]})
output.append({"title": r["title"], "url": r["url"], "content": truncate_content_by_tokens(r["content"])})
return output if output else [{"error": "No results found."}]