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, )