目录
MemTensor

MemOS initial commit by MemTensor

Co-authored-by: Shichao Song 60967965+Ki-Seki@users.noreply.github.com Co-authored-by: CaralHsi caralhsi@gmail.com Co-authored-by: chunyu li 78344051+fridayL@users.noreply.github.com Co-authored-by: jiawei yang 103758578+J1awei-Yang@users.noreply.github.com Co-authored-by: Wang Hanyu 128483880+MarrytheToilet@users.noreply.github.com Co-authored-by: Jiahao Huo 2151773@tongji.edu.cn Co-authored-by: Hao 42795704+Nyakult@users.noreply.github.com Co-authored-by: Travis Tang travistang@foxmail.com Co-authored-by: Jihao Zhao 119649807+Robot2050@users.noreply.github.com Co-authored-by: spitzblattr 36479200+spitzblattr@users.noreply.github.com Co-authored-by: siminniu 146621079+siminniu@users.noreply.github.com Co-authored-by: hush 61102027+hush-cd@users.noreply.github.com Co-authored-by: Qingchen Yu 92162236+Duguce@users.noreply.github.com Co-authored-by: Kyrie Chen kyriiiechen@gmail.com Co-authored-by: lijicode 34564964+lijicode@users.noreply.github.com Co-authored-by: 王鹏远(Pengyuan Wang) 18237423458@163.com Co-authored-by: Zhiyu Li zhiyulee@ruc.edu.cn Co-authored-by: Yezhaohui Wang yezhaohuiwang@gmail.com

2天前1次提交
目录README.md
MemOS Banner

MemOS Logo MemOS 1.0: 星河 (Stellar) Preview Badge

Static Badge PyPI Version Supported Python versions Documentation ArXiv Paper GitHub Discussions Discord WeChat Group License


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.

📈 Performance Benchmark

MemOS demonstrates significant improvements over baseline memory solutions in multiple reasoning tasks.

Model Avg. Score Multi-Hop Open Domain Single-Hop Temporal Reasoning
OpenAI 0.5275 0.6028 0.3299 0.6183 0.2825
MemOS 0.7331 0.6430 0.5521 0.7844 0.7321
Improvement +38.98% +6.67% +67.35% +26.86% +159.15%

💡 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.

END2END SCORE

✨ Key Features

  • 🧠 Memory-Augmented Generation (MAG): Provides a unified API for memory operations, integrating with LLMs to enhance chat and reasoning with contextual memory retrieval.
  • 📦 Modular Memory Architecture (MemCube): A flexible and modular architecture that allows for easy integration and management of different memory types.
  • 💾 Multiple Memory Types:
    • Textual Memory: For storing and retrieving unstructured or structured text knowledge.
    • Activation Memory: Caches key-value pairs (KVCacheMemory) to accelerate LLM inference and context reuse.
    • Parametric Memory: Stores model adaptation parameters (e.g., LoRA weights).
  • 🔌 Extensible: Easily extend and customize memory modules, data sources, and LLM integrations.

🚀 Getting Started

Here’s a quick example of how to create a MemCube, load it from a directory, access its memories, and save it.

from memos.mem_cube.general import GeneralMemCube

# Initialize a MemCube from a local directory
mem_cube = GeneralMemCube.init_from_dir("examples/data/mem_cube_2")

# Access and print all memories
print("--- Textual Memories ---")
for item in mem_cube.text_mem.get_all():
    print(item)

print("\n--- Activation Memories ---")
for item in mem_cube.act_mem.get_all():
    print(item)

# Save the MemCube to a new directory
mem_cube.dump("tmp/mem_cube")

What about MOS (Memory Operating System)? It’s a higher-level orchestration layer that manages multiple MemCubes and provides a unified API for memory operations. Here’s a quick example of how to use MOS:

from memos.configs.mem_os import MOSConfig
from memos.mem_os.main import MOS


# init MOS
mos_config = MOSConfig.from_json_file("examples/data/config/simple_memos_config.json")
memory = MOS(mos_config)

# create user
user_id = "b41a34d5-5cae-4b46-8c49-d03794d206f5"
memory.create_user(user_id=user_id)

# register cube for user
memory.register_mem_cube("examples/data/mem_cube_2", user_id=user_id)

# add memory for user
memory.add(
    messages=[
        {"role": "user", "content": "I like playing football."},
        {"role": "assistant", "content": "I like playing football too."},
    ],
    user_id=user_id,
)

# Later, when you want to retrieve memory for user
retrieved_memories = memory.search(query="What do you like?", user_id=user_id)
# output text_memories: I like playing football, act_memories, para_memories
print(f"text_memories: {retrieved_memories['text_mem']}")

For more detailed examples, please check out the examples directory.

📦 Installation

[!WARNING] Currently, MemOS primarily supports Linux platforms. You may encounter issues on Windows and macOS temporarily.

Install via pip

pip install MemoryOS

Development Install

To contribute to MemOS, clone the repository and install it in editable mode:

git clone https://github.com/MemTensor/MemOS.git
cd MemOS
make install

Optional Dependencies

Ollama Support

To use MemOS with Ollama, first install the Ollama CLI:

curl -fsSL https://ollama.com/install.sh | sh

Transformers Support

To use functionalities based on the transformers library, ensure you have PyTorch installed (CUDA version recommended for GPU acceleration).

💬 Community & Support

Join our community to ask questions, share your projects, and connect with other developers.

  • GitHub Issues: Report bugs or request features in our GitHub Issues.
  • GitHub Pull Requests: Contribute code improvements via Pull Requests.
  • GitHub Discussions: Participate in our GitHub Discussions to ask questions or share ideas.
  • Discord: Join our Discord Server.
  • WeChat: Scan the QR code to join our WeChat group.
QR Code

📜 Citation

If you use MemOS in your research, please cite our paper:

@article{li2025memos,
  title={MemOS: An Operating System for Memory-Augmented Generation (MAG) in Large Language Models},
  author={Li, Zhiyu and Song, Shichao and Wang, Hanyu and Niu, Simin and Chen, Ding and Yang, Jiawei and Xi, Chenyang and Lai, Huayi and Zhao, Jihao and Wang, Yezhaohui and others},
  journal={arXiv preprint arXiv:2505.22101},
  year={2025}
}

🙌 Contributing

We welcome contributions from the community! Please read our contribution guidelines to get started.

📄 License

MemOS is licensed under the Apache 2.0 License.

📰 News

Stay up to date with the latest MemOS announcements, releases, and community highlights.

关于
11.9 MB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

©Copyright 2023 CCF 开源发展委员会
Powered by Trustie& IntelliDE 京ICP备13000930号