Skip to content

Commit b6efb0c

Browse files
Wang-Daojiyuan.wangfridayL
authored
Feat/fix palyground bug (#662)
* fix playground bug, internet search judge * fix playground internet bug * modify delete mem * modify tool resp bug in multi cube * fix bug in playground chat handle and search inter * modify prompt * fix bug in playground * fix bug playfround * fix bug * fix code * fix model bug in playground * modify plan b * llm param modify * add logger in playground * modify code * fix bug * modify code * modify code * fix bug * fix search bug in plarground * fixx bug * move schadualr to back --------- Co-authored-by: yuan.wang <[email protected]> Co-authored-by: chunyu li <[email protected]>
1 parent e7b4ea4 commit b6efb0c

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

src/memos/api/handlers/chat_handler.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -405,17 +405,6 @@ def generate_chat_response() -> Generator[str, None, None]:
405405
async_mode="sync",
406406
)
407407

408-
# Use first readable cube ID for scheduler (backward compatibility)
409-
scheduler_cube_id = (
410-
readable_cube_ids[0] if readable_cube_ids else chat_req.user_id
411-
)
412-
self._send_message_to_scheduler(
413-
user_id=chat_req.user_id,
414-
mem_cube_id=scheduler_cube_id,
415-
query=chat_req.query,
416-
label=QUERY_TASK_LABEL,
417-
)
418-
419408
# ====== first search text mem with parse goal ======
420409
search_req = APISearchPlaygroundRequest(
421410
query=chat_req.query,
@@ -454,6 +443,17 @@ def generate_chat_response() -> Generator[str, None, None]:
454443

455444
yield f"data: {json.dumps({'type': 'reference', 'data': reference})}\n\n"
456445

446+
# Use first readable cube ID for scheduler (backward compatibility)
447+
scheduler_cube_id = (
448+
readable_cube_ids[0] if readable_cube_ids else chat_req.user_id
449+
)
450+
self._send_message_to_scheduler(
451+
user_id=chat_req.user_id,
452+
mem_cube_id=scheduler_cube_id,
453+
query=chat_req.query,
454+
label=QUERY_TASK_LABEL,
455+
)
456+
457457
# parse goal for internet search
458458
searcher = self.dependencies.searcher
459459
parsed_goal = searcher.task_goal_parser.parse(
@@ -476,14 +476,14 @@ def generate_chat_response() -> Generator[str, None, None]:
476476
# internet status
477477
yield f"data: {json.dumps({'type': 'status', 'data': 'start_internet_search'})}\n\n"
478478

479-
# ====== internet search with parse goal ======
479+
# ====== second deep search ======
480480
search_req = APISearchPlaygroundRequest(
481481
query=parsed_goal.rephrased_query
482482
or chat_req.query + (f"{parsed_goal.tags}" if parsed_goal.tags else ""),
483483
user_id=chat_req.user_id,
484484
readable_cube_ids=readable_cube_ids,
485485
mode="fast",
486-
internet_search=chat_req.internet_search,
486+
internet_search=chat_req.internet_search or parsed_goal.internet_search,
487487
top_k=chat_req.top_k,
488488
chat_history=chat_req.history,
489489
session_id=chat_req.session_id,

0 commit comments

Comments
 (0)