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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: MemoryOS
-Version: 0.1.12
+Version: 0.1.13
 Summary: Intelligence Begins with Memory
 License: Apache-2.0
 Keywords: memory,llm,language model,memoryOS,agent
@@ -53,8 +53,8 @@ Description-Content-Type: text/markdown
     <a href="https://memos.openmem.net/docs/home">
       <img src="https://img.shields.io/badge/Documentation-view-blue.svg" alt="Documentation">
     </a>
-    <a href="https://arxiv.org/abs/2505.22101">
-        <img src="https://img.shields.io/badge/arXiv-2505.22101-b31b1b.svg" alt="ArXiv Paper">
+    <a href="https://arxiv.org/abs/2507.03724">
+        <img src="https://img.shields.io/badge/arXiv-2507.03724-b31b1b.svg" alt="ArXiv Paper">
     </a>
     <a href="https://github.com/MemTensor/MemOS/discussions">
       <img src="https://img.shields.io/badge/GitHub-Discussions-181717.svg?logo=github" alt="GitHub Discussions">
@@ -62,7 +62,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/memos-wechat-group.png">
+    <a href="docs/assets/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/">
@@ -74,7 +74,7 @@ Description-Content-Type: text/markdown
 ---
 
   <a href="https://memos.openmem.net/">
-    <img src="docs/assets/SOTA_Score.jpg" alt="SOTA SCORE">
+    <img src="docs/assets/sota_score.jpg" alt="SOTA SCORE">
   </a>
 
 
@@ -182,7 +182,11 @@ For more detailed examples, please check out the [`examples`](./examples) direct
 ## 📦 Installation
 
 > [!WARNING]
-> Currently, MemOS primarily supports Linux platforms. You may encounter issues on Windows and macOS temporarily.
+> MemOS is compatible with Linux, Windows, and macOS.
+>
+> However, if you're using macOS, please note that there may be dependency issues that are difficult to resolve.
+>
+> For example, compatibility with macOS 13 Ventura is currently challenging.
 
 ### Install via pip
 
@@ -223,24 +227,50 @@ 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="docs/assets/qr_code.png" alt="QR Code" width="600">
 
 ## 📜 Citation
 
-If you use MemOS in your research, please cite our paper:
+> [!NOTE]
+> We publicly released the Short Version on **May 28, 2025**, making it the earliest work to propose the concept of a Memory Operating System for LLMs.
+
+If you use MemOS in your research, we would appreciate citations to our papers.
 
 ```bibtex
-@article{li2025memos,
+
+@article{li2025memos_long,
+  title={MemOS: A Memory OS for AI System},
+  author={Li, Zhiyu and Song, Shichao and Xi, Chenyang and Wang, Hanyu and Tang, Chen and Niu, Simin and Chen, Ding and Yang, Jiawei and Li, Chunyu and Yu, Qingchen and Zhao, Jihao and Wang, Yezhaohui and Liu, Peng and Lin, Zehao and Wang, Pengyuan and Huo, Jiahao and Chen, Tianyi and Chen, Kai and Li, Kehang and Tao, Zhen and Ren, Junpeng and Lai, Huayi and Wu, Hao and Tang, Bo and Wang, Zhenren and Fan, Zhaoxin and Zhang, Ningyu and Zhang, Linfeng and Yan, Junchi and Yang, Mingchuan and Xu, Tong and Xu, Wei and Chen, Huajun and Wang, Haofeng and Yang, Hongkang and Zhang, Wentao and Xu, Zhi-Qin John and Chen, Siheng and Xiong, Feiyu},
+  journal={arXiv preprint arXiv:2507.03724},
+  year={2025},
+  url={https://arxiv.org/abs/2507.03724}
+}
+
+@article{li2025memos_short,
   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}
+  year={2025},
+  url={https://arxiv.org/abs/2505.22101}
+}
+
+@article{yang2024memory3,
+author = {Yang, Hongkang and Zehao, Lin and Wenjin, Wang and Wu, Hao and Zhiyu, Li and Tang, Bo and Wenqiang, Wei and Wang, Jinbo and Zeyun, Tang and Song, Shichao and Xi, Chenyang and Yu, Yu and Kai, Chen and Xiong, Feiyu and Tang, Linpeng and Weinan, E},
+title = {Memory$^3$: Language Modeling with Explicit Memory},
+journal = {Journal of Machine Learning},
+year = {2024},
+volume = {3},
+number = {3},
+pages = {300--346},
+issn = {2790-2048},
+doi = {https://doi.org/10.4208/jml.240708},
+url = {https://global-sci.com/article/91443/memory3-language-modeling-with-explicit-memory}
 }
 ```
 
 ## 🙌 Contributing
 
