Skip to content

Commit cefeefb

Browse files
author
yuan.wang
committed
modify code
1 parent 27b4fc4 commit cefeefb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/memos/api/handlers/chat_handler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,12 +423,13 @@ def generate_chat_response() -> Generator[str, None, None]:
423423
readable_cube_ids=readable_cube_ids,
424424
mode="fast",
425425
internet_search=False,
426-
top_k=chat_req.top_k,
426+
top_k=5,
427427
chat_history=chat_req.history,
428428
session_id=chat_req.session_id,
429429
include_preference=chat_req.include_preference,
430430
pref_top_k=chat_req.pref_top_k,
431431
filter=chat_req.filter,
432+
search_tool_memory=False,
432433
playground_search_goal_parser=False,
433434
)
434435
start_time = time.time()
@@ -498,6 +499,7 @@ def generate_chat_response() -> Generator[str, None, None]:
498499
include_preference=False,
499500
filter=chat_req.filter,
500501
search_memory_type="All",
502+
search_tool_memory=False,
501503
playground_search_goal_parser=False,
502504
)
503505
start_time = time.time()

0 commit comments

Comments
 (0)