Skip to content

Commit 002f990

Browse files
author
yuan.wang
committed
add logger in playground chat
1 parent a751823 commit 002f990

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/memos/api/handlers/chat_handler.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -465,6 +465,7 @@ def generate_chat_response() -> Generator[str, None, None]:
465465
conversation=chat_req.history,
466466
mode="fine",
467467
)
468+
self.logger.info(f"[PLAYGROUND chat parsed_goal]: {parsed_goal}")
468469

469470
if chat_req.beginner_guide_step == "first":
470471
chat_req.internet_search = False
@@ -479,8 +480,8 @@ def generate_chat_response() -> Generator[str, None, None]:
479480

480481
# ====== second deep search ======
481482
search_req = APISearchRequest(
482-
query=parsed_goal.rephrased_query
483-
or chat_req.query + (f"{parsed_goal.tags}" if parsed_goal.tags else ""),
483+
query=(parsed_goal.rephrased_query or chat_req.query)
484+
+ (f"{parsed_goal.tags}" if parsed_goal.tags else ""),
484485
user_id=chat_req.user_id,
485486
readable_cube_ids=readable_cube_ids,
486487
mode="fast",
@@ -494,6 +495,9 @@ def generate_chat_response() -> Generator[str, None, None]:
494495
search_memory_type="All",
495496
search_tool_memory=False,
496497
)
498+
499+
self.logger.info(f"[PLAYGROUND second search query]: {search_req.query}")
500+
497501
start_time = time.time()
498502
search_response = self.search_handler.handle_search_memories(search_req)
499503
end_time = time.time()

0 commit comments

Comments
 (0)