-We welcome contributions from the community! Please read our [contribution guidelines](./docs/contribution/overview.md) to get started.
+We welcome contributions from the community! Please read our [contribution guidelines](https://memos.openmem.net/docs/contribution/overview) to get started.
 
 ## 📄 License
 
@@ -251,6 +281,7 @@ MemOS is licensed under the [Apache 2.0 License](./LICENSE).
 Stay up to date with the latest MemOS announcements, releases, and community highlights.
 
 - **2025-07-07** – 🎉 *MemOS 1.0 (Stellar) Preview Release*: A SOTA Memory OS for LLMs is now open-sourced.
+- **2025-07-04** – 🎉 *MemOS Paper Released*: [MemOS: A Memory OS for AI System](https://arxiv.org/abs/2507.03724) was published on arXiv.
 - **2025-05-28** – 🎉 *Short Paper Uploaded*: [MemOS: An Operating System for Memory-Augmented Generation (MAG) in Large Language Models](https://arxiv.org/abs/2505.22101) was published on arXiv.
 - **2024-07-04** – 🎉 *Memory3 Model Released at WAIC 2024*: The new memory-layered architecture model was unveiled at the 2024 World Artificial Intelligence Conference.
 - **2024-07-01** – 🎉 *Memory3 Paper Released*: [Memory3: Language Modeling with Explicit Memory](https://arxiv.org/abs/2407.01178) introduces the new approach to structured memory in LLMs.

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

@@ -1,4 +1,4 @@
-memos/__init__.py,sha256=qFk4znjrbZ17ptByd39yT-S-ukxsBEzte0uV3utCSe0,576
+memos/__init__.py,sha256=c6S03YvsixngtCDQY7KeMPbLwPXHedu2T0EymeXAejY,576
 memos/api/start_api.py,sha256=vDKS3D8AEwlDAkpWxHiblzVqcQ3cvGSnnpP4Jhjm3GE,14656
 memos/chunkers/__init__.py,sha256=7lZOTN3e9Yp5XBsDX5wnWJ3tY126cRU9GmfevzJXAtU,67
 memos/chunkers/base.py,sha256=z0rG5vM7FGremQdSZ_3jlTGbsDtlkWAYWdtSAGqpaR4,655
@@ -9,13 +9,14 @@ memos/configs/base.py,sha256=bOOhEU6HGFTq5jne_TXn8Br72rvAfJHYvYllFUPs12A,2588
 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/mem_chat.py,sha256=TjEQHRG1HpLwCBo3hrn5aVK23rykNtV6Be5p4YIg7F4,2571
 memos/configs/mem_cube.py,sha256=AvffqpFC2WmzshouxLJbabuzt0PKLWSauQMnagP_EM0,3088
-memos/configs/mem_os.py,sha256=XEfg8CA6A65vJ0X-OwLQSMsFhGNeBVwCGWGYu4Phuo0,2348
+memos/configs/mem_os.py,sha256=bUuyi6boIjjvYzrlFoCSqchbWF37IcA5BQi03feq7bg,2476
 memos/configs/mem_reader.py,sha256=y1z4aABKgg-QzROApy3gAPyYX9M5CLi3vT_xKye8Ohs,1864
 memos/configs/mem_scheduler.py,sha256=bTsBfl9GoXbjkiPDCenH8K2QY4RfAJNk-XN1UBl2W4Q,2959
-memos/configs/memory.py,sha256=VIu73GIyrrrujZHOB91quSu2XP5oWsbSF3yjKm1o1dE,6606
+memos/configs/memory.py,sha256=-f9NpWNbZ11AMHrKngmqygXpb90mXKXIv8WxcQap6Nk,6842
 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
@@ -49,7 +50,7 @@ memos/mem_cube/general.py,sha256=ad7Qzshf-psBuu24BX_K0NDN0KDuwliAi06TG4wyH1M,553
 memos/mem_cube/utils.py,sha256=-6KiCCzy1PehbGAZY20BVcelvdiKFSu4CXMXi8qNrGU,777
 memos/mem_os/client.py,sha256=0M-WRTlQr7fDAYtq4B8dsMR0PfmyvD-ySMhKcW3Umd0,43
 memos/mem_os/core.py,sha256=csZQqZfbdACapcA85Ahh8vBfi0VPFGM_6yilfmln008,34836
-memos/mem_os/main.py,sha256=QczPzG7P8aIv8oZgoSDAoa3JfzKqMS1qutKbMidsRXE,344
+memos/mem_os/main.py,sha256=sGMleiO1_CG8-v3ctIMFe1hx8Jy_OohFtqqpWgRMhbw,21099
 memos/mem_os/product.py,sha256=SFLFEleZISCeBU19b-uohFsaUNXB9tN-GGypuhDFx1I,3043
 memos/mem_reader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_reader/base.py,sha256=SSuaD3J88XbHsME1Qa-EAgZ57xApPHPHppMSeeS3JZ4,957
@@ -68,7 +69,7 @@ memos/mem_scheduler/modules/retriever.py,sha256=NJEFpX2ap4mSQZ9GYBvBr9QMD4QpgLEY
 memos/mem_scheduler/modules/schemas.py,sha256=J7Rv4M932Wdm-NoOWQne9txOK7j0V9ScdsRH7klPGTI,5301
 memos/mem_scheduler/scheduler_factory.py,sha256=pw6FKO0EuxFRipMnD22DvcjbM6pdmVfJmRjUvdTaU5E,800
 memos/mem_scheduler/utils.py,sha256=xa2BotLf7iAQ4bVJEujPvRjuzWItOA2gP2H3BkBM5XY,582
-memos/mem_user/user_manager.py,sha256=b5MD2vZd3nrl0jb6xWHKchjOnqpCfaaoxomzYYphmms,15269
+memos/mem_user/user_manager.py,sha256=9XIt4Q2HA1fhh9dbjKRAXh-GTdT43WGYSx-IsST7jZM,15654
 memos/memories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/activation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/activation/base.py,sha256=kiC35AOnuofgYMzLxZyVULsfOoVq03BGvi6AXtNvF-I,1151
@@ -85,18 +86,21 @@ memos/memories/textual/base.py,sha256=3XVeuXZY3ZFzZ6zHIE3l5hx7X2PiH7orxvCwrNZ1l3
 memos/memories/textual/general.py,sha256=7iektZA-_iR80sOlADW2EFkPT5Utbunu9hdZgT7Ocrk,11072
 memos/memories/textual/item.py,sha256=v6YYDPdPt5uwsJxLlaMwi5m1r7Als2EwgFQhqBvNHRQ,6296
 memos/memories/textual/naive.py,sha256=Z_gfbxI6cQGJ_raOTQic4fnpo493Xq3yEQ8qDV4xplo,6954
-memos/memories/textual/tree.py,sha256=VxrdhKEsWCeo1T4cNtBufdXKfqz3Di9ieuGty2iGzz0,12156
+memos/memories/textual/tree.py,sha256=GDG2KS6P9MQaBUmOrHEBUDgxbWCDzELF9XZshyC5zXk,13390
 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/retrieve/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/memories/textual/tree_text_memory/retrieve/reasoner.py,sha256=hwy8iOEZwQFHzzvUQ2mwjGly-a4K8PhYQ1PN8H-Gdqo,2305
+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
+memos/memories/textual/tree_text_memory/retrieve/reasoner.py,sha256=5csoGjviFbN9RJ8dm3B6kjvoC8xbPD8UMiGusefHaf0,2228
 memos/memories/textual/tree_text_memory/retrieve/recall.py,sha256=a1P-L3ZER2wh6qTy6p2EhJ-6mSeSisbZgJnHjQv4XkM,5846
 memos/memories/textual/tree_text_memory/retrieve/reranker.py,sha256=U_O0uEKpWHAURhSvkzjYoR2LC8v8vzBjRo8MTeW_6i4,3680
 memos/memories/textual/tree_text_memory/retrieve/retrieval_mid_structs.py,sha256=jZs2S1jtd727gqDKs8clYCOwcqOBnrl_UmFU1FPE0w8,360
-memos/memories/textual/tree_text_memory/retrieve/searcher.py,sha256=dKjyI3OrEhq224XxN0xod-lYHJfPPOEGkGPZHm09kdo,6547
-memos/memories/textual/tree_text_memory/retrieve/task_goal_parser.py,sha256=Jz1ZGmceTa3CBGIPX3j728EawgFkGtGiJFsV4QcSrEw,2620
+memos/memories/textual/tree_text_memory/retrieve/searcher.py,sha256=2TQsYxWXxZns7jQo1_5rZxbdjqReaQIzGDYJb9eZ4RM,8367
+memos/memories/textual/tree_text_memory/retrieve/task_goal_parser.py,sha256=mAPxg5P3wmAvRVrutpeE3i2yC1R90egUUGdTWyWmq1U,2653
 memos/memories/textual/tree_text_memory/retrieve/utils.py,sha256=JU9_3OTRp1lXHWV6SuRUIO_GjRCPclEu9MrjNZX_kck,1406
+memos/memories/textual/tree_text_memory/retrieve/xinyusearch.py,sha256=0bewmGoHE2n-inHDMlHY2wfpGUzTPSRBm9JkGl63rag,10437
 memos/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/parsers/base.py,sha256=AWgWIZzReDiTqiv6z08_9aG2KHVzc4Bdr0Lowz0mWVI,435
 memos/parsers/factory.py,sha256=hPKTR0wVgMQ5Z9ZL-a9FyHWCpz9UcrG2oSEDYBAoB3g,704
@@ -105,13 +109,14 @@ 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_scheduler_prompts.py,sha256=53bcOCC32790tVJ7_s7o0jwu6hg7jUs-z8dyahlLv24,2417
+memos/templates/mos_prompts.py,sha256=QJrMb4FwMOAVb-SFSCGTuqqzvD_YhBTpdVBvpUIftK0,3595
 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.12.dist-info/LICENSE,sha256=FU-b6N8tVc7dzUZGyNjUIG1Ihnrh2iuBziq4a1Gl8HU,11358
-memoryos-0.1.12.dist-info/METADATA,sha256=_XKe1CwEwEL77cX3bF2FRAhu6641gaPTCnFVuOH-cSQ,10818
-memoryos-0.1.12.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-memoryos-0.1.12.dist-info/RECORD,,
+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,,

memos/__init__.py

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

memos/configs/internet_retriever.py

@@ -0,0 +1,81 @@
+"""Configuration classes for internet retrievers."""
+
+from typing import Any, ClassVar
+
+from pydantic import Field, field_validator, model_validator
+
+from memos.configs.base import BaseConfig
+from memos.exceptions import ConfigurationError
+
+
+class BaseInternetRetrieverConfig(BaseConfig):
+    """Base configuration class for internet retrievers."""
+
+    api_key: str = Field(..., description="API key for the search service")
+    search_engine_id: str | None = Field(
+        None, description="Search engine ID (required for Google Custom Search)"
+    )
+
+
+class GoogleCustomSearchConfig(BaseInternetRetrieverConfig):
+    """Configuration class for Google Custom Search API."""
+
+    search_engine_id: str = Field(..., description="Google Custom Search Engine ID (cx parameter)")
+    max_results: int = Field(default=20, description="Maximum number of results to retrieve")
+    num_per_request: int = Field(
+        default=10, description="Number of results per API request (max 10 for Google)"
+    )
+
+
+class BingSearchConfig(BaseInternetRetrieverConfig):
+    """Configuration class for Bing Search API."""
+
+    endpoint: str = Field(
+        default="https://api.bing.microsoft.com/v7.0/search", description="Bing Search API endpoint"
+    )
+    max_results: int = Field(default=20, description="Maximum number of results to retrieve")
+    num_per_request: int = Field(default=10, description="Number of results per API request")
+
+
+class XinyuSearchConfig(BaseInternetRetrieverConfig):
+    """Configuration class for Xinyu Search API."""
+
+    search_engine_id: str | None = Field(
+        None, description="Not used for Xinyu Search (kept for compatibility)"
+    )
+    max_results: int = Field(default=20, description="Maximum number of results to retrieve")
+    num_per_request: int = Field(
+        default=10, description="Number of results per API request (not used for Xinyu)"
+    )
+
+
+class InternetRetrieverConfigFactory(BaseConfig):
+    """Factory class for creating internet retriever configurations."""
+
+    backend: str | None = Field(
+        None, description="Backend for internet retriever (google, bing, etc.)"
+    )
+    config: dict[str, Any] | None = Field(
+        None, description="Configuration for the internet retriever backend"
+    )
+
+    backend_to_class: ClassVar[dict[str, Any]] = {
+        "google": GoogleCustomSearchConfig,
+        "bing": BingSearchConfig,
+        "xinyu": XinyuSearchConfig,
+    }
+
+    @field_validator("backend")
+    @classmethod
+    def validate_backend(cls, backend: str | None) -> str | None:
+        """Validate the backend field."""
+        if backend is not None and backend not in cls.backend_to_class:
+            raise ConfigurationError(f"Invalid internet retriever backend: {backend}")
+        return backend
+
+    @model_validator(mode="after")
+    def create_config(self) -> "InternetRetrieverConfigFactory":
+        if self.backend is not None:
+            config_class = self.backend_to_class[self.backend]
+            self.config = config_class(**self.config)
+        return self

memos/configs/mem_os.py

@@ -57,6 +57,10 @@ class MOSConfig(BaseConfig):
         default=False,
         description="Enable memory scheduler for automated memory management",
     )
+    PRO_MODE: bool = Field(
+        default=False,
+        description="Enable PRO mode for complex query decomposition",
+    )
 
 
 class MemOSConfigFactory(BaseConfig):

memos/configs/memory.py

@@ -5,6 +5,7 @@ from pydantic import Field, field_validator, model_validator
 from memos.configs.base import BaseConfig
 from memos.configs.embedder import EmbedderConfigFactory
 from memos.configs.graph_db import GraphDBConfigFactory
+from memos.configs.internet_retriever import InternetRetrieverConfigFactory
 from memos.configs.llm import LLMConfigFactory
 from memos.configs.vec_db import VectorDBConfigFactory
 from memos.exceptions import ConfigurationError
@@ -133,7 +134,7 @@ class GeneralTextMemoryConfig(BaseTextMemoryConfig):
 
 
 class TreeTextMemoryConfig(BaseTextMemoryConfig):
-    """General memory configuration class."""
+    """Tree text memory configuration class."""
 
     extractor_llm: LLMConfigFactory = Field(
         ...,
@@ -155,6 +156,10 @@ class TreeTextMemoryConfig(BaseTextMemoryConfig):
         default_factory=GraphDBConfigFactory,
         description="Graph database configuration for the tree-memory storage",
     )
+    internet_retriever: InternetRetrieverConfigFactory | None = Field(
+        None,
+        description="Internet retriever configuration (optional)",
+    )
 
 
 # ─── 3. Global Memory Config Factory ──────────────────────────────────────────