Skip to content

Commit 9898551

Browse files
authored
Merge branch 'main' into fix/nested-data-handling
2 parents 721166f + d131bce commit 9898551

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/memos/mem_os/product.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,9 @@ def _get_or_create_user_config(
327327

328328
return self._create_user_config(user_id, user_config)
329329

330-
def _build_system_prompt(self, user_id: str, memories_all: list[TextualMemoryItem]) -> str:
330+
def _build_system_prompt(
331+
self, memories_all: list[TextualMemoryItem], base_prompt: str | None = None
332+
) -> str:
331333
"""
332334
Build custom system prompt for the user with memory references.
333335
@@ -760,7 +762,7 @@ def chat_with_references(
760762
memories_list = memories_result[0]["memories"]
761763

762764
# Build custom system prompt with relevant memories
763-
system_prompt = self._build_system_prompt(user_id, memories_list)
765+
system_prompt = self._build_system_prompt(memories_list, base_prompt=None)
764766

765767
# Get chat history
766768
target_user_id = user_id if user_id is not None else self.user_id

0 commit comments

Comments
 (0)