Support Custom Prompt for Memory Action Decision (#2371)

2025-03-18 14:13:01 +09:00
parent b8f40f728f
commit 66d3f9b93c
9 changed files with 498 additions and 165 deletions
--- a/mem0/configs/base.py
+++ b/mem0/configs/base.py
@@ -48,8 +48,12 @@ class MemoryConfig(BaseModel):
        description="The version of the API",
        default="v1.1",
    )
-    custom_prompt: Optional[str] = Field(
-        description="Custom prompt for the memory",
+    custom_fact_extraction_prompt: Optional[str] = Field(
+        description="Custom prompt for the fact extraction",
+        default=None,
+    )
+    custom_update_memory_prompt: Optional[str] = Field(
+        description="Custom prompt for the update memory",
        default=None,
    )