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-03-07 17:18:46 -06:00
parent dfa2511199
commit b04de83c20
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/middleware.py
+5 -4
View File
@@ -3093,8 +3093,7 @@ async def non_streaming_chat_response_handler(response, ctx):
) )
# Save message in the database # Save message in the database
raw_usage = response_data.get("usage", {}) or {} usage = normalize_usage(response_data.get("usage", {}) or {})
usage = normalize_usage(raw_usage) if raw_usage else None
Chats.upsert_message_to_chat_by_id_and_message_id( Chats.upsert_message_to_chat_by_id_and_message_id(
metadata["chat_id"], metadata["chat_id"],
@@ -4468,7 +4467,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}
@@ -4484,7 +4484,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 (