From 058369adeaf6873801d89c6a30ca16da2a320e06 Mon Sep 17 00:00:00 2001 From: Sihyeon Jang Date: Tue, 15 Jul 2025 10:14:30 +0900 Subject: [PATCH] fix: improve cleanup_response positioning for better resource management Signed-off-by: Sihyeon Jang --- backend/open_webui/routers/ollama.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/routers/ollama.py b/backend/open_webui/routers/ollama.py index 000f4b48d2..7fa7aef249 100644 --- a/backend/open_webui/routers/ollama.py +++ b/backend/open_webui/routers/ollama.py @@ -184,7 +184,6 @@ async def send_post_request( ) else: res = await r.json() - await cleanup_response(r, session) return res except HTTPException as e: @@ -196,6 +195,9 @@ async def send_post_request( status_code=r.status if r else 500, detail=detail if e else "Open WebUI: Server Connection Error", ) + finally: + if not stream: + await cleanup_response(r, session) def get_api_key(idx, url, configs):