Skip to content

Commit f605f79

Browse files
CaralHsifridayLmuzzlol
authored
fix: align MOSProduct._build_system_prompt signature with MOSCore (#169)
* fix: align MOSProduct._build_system_prompt signature with MOSCore - Fix TypeError when calling MOSProduct.chat() method - MOSCore.chat() expects _build_system_prompt(memories, base_prompt=...) - MOSProduct._build_system_prompt had incompatible signature (user_id, memories) - Updated signature to match parent class interface - Removed unused user_id parameter from method * ran format --------- Co-authored-by: chunyu li <[email protected]> Co-authored-by: muzzlol <[email protected]>
1 parent 9764d5e commit f605f79

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
@@ -339,7 +339,9 @@ def _get_or_create_user_config(
339339

340340
return self._create_user_config(user_id, user_config)
341341

342-
def _build_system_prompt(self, user_id: str, memories_all: list[TextualMemoryItem]) -> str:
342+
def _build_system_prompt(
343+
self, memories_all: list[TextualMemoryItem], base_prompt: str | None = None
344+
) -> str:
343345
"""
344346
Build custom system prompt for the user with memory references.
345347
@@ -751,7 +753,7 @@ def chat_with_references(
751753
memories_list = memories_result[0]["memories"]
752754
memories_list = self._filter_memories_by_threshold(memories_list)
753755
# Build custom system prompt with relevant memories
754-
system_prompt = self._build_system_prompt(user_id, memories_list)
756+
system_prompt = self._build_system_prompt(memories_list, base_prompt=None)
755757

756758
# Get chat history
757759
if user_id not in self.chat_history_manager:

0 commit comments

Comments
 (0)