github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-05-03 18:59:38 -05:00
Code Issues 915 Packages Projects Releases 126 Wiki Activity

refac

Browse Source
This commit is contained in:
Tim Baek
2026-01-07 10:21:05 -05:00
parent ab400e3eae
commit 35d385e9cc
3 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/utils/middleware.py
View File

@@ -159,7 +159,7 @@ def get_citation_source_from_tool_result(
Returns a list of sources (usually one, but query_knowledge_bases may return multiple).
"""
try:
if tool_name == "web_search":
if tool_name == "search_web":
# Parse JSON array: [{"title": "...", "link": "...", "snippet": "..."}]
results = json.loads(tool_result)
documents = []
@@ -178,7 +178,7 @@ def get_citation_source_from_tool_result(
})
return [{
"source": {"name": "web_search", "id": "web_search"},
"source": {"name": "search_web", "id": "search_web"},
"document": documents,
"metadata": metadata,
}]
@@ -3233,7 +3233,7 @@ async def process_chat_response(
)
# Extract citation sources from tool results
if tool_function_name in ["web_search", "view_knowledge_file", "query_knowledge_bases"] and tool_result:
if tool_function_name in ["search_web", "view_knowledge_file", "query_knowledge_bases"] and tool_result:
try:
citation_sources = get_citation_source_from_tool_result(
tool_name=tool_function_name,