Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-12 15:57:27 -06:00
parent 8919d8a82a
commit 3238d94a0e
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/middleware.py
+14 -8
View File
@@ -171,9 +171,13 @@ def get_citation_source_from_tool_result(
Returns a list of sources (usually one, but query_knowledge_files may return multiple). Returns a list of sources (usually one, but query_knowledge_files may return multiple).
""" """
try: try:
tool_result = json.loads(tool_result)
if isinstance(tool_result, dict) and "error" in tool_result:
return []
if tool_name == "search_web": if tool_name == "search_web":
# Parse JSON array: [{"title": "...", "link": "...", "snippet": "..."}] # Parse JSON array: [{"title": "...", "link": "...", "snippet": "..."}]
results = json.loads(tool_result) results = tool_result
documents = [] documents = []
metadata = [] metadata = []
@@ -200,7 +204,7 @@ def get_citation_source_from_tool_result(
] ]
elif tool_name == "view_knowledge_file": elif tool_name == "view_knowledge_file":
file_data = json.loads(tool_result) file_data = tool_result
filename = file_data.get("filename", "Unknown File") filename = file_data.get("filename", "Unknown File")
file_id = file_data.get("id", "") file_id = file_data.get("id", "")
knowledge_name = file_data.get("knowledge_name", "") knowledge_name = file_data.get("knowledge_name", "")
@@ -229,7 +233,7 @@ def get_citation_source_from_tool_result(
] ]
elif tool_name == "query_knowledge_files": elif tool_name == "query_knowledge_files":
chunks = json.loads(tool_result) chunks = tool_result
# Group chunks by source for better citation display # Group chunks by source for better citation display
# Each unique source becomes a separate source entry # Each unique source becomes a separate source entry
@@ -2249,11 +2253,11 @@ async def process_chat_payload(request, form_data, user, metadata, model):
headers = include_user_info_headers(headers, user) headers = include_user_info_headers(headers, user)
if metadata and metadata.get("chat_id"): if metadata and metadata.get("chat_id"):
headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = metadata.get( headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = metadata.get(
"chat_id" "chat_id"
) )
if metadata and metadata.get("message_id"): if metadata and metadata.get("message_id"):
headers[FORWARD_SESSION_INFO_HEADER_MESSAGE_ID] = metadata.get( headers[FORWARD_SESSION_INFO_HEADER_MESSAGE_ID] = (
"message_id" metadata.get("message_id")
) )
mcp_clients[server_id] = MCPClient() mcp_clients[server_id] = MCPClient()
@@ -4121,7 +4125,8 @@ async def streaming_chat_response_handler(response, ctx):
code = sanitize_code(code) code = sanitize_code(code)
if CODE_INTERPRETER_BLOCKED_MODULES: if CODE_INTERPRETER_BLOCKED_MODULES:
blocking_code = textwrap.dedent(f""" blocking_code = textwrap.dedent(
f"""
import builtins import builtins
BLOCKED_MODULES = {CODE_INTERPRETER_BLOCKED_MODULES} BLOCKED_MODULES = {CODE_INTERPRETER_BLOCKED_MODULES}
@@ -4137,7 +4142,8 @@ async def streaming_chat_response_handler(response, ctx):
return _real_import(name, globals, locals, fromlist, level) return _real_import(name, globals, locals, fromlist, level)
builtins.__import__ = restricted_import builtins.__import__ = restricted_import
""") """
)
code = blocking_code + "\n" + code code = blocking_code + "\n" + code
if ( if (