restore type/visibility

Author: Nyakult

examples/core_memories/general_textual_memory.py

@@ -35,7 +35,7 @@
     {
         "memory": "I'm a RUCer, I'm happy.",
         "metadata": {
-            "title": "happy RUCer",
+            "key": "happy RUCer",
             "source": "conversation",
             "tags": ["happy"],
             "updated_at": "2025-05-19T00:00:00",
@@ -44,7 +44,7 @@
     {
         "memory": "MemOS is awesome!",
         "metadata": {
-            "title": "MemOS",
+            "key": "MemOS",
             "source": "conversation",
             "tags": ["awesome"],
             "updated_at": "2025-05-19T00:00:00",
@@ -71,7 +71,7 @@
             "id": example_id,
             "memory": "User is Chinese.",
             "metadata": {
-                "title": "User Nationality",
+                "key": "User Nationality",
                 "source": "conversation",
                 "tags": ["Nationality"],
                 "updated_at": "2025-05-18T00:00:00",
@@ -98,7 +98,7 @@
         "id": example_id,
         "memory": "User is Canadian.",
         "metadata": {
-            "title": "User Nationality",
+            "key": "User Nationality",
             "source": "conversation",
             "tags": ["Nationality"],
             "updated_at": "2025-05-19T00:00:00",

src/memos/memories/textual/general.py

@@ -66,7 +66,7 @@ def extract(self, messages: MessageList) -> list[TextualMemoryItem]:
             TextualMemoryItem(
                 memory=memory_dict["value"],
                 metadata={
-                    "title": memory_dict["key"],
+                    "key": memory_dict["key"],
                     "source": "conversation",
                     "tags": memory_dict["tags"],
                     "updated_at": datetime.now().isoformat(),

src/memos/memories/textual/item.py

@@ -27,7 +27,8 @@ class TextualMemoryMetadata(BaseModel):
         default="activated",
         description="The status of the memory, e.g., 'activated', 'archived', 'deleted'.",
     )
-    title: str | None = Field(default=None)
+    type: str | None = Field(default=None)
+    key: str | None = Field(default=None, description="Memory key or title.")
     confidence: float | None = Field(
         default=None,
         description="A numeric score (float between 0 and 100) indicating how certain you are about the accuracy or reliability of the memory.",
@@ -39,6 +40,9 @@ class TextualMemoryMetadata(BaseModel):
         default=None,
         description='A list of keywords or thematic labels associated with the memory for categorization or retrieval, e.g., `["travel", "health", "project-x"]`.',
     )
+    visibility: Literal["private", "public", "session"] | None = Field(
+        default=None, description="e.g., 'private', 'public', 'session'"
+    )
     updated_at: str | None = Field(
         default_factory=lambda: datetime.now().isoformat(),
         description="The timestamp of the last modification to the memory. Useful for tracking memory freshness or change history. Format: ISO 8601.",
@@ -58,7 +62,6 @@ class TreeNodeTextualMemoryMetadata(TextualMemoryMetadata):
     memory_type: Literal["WorkingMemory", "LongTermMemory", "UserMemory"] = Field(
         default="WorkingMemory", description="Memory lifecycle type."
     )
-    key: str | None = Field(default=None, description="Memory key or title.")
     sources: list[str] | None = Field(
         default=None, description="Multiple origins of the memory (e.g., URLs, notes)."
     )

tests/memories/textual/test_general.py

@@ -117,15 +117,15 @@ def test_extract(self):
         self.assertEqual(len(result), 1)
         self.assertIsInstance(result[0], TextualMemoryItem)
         self.assertEqual(result[0].memory, "Hello")
-        self.assertEqual(result[0].metadata.title, "greeting")
+        self.assertEqual(result[0].metadata.key, "greeting")
 
     def test_add_memories(self):
         """Test adding memories."""
         memories_to_add = [
             {
                 "memory": "I'm a RUCer, I'm happy.",
                 "metadata": {
-                    "title": "happy RUCer",
+                    "key": "happy RUCer",
                     "source": "conversation",
                     "tags": ["happy"],
                     "updated_at": "2025-05-19T00:00:00",
@@ -134,7 +134,7 @@ def test_add_memories(self):
             {
                 "memory": "MemOS is awesome!",
                 "metadata": {
-                    "title": "MemOS",
+                    "key": "MemOS",
                     "source": "conversation",
                     "tags": ["awesome"],
                     "updated_at": "2025-05-19T00:00:00",
@@ -154,7 +154,7 @@ def test_update_memory(self):
             "id": memory_id_to_update,
             "memory": "This is the updated memory content via dict.",
             "metadata": {
-                "title": "MemOS",
+                "key": "MemOS",
                 "source": "conversation",
                 "tags": ["awesome"],
                 "updated_at": "2025-05-19T00:00:00",
@@ -178,7 +178,7 @@ def test_update_memory(self):
 
         memory_dict = updated_data_to_db.payload
         self.assertEqual(memory_dict["memory"], "This is the updated memory content via dict.")
-        self.assertEqual(memory_dict["metadata"]["title"], "MemOS")
+        self.assertEqual(memory_dict["metadata"]["key"], "MemOS")
         self.assertEqual(memory_dict["metadata"]["source"], "conversation")
 
     def test_search_memories(self):