From ffbd6ec7f2effad1e33f7a92f0a02606b0fd43fd Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Fri, 9 Jan 2026 03:03:25 +0400
Subject: [PATCH] refac

---
 backend/open_webui/tools/builtin.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/backend/open_webui/tools/builtin.py b/backend/open_webui/tools/builtin.py
index f477a500ee..2f57986184 100644
--- a/backend/open_webui/tools/builtin.py
+++ b/backend/open_webui/tools/builtin.py
@@ -346,6 +346,7 @@ async def edit_image(
 
 async def search_memories(
     query: str,
+    count: int = 5,
     __request__: Request = None,
     __user__: dict = None,
 ) -> str:
@@ -353,6 +354,7 @@ async def search_memories(
     Search the user's stored memories for relevant information.
 
     :param query: The search query to find relevant memories
+    :param count: Number of memories to return (default 5)
     :return: JSON with matching memories and their dates
     """
     if __request__ is None:
@@ -363,7 +365,7 @@ async def search_memories(
 
         results = await query_memory(
             __request__,
-            QueryMemoryForm(content=query, k=5),
+            QueryMemoryForm(content=query, k=count),
             user,
         )