Fix discrete_memory_extracted flag in MCP memory search

- Add ExtractedMemoryRecord class for explicit memories (defaults to 't')
- Keep MemoryRecord for raw messages (defaults to 'f')
- Update CreateMemoryRecordRequest to use ExtractedMemoryRecord
- Update LenientMemoryRecord to inherit from ExtractedMemoryRecord
- Fix Redis vector store adapter to preserve discrete_memory_extracted field during search
- Update long_term_memory.py type signatures to accept both memory classes

This resolves the issue where MCP-created memories were incorrectly showing
discrete_memory_extracted='f' instead of 't' in search results.

Author: abrookins

agent-memory-client/agent_memory_client/models.py

@@ -110,7 +110,7 @@ class MemoryRecord(BaseModel):
     )
     discrete_memory_extracted: Literal["t", "f"] = Field(
         default="f",
-        description="Whether memory extraction has run for this memory (only messages)",
+        description="Whether memory extraction has run for this memory",
     )
     memory_type: MemoryTypeEnum = Field(
         default=MemoryTypeEnum.MESSAGE,
@@ -130,6 +130,19 @@ class MemoryRecord(BaseModel):
     )
 
 
+class ExtractedMemoryRecord(MemoryRecord):
+    """A memory record that has already been extracted (e.g., explicit memories from API/MCP)"""
+
+    discrete_memory_extracted: Literal["t", "f"] = Field(
+        default="t",
+        description="Whether memory extraction has run for this memory",
+    )
+    memory_type: MemoryTypeEnum = Field(
+        default=MemoryTypeEnum.SEMANTIC,
+        description="Type of memory",
+    )
+
+
 class ClientMemoryRecord(MemoryRecord):
     """A memory record with a client-provided ID"""
 

agent_memory_server/cli.py

@@ -128,11 +128,14 @@ async def setup_and_run():
             logger.info(f"Starting MCP server on port {port}\n")
             await mcp_app.run_sse_async()
         elif mode == "stdio":
-            # Logging already configured above
+            # Don't run a task worker in stdio mode.
+            # TODO: Make configurable with a CLI flag?
+            settings.use_docket = False
             await mcp_app.run_stdio_async()
         else:
             raise ValueError(f"Invalid mode: {mode}")
 
+    # TODO: Do we really need to update the port again?
     # Update the port in settings
     settings.mcp_port = port
 

agent_memory_server/long_term_memory.py

@@ -1204,7 +1204,6 @@ async def promote_working_memory_to_long_term(
                     text=f"{msg.role}: {msg.content}",
                     namespace=namespace,
                     user_id=current_working_memory.user_id,
-                    memory_type=MemoryTypeEnum.MESSAGE,
                     persisted_at=None,
                 )
 

agent_memory_server/mcp.py

@@ -310,9 +310,7 @@ async def create_long_term_memories(
         if mem.user_id is None and settings.default_mcp_user_id:
             mem.user_id = settings.default_mcp_user_id
 
-    payload = CreateMemoryRecordRequest(
-        memories=[MemoryRecord(**mem.model_dump()) for mem in memories]
-    )
+    payload = CreateMemoryRecordRequest(memories=memories)
     return await core_create_long_term_memory(
         payload, background_tasks=get_background_tasks()
     )

agent_memory_server/models.py

@@ -130,7 +130,7 @@ class MemoryRecord(BaseModel):
     )
     discrete_memory_extracted: Literal["t", "f"] = Field(
         default="f",
-        description="Whether memory extraction has run for this memory (only messages)",
+        description="Whether memory extraction has run for this memory",
     )
     memory_type: MemoryTypeEnum = Field(
         default=MemoryTypeEnum.MESSAGE,
@@ -150,6 +150,19 @@ class MemoryRecord(BaseModel):
     )
 
 
+class ExtractedMemoryRecord(MemoryRecord):
+    """A memory record that has already been extracted (e.g., explicit memories from API/MCP)"""
+
+    discrete_memory_extracted: Literal["t", "f"] = Field(
+        default="t",
+        description="Whether memory extraction has run for this memory",
+    )
+    memory_type: MemoryTypeEnum = Field(
+        default=MemoryTypeEnum.SEMANTIC,
+        description="Type of memory",
+    )
+
+
 class ClientMemoryRecord(MemoryRecord):
     """A memory record with a client-provided ID"""
 
@@ -268,7 +281,7 @@ class MemoryRecordResultsResponse(MemoryRecordResults):
 class CreateMemoryRecordRequest(BaseModel):
     """Payload for creating memory records"""
 
-    memories: list[MemoryRecord]
+    memories: list[ExtractedMemoryRecord]
 
 
 class GetSessionsQuery(BaseModel):
@@ -401,7 +414,7 @@ class MemoryPromptResponse(BaseModel):
     messages: list[base.Message | SystemMessage]
 
 
-class LenientMemoryRecord(MemoryRecord):
+class LenientMemoryRecord(ExtractedMemoryRecord):
     """A memory record that can be created without an ID"""
 
     id: str | None = Field(default_factory=lambda: str(ULID()))