oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-08 08:03:39 +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

14
user_data/tools/web_search.py
View file

@ -1,16 +1,15 @@
from modules.web_search import perform_web_search, truncate_content_by_tokens
from modules.web_search import perform_web_search
tool = {
"type": "function",
"function": {
"name": "web_search",
"description": "Search the web using DuckDuckGo and return page contents.",
"description": "Search the web using DuckDuckGo and return a list of result titles and URLs. Use fetch_webpage to read the contents of a specific result.",
"parameters": {
"type": "object",
"properties": {
"query": {"type": "string", "description": "The search query."},
"num_pages": {"type": "integer", "description": "Number of search result pages to fetch (default: 3)."},
"max_tokens": {"type": "integer", "description": "Maximum number of tokens per page result (default: 2048)."},
"num_pages": {"type": "integer", "description": "Number of search results to return (default: 3)."},
},
"required": ["query"]
}
@ -21,11 +20,10 @@ tool = {
def execute(arguments):
query = arguments.get("query", "")
num_pages = arguments.get("num_pages", 3)
max_tokens = arguments.get("max_tokens", 2048)
results = perform_web_search(query, num_pages=num_pages)
results = perform_web_search(query, num_pages=num_pages, fetch_content=False)
output = []
for r in results:
if r and r["content"].strip():
output.append({"title": r["title"], "url": r["url"], "content": truncate_content_by_tokens(r["content"], max_tokens=max_tokens)})
if r:
output.append({"title": r["title"], "url": r["url"]})
return output if output else [{"error": "No results found."}]