From 16bae8f33a1994798e17681ffee10860e1b561fd Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 10 Feb 2025 13:53:16 -0800 Subject: [PATCH] refac --- backend/open_webui/config.py | 3 +-- backend/open_webui/utils/middleware.py | 9 +++++++++ 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py index 6993b2801..b6d8fa274 100644 --- a/backend/open_webui/config.py +++ b/backend/open_webui/config.py @@ -1378,9 +1378,8 @@ DEFAULT_CODE_INTERPRETER_PROMPT = """ - When coding, **always aim to print meaningful outputs** (e.g., results, tables, summaries, or visuals) to better interpret and verify the findings. Avoid relying on implicit outputs; prioritize explicit and clear print statements so the results are effectively communicated to the user. - After obtaining the printed output, **always provide a concise analysis, interpretation, or next steps to help the user understand the findings or refine the outcome further.** - If the results are unclear, unexpected, or require validation, refine the code and execute it again as needed. Always aim to deliver meaningful insights from the results, iterating if necessary. - - If a link is provided for an image, audio, or any file, include it in the response exactly as given to ensure the user has access to the original resource. + - **If a link to an image, audio, or any file is provided in markdown format in the output, ALWAYS regurgitate word for word, explicitly display it as part of the response to ensure the user can access it easily, do NOT change the link.** - All responses should be communicated in the chat's primary language, ensuring seamless understanding. If the chat is multilingual, default to English for clarity. - - **If a link to an image, audio, or any file is provided in markdown format, ALWAYS regurgitate explicitly display it as part of the response to ensure the user can access it easily, do NOT change the link.** Ensure that the tools are effectively utilized to achieve the highest-quality analysis for the user.""" diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index f855e6434..dddf5235f 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -1645,6 +1645,15 @@ async def process_chat_response( content_blocks[-1]["type"] == "code_interpreter" and retries < MAX_RETRIES ): + await event_emitter( + { + "type": "chat:completion", + "data": { + "content": serialize_content_blocks(content_blocks), + }, + } + ) + retries += 1 log.debug(f"Attempt count: {retries}")