Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 19:37:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac: code interpreter

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-02-13 00:40:04 -08:00
parent 6d899b80d0
commit 6acda2e6ec
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1704,10 +1704,12 @@ async def process_chat_response(
"stdout": "Code interpreter engine not configured." "stdout": "Code interpreter engine not configured."
} }
log.debug(f"Code interpreter output: {output}")
if isinstance(output, dict): if isinstance(output, dict):
stdout = output.get("stdout", "") stdout = output.get("stdout", "")
if stdout: if isinstance(stdout, str):
stdoutLines = stdout.split("\n") stdoutLines = stdout.split("\n")
for idx, line in enumerate(stdoutLines): for idx, line in enumerate(stdoutLines):
if "data:image/png;base64" in line: if "data:image/png;base64" in line:
@@ -1739,7 +1741,7 @@ async def process_chat_response(
result = output.get("result", "") result = output.get("result", "")
if result: if isinstance(result, str):
resultLines = result.split("\n") resultLines = result.split("\n")
for idx, line in enumerate(resultLines): for idx, line in enumerate(resultLines):
if "data:image/png;base64" in line: if "data:image/png;base64" in line:
@@ -1789,6 +1791,8 @@ async def process_chat_response(
} }
) )
print(content_blocks, serialize_content_blocks(content_blocks))
try: try:
res = await generate_chat_completion( res = await generate_chat_completion(
request, request,