From be989f3645f2994af95a1ae9e58993da067ddde4 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Fri, 30 May 2025 00:12:28 +0400
Subject: [PATCH] refac: better memory error handling

---
 backend/open_webui/utils/middleware.py | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index ab5301a6f..7b5659d51 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -299,13 +299,20 @@ async def chat_completion_tools_handler(
 async def chat_memory_handler(
     request: Request, form_data: dict, extra_params: dict, user
 ):
-    results = await query_memory(
-        request,
-        QueryMemoryForm(
-            **{"content": get_last_user_message(form_data["messages"]) or "", "k": 3}
-        ),
-        user,
-    )
+    try:
+        results = await query_memory(
+            request,
+            QueryMemoryForm(
+                **{
+                    "content": get_last_user_message(form_data["messages"]) or "",
+                    "k": 3,
+                }
+            ),
+            user,
+        )
+    except Exception as e:
+        log.debug(e)
+        results = None
 
     user_context = ""
     if results and hasattr(results, "documents"):