diff --git a/backend/open_webui/utils/code_interpreter.py b/backend/open_webui/utils/code_interpreter.py index cd6ce91ada..9a2cc45fc9 100644 --- a/backend/open_webui/utils/code_interpreter.py +++ b/backend/open_webui/utils/code_interpreter.py @@ -7,6 +7,7 @@ from typing import Optional import aiohttp import websockets from open_webui.env import AIOHTTP_CLIENT_ALLOW_REDIRECTS +from open_webui.utils.json_codec import JSONCodec from pydantic import BaseModel logger = logging.getLogger(__name__) @@ -157,7 +158,7 @@ class JupyterCodeExecuter: try: # wait for message message = await asyncio.wait_for(ws.recv(), self.timeout) - message_data = json.loads(message) + message_data = JSONCodec.loads(message) # msg id not match, skip if message_data.get('parent_header', {}).get('msg_id') != msg_id: continue