Skip to content

Commit 7e18cae

Browse files
author
yuan.wang
committed
modify bug remove redundant field
1 parent 1b4ef23 commit 7e18cae

File tree

5 files changed

+1
-20
lines changed

5 files changed

+1
-20
lines changed

src/memos/api/handlers/chat_handler.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,6 @@ def generate_chat_response() -> Generator[str, None, None]:
410410
pref_top_k=chat_req.pref_top_k,
411411
filter=chat_req.filter,
412412
search_tool_memory=False,
413-
playground_search_goal_parser=False,
414413
)
415414
start_time = time.time()
416415
search_response = self.search_handler.handle_search_memories(search_req)
@@ -492,7 +491,6 @@ def generate_chat_response() -> Generator[str, None, None]:
492491
filter=chat_req.filter,
493492
search_memory_type="All",
494493
search_tool_memory=False,
495-
playground_search_goal_parser=False,
496494
)
497495
start_time = time.time()
498496
search_response = self.search_handler.handle_search_memories(search_req)
@@ -538,8 +536,6 @@ def generate_chat_response() -> Generator[str, None, None]:
538536
filtered_memories, pref_string, lang=lang
539537
)
540538

541-
print(f"system_prompt: \n{system_prompt}")
542-
543539
# Prepare messages
544540
history_info = chat_req.history[-20:] if chat_req.history else []
545541
current_messages = [

src/memos/api/product_models.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -461,9 +461,6 @@ def _convert_deprecated_fields(self) -> "APISearchRequest":
461461
class APISearchPlaygroundRequest(APISearchRequest):
462462
"""Request model for searching memories in playground."""
463463

464-
# TODO: tmp field for playground search goal parser, will be removed later
465-
playground_search_goal_parser: bool = Field(False, description="Playground search goal parser")
466-
467464

468465
class APIADDRequest(BaseRequest):
469466
"""Request model for creating memories."""

src/memos/memories/textual/tree_text_memory/retrieve/searcher.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -227,8 +227,7 @@ def _parse_task(
227227
query_embedding = None
228228

229229
# fine mode will trigger initial embedding search
230-
# TODO: tmp "playground_search_goal_parser" for playground search goal parser, will be removed later
231-
if mode == "fine_old" or kwargs.get("playground_search_goal_parser", False):
230+
if mode == "fine_old":
232231
logger.info("[SEARCH] Fine mode: embedding search")
233232
query_embedding = self.embedder.embed([query])[0]
234233

@@ -275,10 +274,6 @@ def _parse_task(
275274
**kwargs,
276275
)
277276

278-
# TODO: tmp field playground_search_goal_parser for playground, will be removed later
279-
if kwargs.get("playground_search_goal_parser", False):
280-
parsed_goal.internet_search = False
281-
282277
query = parsed_goal.rephrased_query or query
283278
# if goal has extra memories, embed them too
284279
if parsed_goal.memories:

src/memos/memories/textual/tree_text_memory/retrieve/task_goal_parser.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,6 @@ def parse(
3939
- mode == 'fast': use jieba to split words only
4040
- mode == 'fine': use LLM to parse structured topic/keys/tags
4141
"""
42-
# TODO: tmp mode for playground search goal parser, will be removed later
43-
if kwargs.get("playground_search_goal_parser", False):
44-
mode = "fine"
4542

4643
if mode == "fast":
4744
return self._parse_fast(task_description, context=context, **kwargs)

src/memos/multi_mem_cube/single_cube.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -444,10 +444,6 @@ def _fast_search(
444444
plugin=plugin,
445445
search_tool_memory=search_req.search_tool_memory,
446446
tool_mem_top_k=search_req.tool_mem_top_k,
447-
# TODO: tmp field for playground search goal parser, will be removed later
448-
playground_search_goal_parser=search_req.playground_search_goal_parser
449-
if hasattr(search_req, "playground_search_goal_parser")
450-
else False,
451447
)
452448

453449
formatted_memories = [format_memory_item(data) for data in search_results]

0 commit comments

Comments
 (0)