Added custom prompt support (#1849)

2024-09-10 16:57:32 -07:00
parent 5eeeb4e38c
commit ac7b7aa20a
5 changed files with 122 additions and 3 deletions
--- a/mem0/configs/base.py
+++ b/mem0/configs/base.py
@@ -56,6 +56,10 @@ class MemoryConfig(BaseModel):
        description="The version of the API",
        default="v1.0",
    )
+    custom_prompt: Optional[str] = Field(
+        description="Custom prompt for the memory",
+        default=None,
+    )
    

 class AzureConfig(BaseModel):