OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-23 02:16:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: better memory error handling

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-30 00:12:28 +04:00
parent 4c45d67677
commit be989f3645
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -299,13 +299,20 @@ async def chat_completion_tools_handler(
async def chat_memory_handler( async def chat_memory_handler(
request: Request, form_data: dict, extra_params: dict, user request: Request, form_data: dict, extra_params: dict, user
): ):
results = await query_memory( try:
request, results = await query_memory(
QueryMemoryForm( request,
**{"content": get_last_user_message(form_data["messages"]) or "", "k": 3} QueryMemoryForm(
), **{
user, "content": get_last_user_message(form_data["messages"]) or "",
) "k": 3,
}
),
user,
)
except Exception as e:
log.debug(e)
results = None
user_context = "" user_context = ""
if results and hasattr(results, "documents"): if results and hasattr(results, "documents"):