MemoryOS 0.1.13__py3-none-any.whl → 0.2.0__py3-none-any.whl

{memoryos-0.1.13.dist-info → memoryos-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: MemoryOS
-Version: 0.1.13
+Version: 0.2.0
 Summary: Intelligence Begins with Memory
 License: Apache-2.0
 Keywords: memory,llm,language model,memoryOS,agent
@@ -22,6 +22,7 @@ Requires-Dist: ollama (>=0.4.8,<0.5.0)
 Requires-Dist: openai (>=1.77.0,<2.0.0)
 Requires-Dist: qdrant-client (>=1.14.2,<2.0.0)
 Requires-Dist: redis (>=6.2.0,<7.0.0)
+Requires-Dist: schedule (>=1.2.2,<2.0.0)
 Requires-Dist: sentence-transformers (>=4.1.0,<5.0.0)
 Requires-Dist: sqlalchemy (>=2.0.41,<3.0.0)
 Requires-Dist: tenacity (>=9.1.2,<10.0.0)
@@ -31,13 +32,11 @@ Description-Content-Type: text/markdown
 
 <div align="center">
   <a href="https://memos.openmem.net/">
-    <img src="docs/assets/banner_new.gif" alt="MemOS Banner">
+    <img src="https://statics.memtensor.com.cn/memos/memos-banner.gif" alt="MemOS Banner">
   </a>
 
-
-
 <h1 align="center">
-  <img src="docs/assets/memos_logo.png" alt="MemOS Logo" width="50"/> MemOS 1.0: 星河 (Stellar)  <img src="https://img.shields.io/badge/status-Preview-blue" alt="Preview Badge"/>
+  <img src="https://statics.memtensor.com.cn/logo/memos_color_m.png" alt="MemOS Logo" width="50"/> MemOS 1.0: 星河 (Stellar)  <img src="https://img.shields.io/badge/status-Preview-blue" alt="Preview Badge"/>
 </h1>
 
   <p>
@@ -50,7 +49,7 @@ Description-Content-Type: text/markdown
     <a href="https://pypi.org/project/MemoryOS">
       <img src="https://img.shields.io/pypi/pyversions/MemoryOS.svg" alt="Supported Python versions">
     </a>
-    <a href="https://memos.openmem.net/docs/home">
+    <a href="https://memos-docs.openmem.net/home/overview/">
       <img src="https://img.shields.io/badge/Documentation-view-blue.svg" alt="Documentation">
     </a>
     <a href="https://arxiv.org/abs/2507.03724">
@@ -62,7 +61,7 @@ Description-Content-Type: text/markdown
     <a href="https://discord.gg/Txbx3gebZR">
       <img src="https://img.shields.io/badge/Discord-join%20chat-7289DA.svg?logo=discord" alt="Discord">
     </a>
-    <a href="docs/assets/qr_code.png">
+    <a href="https://statics.memtensor.com.cn/memos/qr-code.png">
       <img src="https://img.shields.io/badge/WeChat-Group-07C160.svg?logo=wechat" alt="WeChat Group">
     </a>
     <a href="https://opensource.org/license/apache-2-0/">
@@ -73,17 +72,14 @@ Description-Content-Type: text/markdown
 
 ---
 
-  <a href="https://memos.openmem.net/">
-    <img src="docs/assets/sota_score.jpg" alt="SOTA SCORE">
-  </a>
-
+<img src="https://statics.memtensor.com.cn/memos/sota_score.jpg" alt="SOTA SCORE">
 
 **MemOS** is an operating system for Large Language Models (LLMs) that enhances them with long-term memory capabilities. It allows LLMs to store, retrieve, and manage information, enabling more context-aware, consistent, and personalized interactions.
 
-- **Website**: <a href="https://memos.openmem.net/" target="_blank">https://memos.openmem.net/</a>
-- **Documentation**: <a href="https://memos.openmem.net/docs/home" target="_blank">https://memos.openmem.net/docs/home</a>
-- **API Reference**: <a href="https://memos.openmem.net/docs/api/info" target="_blank">https://memos.openmem.net/docs/api/info</a>
-- **Source Code**: <a href="https://github.com/MemTensor/MemOS" target="_blank">https://github.com/MemTensor/MemOS</a>
+- **Website**: https://memos.openmem.net/
+- **Documentation**: https://memos-docs.openmem.net/home/overview/
+- **API Reference**: https://memos-docs.openmem.net/docs/api/info/
+- **Source Code**: https://github.com/MemTensor/MemOS
 
 ## 📈 Performance Benchmark
 
@@ -97,19 +93,12 @@ MemOS demonstrates significant improvements over baseline memory solutions in mu
 
 > 💡 **Temporal reasoning accuracy improved by 159% compared to the OpenAI baseline.**
 
-
-
 ### Details of End-to-End Evaluation on LOCOMO
 
 > [!NOTE]
 > Comparison of LLM Judge Scores across five major tasks in the LOCOMO benchmark. Each bar shows the mean evaluation score judged by LLMs for a given method-task pair, with standard deviation as error bars. MemOS-0630 consistently outperforms baseline methods (LangMem, Zep, OpenAI, Mem0) across all task types, especially in multi-hop and temporal reasoning scenarios.
 
-<a href="https://memos.openmem.net/">
-  <img src="docs/assets/score_all_end2end.jpg" alt="END2END SCORE">
-</a>
-
-
-
+<img src="https://statics.memtensor.com.cn/memos/score_all_end2end.jpg" alt="END2END SCORE">
 
 ## ✨ Key Features
 
@@ -227,7 +216,7 @@ Join our community to ask questions, share your projects, and connect with other
 - **Discord**: Join our <a href="https://discord.gg/Txbx3gebZR" target="_blank">Discord Server</a>.
 - **WeChat**: Scan the QR code to join our WeChat group.
 
-<img src="docs/assets/qr_code.png" alt="QR Code" width="600">
+<img src="https://statics.memtensor.com.cn/memos/qr-code.png" alt="QR Code" width="600">
 
 ## 📜 Citation
 
@@ -270,7 +259,7 @@ url = {https://global-sci.com/article/91443/memory3-language-modeling-with-expli
 
 ## 🙌 Contributing
 
-We welcome contributions from the community! Please read our [contribution guidelines](https://memos.openmem.net/docs/contribution/overview) to get started.
+We welcome contributions from the community! Please read our [contribution guidelines](https://memos-docs.openmem.net/contribution/overview) to get started.
 
 ## 📄 License
 

{memoryos-0.1.13.dist-info → memoryos-0.2.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-memos/__init__.py,sha256=c6S03YvsixngtCDQY7KeMPbLwPXHedu2T0EymeXAejY,576
+memos/__init__.py,sha256=9C9wxypP0Ch_j6yjvoNgmbr8spHUAsdC1hd49UswG2U,575
 memos/api/start_api.py,sha256=vDKS3D8AEwlDAkpWxHiblzVqcQ3cvGSnnpP4Jhjm3GE,14656
 memos/chunkers/__init__.py,sha256=7lZOTN3e9Yp5XBsDX5wnWJ3tY126cRU9GmfevzJXAtU,67
 memos/chunkers/base.py,sha256=z0rG5vM7FGremQdSZ_3jlTGbsDtlkWAYWdtSAGqpaR4,655
@@ -10,13 +10,13 @@ memos/configs/chunker.py,sha256=Ulci0MyhS_FkuV95l7Pr9Vei179aEFEUigwVmtmSDfI,1593
 memos/configs/embedder.py,sha256=a7EbWB3tnqeK_pdK6-CneoKIwSqQg6sJrXuHX3J1PnI,1769
 memos/configs/graph_db.py,sha256=h5wz2lJSdB1IVaidzWHXDfvL4_IAjrjYrmypu1QCCq4,1432
 memos/configs/internet_retriever.py,sha256=7x1Sp--skifHQqWvcPFiEOkda6HCCocw6HSO5H127jk,3046
-memos/configs/llm.py,sha256=rfIyuxx2xefBD5NpZaeEy3M0gEgEEhvTp1RmRG-zodw,2378
+memos/configs/llm.py,sha256=iNLGnZYbeqkLpjP4L0sGzoCnH4NH8SBUzBt6bTohxZI,2446
 memos/configs/mem_chat.py,sha256=TjEQHRG1HpLwCBo3hrn5aVK23rykNtV6Be5p4YIg7F4,2571
 memos/configs/mem_cube.py,sha256=AvffqpFC2WmzshouxLJbabuzt0PKLWSauQMnagP_EM0,3088
 memos/configs/mem_os.py,sha256=bUuyi6boIjjvYzrlFoCSqchbWF37IcA5BQi03feq7bg,2476
-memos/configs/mem_reader.py,sha256=y1z4aABKgg-QzROApy3gAPyYX9M5CLi3vT_xKye8Ohs,1864
+memos/configs/mem_reader.py,sha256=m5pRr3DxtmOKBFdbnCki_XgG47oKfcasJNwZGXg9gN4,2022
 memos/configs/mem_scheduler.py,sha256=bTsBfl9GoXbjkiPDCenH8K2QY4RfAJNk-XN1UBl2W4Q,2959
-memos/configs/memory.py,sha256=-f9NpWNbZ11AMHrKngmqygXpb90mXKXIv8WxcQap6Nk,6842
+memos/configs/memory.py,sha256=kAGmniEgBdHgAsZGXeUF7J-ekwU5H9HnxQp_3jhUtLQ,6976
 memos/configs/parser.py,sha256=dy-QoevJbCnkJePKgpzR4oziOzYnS4jB6XH-YrpeMns,1145
 memos/configs/utils.py,sha256=X9NQ6-xURsZROAdS3WT96phVfHcOHgDPOo2Yq68QoKM,242
 memos/configs/vec_db.py,sha256=Gjhhda94pyTDjyJGe2Z6rVEqH4FtViiwq1-7QWhjarM,2335
@@ -30,14 +30,15 @@ memos/exceptions.py,sha256=UnBoZUYdwb1KoQPE-pXSLT4yOjkwxse9fx0rb_MhEzo,531
 memos/graph_dbs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/graph_dbs/base.py,sha256=dTCdoYNS6FZQukpmpqhh8c_NwZ9Myql41g1Im8aDR6E,6873
 memos/graph_dbs/factory.py,sha256=9fS8w0GowWNMSupEyXyTRnivX-RJxTkxPy-tNHs15OM,732
-memos/graph_dbs/neo4j.py,sha256=95x1SJVaXbqryykvUqQXqHMBTtZ4za1iew_8T1hlGq4,31268
+memos/graph_dbs/item.py,sha256=BfK17hA_hHu7I348jWugP_rb9GS5hpKfgqtYGGHBohk,1450
+memos/graph_dbs/neo4j.py,sha256=bOTBvHGw_65dudqIpQep9D95qBPyx-NHfTybsNAdjiM,33672
 memos/hello_world.py,sha256=RV1vXfK1_U_xAvSusqc-4A8wk3yr8WEQ9q88dmBxvnI,3057
 memos/llms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/llms/base.py,sha256=E7qlIWUi_bxRwbKw0vGG7NjHksz45D_-gcrHKT3Ak1k,438
 memos/llms/factory.py,sha256=1JP2ZwUKN4ywaNgdZcRM6wSjahnEXqyuFiGzoxLCadY,792
 memos/llms/hf.py,sha256=rxcR8nAvhezsIQNeOeow-xn-z2h9j1poJO2jPNcJAg8,9008
 memos/llms/ollama.py,sha256=_nZYe0OauPWWcFFYVPSNtvyDGblrL49-EznApa-Mr5I,2565
-memos/llms/openai.py,sha256=QoksbFh0ptsRFDOJzUyTl-Qovd3LzVYid9JqGSBH39U,1145
+memos/llms/openai.py,sha256=agIFI58UXz-fOEth4EBTUlQ5p6ECBg9v2gFed6OVFzw,1192
 memos/llms/utils.py,sha256=OcbM9iSpFJpio7sTT5wvxVx-JnqjIx7eSgiRk7dt0ZI,292
 memos/log.py,sha256=ocQi2NWlpefgxOJi9EoB0as6tXhLA4dmXuVIgZT52FE,2234
 memos/mem_chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -56,7 +57,7 @@ memos/mem_reader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 memos/mem_reader/base.py,sha256=SSuaD3J88XbHsME1Qa-EAgZ57xApPHPHppMSeeS3JZ4,957
 memos/mem_reader/factory.py,sha256=emKnId9BhScSkqCZyLSLWntixnU3wAMIOVdsRpHldJA,766
 memos/mem_reader/memory.py,sha256=f3fAjrs8Jf6mBZWTgzkEZle7XjDwTJHxM1L2sOb85Tg,13288
-memos/mem_reader/simple_struct.py,sha256=QQOBX-26cysWH-vudf9FzxGa2O0pkZ8_bQ7zGmGXz4o,9493
+memos/mem_reader/simple_struct.py,sha256=SfFxtZ_YGcDh-A4Ztx4ouEBC6m0j8mLCVF2HN4oGRy8,9739
 memos/mem_scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_scheduler/base_scheduler.py,sha256=bYxLm6nUorJGm8DCiBOJ6DG8eqCK3cCIrm3cEe2be7M,6617
 memos/mem_scheduler/general_scheduler.py,sha256=Tp9Nh8qFVR9TklmmISSoBoeHoWTqM0P977gmb-HS6rM,12895
@@ -84,12 +85,16 @@ memos/memories/parametric/lora.py,sha256=TqSI2OjmFi-XXCeM-MchSwh1sAhOwL7_JnOwSy9
 memos/memories/textual/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/textual/base.py,sha256=3XVeuXZY3ZFzZ6zHIE3l5hx7X2PiH7orxvCwrNZ1l3U,2812
 memos/memories/textual/general.py,sha256=7iektZA-_iR80sOlADW2EFkPT5Utbunu9hdZgT7Ocrk,11072
-memos/memories/textual/item.py,sha256=v6YYDPdPt5uwsJxLlaMwi5m1r7Als2EwgFQhqBvNHRQ,6296
+memos/memories/textual/item.py,sha256=8tSsiDztsxfyNXWB1TG2E5a4nhIQcuYMJ_BrnPl6osQ,6323
 memos/memories/textual/naive.py,sha256=Z_gfbxI6cQGJ_raOTQic4fnpo493Xq3yEQ8qDV4xplo,6954
-memos/memories/textual/tree.py,sha256=GDG2KS6P9MQaBUmOrHEBUDgxbWCDzELF9XZshyC5zXk,13390
+memos/memories/textual/tree.py,sha256=xOc4yMO-tISHmndS-T3ZxGQDr3cARapiTU71PTZdJbQ,13514
 memos/memories/textual/tree_text_memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/textual/tree_text_memory/organize/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/memories/textual/tree_text_memory/organize/manager.py,sha256=_nCKzkDk77D_KRSLHwjT1f8Ra9waNVCKs2sBEsmBypk,12341
+memos/memories/textual/tree_text_memory/organize/conflict.py,sha256=AIJ3pZLPkYsOZE_AqqnCykaGsCk00wLZ8dAdpaz29uE,8462
+memos/memories/textual/tree_text_memory/organize/manager.py,sha256=ZT0VZnvR-Nd3P3f2eu6YumoauMUmp94NVP90JpI9C5w,13486
+memos/memories/textual/tree_text_memory/organize/redundancy.py,sha256=Z88OaNs5YvFY3liXDRFqPcdiIVdTTdlJ_esx3oOiRxI,9075
+memos/memories/textual/tree_text_memory/organize/relation_reason_detector.py,sha256=gRG6iMT8Rm2Pk42UO1JzRaj3neIYmmOJmySi1knaGXA,8809
+memos/memories/textual/tree_text_memory/organize/reorganizer.py,sha256=fy1P_9RWIiKBDayAfCviRZzJx48TMY09ynE0OenMX-o,22932
 memos/memories/textual/tree_text_memory/retrieve/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/textual/tree_text_memory/retrieve/internet_retriever.py,sha256=asSzVcSVOiigN4-4NzFc7wzthGDK6P5sODIDF5W9egA,8467
 memos/memories/textual/tree_text_memory/retrieve/internet_retriever_factory.py,sha256=pBlufTc5-mdWcyKR_R0Syev3QPMtV83mydyIw0Saz4o,3192
@@ -107,16 +112,17 @@ memos/parsers/factory.py,sha256=hPKTR0wVgMQ5Z9ZL-a9FyHWCpz9UcrG2oSEDYBAoB3g,704
 memos/parsers/markitdown.py,sha256=Yh7YAv8PFsz3WgegrwtA_hqaC9W4SOMZP7IRXxcnn_w,615
 memos/settings.py,sha256=HYNBFRW0CS2Kh4GmMRMA-Yrh3ZhJno4S4B6gx6P3T30,178
 memos/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/templates/mem_reader_prompts.py,sha256=_I9W5AHpmyQKzbuty-NkTBXufDtHzxQmXGvMgxScTa0,5924
+memos/templates/mem_reader_prompts.py,sha256=tEsrlcYifQoSTZ0JXZr8Oo_--7zo3TPKAFoaHnDHTtE,7701
 memos/templates/mem_scheduler_prompts.py,sha256=53bcOCC32790tVJ7_s7o0jwu6hg7jUs-z8dyahlLv24,2417
 memos/templates/mos_prompts.py,sha256=QJrMb4FwMOAVb-SFSCGTuqqzvD_YhBTpdVBvpUIftK0,3595
+memos/templates/tree_reorganize_prompts.py,sha256=5nN5m8tlxxy46Xz4x9bGhzMPkywVFKDFqvSCjK3xnTA,6347
 memos/types.py,sha256=N7XBYxDTdc50KEsS6YxHvYgs23ykGsZ-wNnaJBVdVi4,1791
 memos/vec_dbs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/vec_dbs/base.py,sha256=a_-Gsh51C1_8R35mK7DykDxQTicmdgyz4fbEw8d5Mlg,3165
 memos/vec_dbs/factory.py,sha256=Noa4caqzPT9b59i2jzdpAHFCSHiMfDmgRox1POkRRfE,710
 memos/vec_dbs/item.py,sha256=mLrcHF0nWtMCUjScBgaeeSqabQ3vJhKr_6wrU_g25ns,1425
 memos/vec_dbs/qdrant.py,sha256=I_XFmz_7SgFojMTFaqkPvwQEdMBl0MsCKZ43qatwuV0,9858
-memoryos-0.1.13.dist-info/LICENSE,sha256=FU-b6N8tVc7dzUZGyNjUIG1Ihnrh2iuBziq4a1Gl8HU,11358
-memoryos-0.1.13.dist-info/METADATA,sha256=RfJVrfNGHwDlsIlXcR6UG6egWsjYXNtWOIbT0SUhNGc,12766
-memoryos-0.1.13.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-memoryos-0.1.13.dist-info/RECORD,,
+memoryos-0.2.0.dist-info/LICENSE,sha256=FU-b6N8tVc7dzUZGyNjUIG1Ihnrh2iuBziq4a1Gl8HU,11358
+memoryos-0.2.0.dist-info/METADATA,sha256=q0T0qM5v5Dkq9J5kKBtahtGWUHlSsu7R_dT3O_0Rb1A,12636
+memoryos-0.2.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+memoryos-0.2.0.dist-info/RECORD,,

memos/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.13"
+__version__ = "0.2.0"
 
 from memos.configs.mem_cube import GeneralMemCubeConfig
 from memos.configs.mem_os import MOSConfig

memos/configs/llm.py

@@ -24,6 +24,7 @@ class OpenAILLMConfig(BaseLLMConfig):
     api_base: str = Field(
         default="https://api.openai.com/v1", description="Base URL for OpenAI API"
     )
+    extra_body: Any = Field(default=None, description="extra body")
 
 
 class OllamaLLMConfig(BaseLLMConfig):

memos/configs/mem_reader.py

@@ -22,6 +22,10 @@ class BaseMemReaderConfig(BaseConfig):
     chunker: ChunkerConfigFactory = Field(
         ..., description="Chunker configuration for the MemReader"
     )
+    remove_prompt_example: bool = Field(
+        default=False,
+        description="whether remove example in memory extraction prompt to save token",
+    )
 
 
 class SimpleStructMemReaderConfig(BaseMemReaderConfig):

memos/configs/memory.py

@@ -161,6 +161,11 @@ class TreeTextMemoryConfig(BaseTextMemoryConfig):
         description="Internet retriever configuration (optional)",
     )
 
+    reorganize: bool | None = Field(
+        False,
+        description="Optional description for this memory configuration.",
+    )
+
 
 # ─── 3. Global Memory Config Factory ──────────────────────────────────────────
 

memos/graph_dbs/item.py

@@ -0,0 +1,46 @@
+import uuid
+
+from typing import Any, Literal
+
+from pydantic import BaseModel, ConfigDict, Field, field_validator
+
+from memos.memories.textual.item import TextualMemoryItem
+
+
+class GraphDBNode(TextualMemoryItem):
+    pass
+
+
+class GraphDBEdge(BaseModel):
+    """Represents an edge in a graph database (corresponds to Neo4j relationship)."""
+
+    id: str = Field(
+        default_factory=lambda: str(uuid.uuid4()), description="Unique identifier for the edge"
+    )
+    source: str = Field(..., description="Source node ID")
+    target: str = Field(..., description="Target node ID")
+    type: Literal["RELATED", "PARENT"] = Field(
+        ..., description="Relationship type (must be one of 'RELATED', 'PARENT')"
+    )
+    properties: dict[str, Any] | None = Field(
+        default=None, description="Additional properties for the edge"
+    )
+
+    model_config = ConfigDict(extra="forbid")
+
+    @field_validator("id")
+    @classmethod
+    def validate_id(cls, v):
+        """Validate that ID is a valid UUID."""
+        if not isinstance(v, str) or not uuid.UUID(v, version=4):
+            raise ValueError("ID must be a valid UUID string")
+        return v
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "GraphDBEdge":
+        """Create GraphDBEdge from dictionary."""
+        return cls(**data)
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to dictionary format."""
+        return self.model_dump(exclude_none=True)

memos/graph_dbs/neo4j.py

@@ -92,6 +92,16 @@ class Neo4jGraphDB(BaseGraphDB):
             result = session.run(query, memory_type=memory_type)
             return result.single()["count"]
 
+    def count_nodes(self, scope: str) -> int:
+        query = """
+        MATCH (n:Memory)
+        WHERE n.memory_type = $scope
+        RETURN count(n) AS count
+        """
+        with self.driver.session(database=self.db_name) as session:
+            result = session.run(query, {"scope": scope}).single()
+            return result["count"]
+
     def remove_oldest_memory(self, memory_type: str, keep_latest: int) -> None:
         """
         Remove all WorkingMemory nodes except the latest `keep_latest` entries.
@@ -336,6 +346,49 @@ class Neo4jGraphDB(BaseGraphDB):
         """
         raise NotImplementedError
 
+    def get_neighbors_by_tag(
+        self,
+        tags: list[str],
+        exclude_ids: list[str],
+        top_k: int = 5,
+        min_overlap: int = 1,
+    ) -> list[dict[str, Any]]:
+        """
+        Find top-K neighbor nodes with maximum tag overlap.
+
+        Args:
+            tags: The list of tags to match.
+            exclude_ids: Node IDs to exclude (e.g., local cluster).
+            top_k: Max number of neighbors to return.
+            min_overlap: Minimum number of overlapping tags required.
+
+        Returns:
+            List of dicts with node details and overlap count.
+        """
+        query = """
+            MATCH (n:Memory)
+            WHERE NOT n.id IN $exclude_ids
+            AND n.status = 'activated'
+            AND n.type <> 'reasoning'
+            AND n.memory_type <> 'WorkingMemory'
+            WITH n, [tag IN n.tags WHERE tag IN $tags] AS overlap_tags
+            WHERE size(overlap_tags) >= $min_overlap
+            RETURN n, size(overlap_tags) AS overlap_count
+            ORDER BY overlap_count DESC
+            LIMIT $top_k
+        """
+
+        params = {
+            "tags": tags,
+            "exclude_ids": exclude_ids,
+            "min_overlap": min_overlap,
+            "top_k": top_k,
+        }
+
+        with self.driver.session(database=self.db_name) as session:
+            result = session.run(query, params)
+            return [_parse_node(dict(record["n"])) for record in result]
+
     def get_children_with_embeddings(self, id: str) -> list[str]:
         query = """
         MATCH (p:Memory)-[:PARENT]->(c:Memory)
@@ -386,14 +439,10 @@ class Neo4jGraphDB(BaseGraphDB):
             record = session.run(query, {"center_id": center_id}).single()
 
             if not record:
-                logger.warning(
-                    f"No active node found for center_id={center_id} with status={center_status}"
-                )
                 return {"core_node": None, "neighbors": [], "edges": []}
 
             centers = record["centers"]
             if not centers or centers[0] is None:
-                logger.warning(f"Center node not found or inactive for id={center_id}")
                 return {"core_node": None, "neighbors": [], "edges": []}
 
             core_node = _parse_node(dict(centers[0]))
@@ -730,6 +779,24 @@ class Neo4jGraphDB(BaseGraphDB):
             results = session.run(query, {"scope": scope})
             return [_parse_node(dict(record["n"])) for record in results]
 
+    def get_structure_optimization_candidates(self, scope: str) -> list[dict]:
+        """
+        Find nodes that are likely candidates for structure optimization:
+        - Isolated nodes, nodes with empty background, or nodes with exactly one child.
+        - Plus: the child of any parent node that has exactly one child.
+        """
+        query = """
+                MATCH (n:Memory)
+                WHERE n.memory_type = $scope
+                  AND n.status = 'activated'
+                  AND NOT ( (n)-[:PARENT]->() OR ()-[:PARENT]->(n) )
+                RETURN n.id AS id, n AS node
+                """
+
+        with self.driver.session(database=self.db_name) as session:
+            results = session.run(query, {"scope": scope})
+            return [_parse_node({"id": record["id"], **dict(record["node"])}) for record in results]
+
     def drop_database(self) -> None:
         """
         Permanently delete the entire database this instance is using.
@@ -744,7 +811,7 @@ class Neo4jGraphDB(BaseGraphDB):
 
     def _ensure_database_exists(self):
         with self.driver.session(database="system") as session:
-            session.run(f"CREATE DATABASE {self.db_name} IF NOT EXISTS")
+            session.run(f"CREATE DATABASE $db_name IF NOT EXISTS", db_name=self.db_name)
 
         # Wait until the database is available
         for _ in range(10):

memos/llms/openai.py

@@ -22,6 +22,7 @@ class OpenAILLM(BaseLLM):
         response = self.client.chat.completions.create(
             model=self.config.model_name_or_path,
             messages=messages,
+            extra_body=self.config.extra_body,
             temperature=self.config.temperature,
             max_tokens=self.config.max_tokens,
             top_p=self.config.top_p,

memos/mem_reader/simple_struct.py

@@ -1,7 +1,7 @@
 import concurrent.futures
 import copy
 import json
-
+import re
 from abc import ABC
 from typing import Any
 
@@ -17,6 +17,7 @@ from memos.parsers.factory import ParserFactory
 from memos.templates.mem_reader_prompts import (
     SIMPLE_STRUCT_DOC_READER_PROMPT,
     SIMPLE_STRUCT_MEM_READER_PROMPT,
+    SIMPLE_STRUCT_MEM_READER_EXAMPLE,
 )
 
 
@@ -39,11 +40,11 @@ class SimpleStructMemReader(BaseMemReader, ABC):
         self.chunker = ChunkerFactory.from_config(config.chunker)
 
     def _process_chat_data(self, scene_data_info, info):
-        prompt = (
-            SIMPLE_STRUCT_MEM_READER_PROMPT.replace("${user_a}", "user")
-            .replace("${user_b}", "assistant")
-            .replace("${conversation}", "\n".join(scene_data_info))
+        prompt = SIMPLE_STRUCT_MEM_READER_PROMPT.replace(
+            "${conversation}", "\n".join(scene_data_info)
         )
+        if self.config.remove_prompt_example:
+            prompt = prompt.replace(SIMPLE_STRUCT_MEM_READER_EXAMPLE, "")
 
         messages = [{"role": "user", "content": prompt}]
 
@@ -228,7 +229,11 @@ class SimpleStructMemReader(BaseMemReader, ABC):
 
     def parse_json_result(self, response_text):
         try:
-            response_text = response_text.replace("```", "").replace("json", "")
+            json_start = response_text.find("{")
+            response_text = response_text[json_start:]
+            response_text = response_text.replace("```", "").strip()
+            if response_text[-1] != "}":
+                response_text += "}"
             response_json = json.loads(response_text)
             return response_json
         except json.JSONDecodeError as e:

memos/memories/textual/item.py

@@ -27,7 +27,9 @@ class TextualMemoryMetadata(BaseModel):
         default="activated",
         description="The status of the memory, e.g., 'activated', 'archived', 'deleted'.",
     )
-    type: Literal["procedure", "fact", "event", "opinion", "topic"] | None = Field(default=None)
+    type: Literal["procedure", "fact", "event", "opinion", "topic", "reasoning"] | None = Field(
+        default=None
+    )
     memory_time: str | None = Field(
         default=None,
         description='The time the memory occurred or refers to. Must be in standard `YYYY-MM-DD` format. Relative expressions such as "yesterday" or "tomorrow" are not allowed.',

memos/memories/textual/tree.py

@@ -35,7 +35,11 @@ class TreeTextMemory(BaseTextMemory):
         self.dispatcher_llm: OpenAILLM | OllamaLLM = LLMFactory.from_config(config.dispatcher_llm)
         self.embedder: OllamaEmbedder = EmbedderFactory.from_config(config.embedder)
         self.graph_store: Neo4jGraphDB = GraphStoreFactory.from_config(config.graph_db)
-        self.memory_manager: MemoryManager = MemoryManager(self.graph_store, self.embedder)
+        self.is_reorganize = config.reorganize
+
+        self.memory_manager: MemoryManager = MemoryManager(
+            self.graph_store, self.embedder, self.extractor_llm, is_reorganize=self.is_reorganize
+        )
 
         # Create internet retriever if configured
         self.internet_retriever = None