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

Add num_pages and max_tokens kwargs to web search tools

Browse source
This commit is contained in:
oobabooga 2026-03-12 22:17:23 -03:00
parent a09f21b9de
commit 4c7a56c18d
2 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
user_data/tools/fetch_webpage.py
View file

@ -9,6 +9,7 @@ tool = {
"type": "object",
"properties": {
"url": {"type": "string", "description": "The URL of the web page to fetch."},
"max_tokens": {"type": "integer", "description": "Maximum number of tokens in the returned content (default: 2048)."},
},
"required": ["url"]
}
@ -18,6 +19,7 @@ tool = {
def execute(arguments):
url = arguments.get("url", "")
max_tokens = arguments.get("max_tokens", 2048)
if not url:
return {"error": "No URL provided."}
@ -25,4 +27,4 @@ def execute(arguments):
if not content or not content.strip():
return {"error": f"Failed to fetch content from {url}"}
return {"url": url, "content": truncate_content_by_tokens(content)}
return {"url": url, "content": truncate_content_by_tokens(content, max_tokens=max_tokens)}