MemoryOS 2.0.5__tar.gz → 2.0.6__tar.gz

{memoryos-2.0.5 → memoryos-2.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MemoryOS
-Version: 2.0.5
+Version: 2.0.6
 Summary: Intelligence Begins with Memory
 License: Apache-2.0
 License-File: LICENSE
@@ -148,12 +148,12 @@ Get Free API: [Try API](https://memos-dashboard.openmem.net/quickstart/?source=g
 ## 🦞 Enhanced OpenClaw with MemOS Plugin
 ![](https://cdn.memtensor.com.cn/img/1770612303123_mnaisk_compressed.png)
 
-- **50-60% lower token usage** – intelligent memory retrieval instead of loading full chat history
-- **Persistent cross-session memory** – remembers user preferences accurately, not just recent chatter
-- **Faster, more accurate responses** – retrieves the right context without hallucinations
+- [**72% lower token usage**](https://x.com/MemOS_dev/status/2020854044583924111) – intelligent memory retrieval instead of loading full chat history
+- [**Multi-agent memory sharing**](https://x.com/MemOS_dev/status/2020538135487062094) – multi-instance agents share memory via same user_id. Automatic context handoff.  
+
+🦞 Your lobster now has a working memory system.
 
 Get your API key: [MemOS Dashboard](https://memos-dashboard.openmem.net/cn/login/)  
-Your lobster now has a working memory system.  
 Try it: Full tutorial → [MemOS-Cloud-OpenClaw-Plugin](https://github.com/MemTensor/MemOS-Cloud-OpenClaw-Plugin)
 
 ## 📌 MemOS: Memory Operating System for AI Agents

{memoryos-2.0.5 → memoryos-2.0.6}/README.md

@@ -67,12 +67,12 @@ Get Free API: [Try API](https://memos-dashboard.openmem.net/quickstart/?source=g
 ## 🦞 Enhanced OpenClaw with MemOS Plugin
 ![](https://cdn.memtensor.com.cn/img/1770612303123_mnaisk_compressed.png)
 
-- **50-60% lower token usage** – intelligent memory retrieval instead of loading full chat history
-- **Persistent cross-session memory** – remembers user preferences accurately, not just recent chatter
-- **Faster, more accurate responses** – retrieves the right context without hallucinations
+- [**72% lower token usage**](https://x.com/MemOS_dev/status/2020854044583924111) – intelligent memory retrieval instead of loading full chat history
+- [**Multi-agent memory sharing**](https://x.com/MemOS_dev/status/2020538135487062094) – multi-instance agents share memory via same user_id. Automatic context handoff.  
+
+🦞 Your lobster now has a working memory system.
 
 Get your API key: [MemOS Dashboard](https://memos-dashboard.openmem.net/cn/login/)  
-Your lobster now has a working memory system.  
 Try it: Full tutorial → [MemOS-Cloud-OpenClaw-Plugin](https://github.com/MemTensor/MemOS-Cloud-OpenClaw-Plugin)
 
 ## 📌 MemOS: Memory Operating System for AI Agents

{memoryos-2.0.5 → memoryos-2.0.6}/pyproject.toml

@@ -4,7 +4,7 @@
 ##############################################################################
 
 name = "MemoryOS"
-version = "2.0.5"
+version = "2.0.6"
 description = "Intelligence Begins with Memory"
 license = {text = "Apache-2.0"}
 readme = "README.md"

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/__init__.py

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

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/config.py

@@ -335,7 +335,6 @@ class APIConfig:
                 # validation requirements during tests/import.
                 "api_base": os.getenv("MEMRADER_API_BASE", "https://api.openai.com/v1"),
                 "remove_think_prefix": True,
-                "extra_body": {"chat_template_kwargs": {"enable_thinking": False}},
             },
         }
 
@@ -531,7 +530,6 @@ class APIConfig:
                                 "api_key": os.getenv("MEMRADER_API_KEY", "EMPTY"),
                                 "api_base": os.getenv("MEMRADER_API_BASE"),
                                 "remove_think_prefix": True,
-                                "extra_body": {"chat_template_kwargs": {"enable_thinking": False}},
                             },
                         },
                         "embedder": APIConfig.get_embedder_config(),

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/handlers/chat_handler.py

@@ -130,6 +130,7 @@ class ChatHandler(BaseHandler):
                 include_preference=chat_req.include_preference,
                 pref_top_k=chat_req.pref_top_k,
                 filter=chat_req.filter,
+                relativity=chat_req.relativity,
             )
 
             search_response = self.search_handler.handle_search_memories(search_req)
@@ -269,6 +270,7 @@ class ChatHandler(BaseHandler):
                         include_preference=chat_req.include_preference,
                         pref_top_k=chat_req.pref_top_k,
                         filter=chat_req.filter,
+                        relativity=chat_req.relativity,
                     )
 
                     search_response = self.search_handler.handle_search_memories(search_req)
@@ -811,6 +813,7 @@ class ChatHandler(BaseHandler):
                             include_preference=chat_req.include_preference,
                             pref_top_k=chat_req.pref_top_k,
                             filter=chat_req.filter,
+                            relativity=chat_req.relativity,
                         )
 
                         search_response = self.search_handler.handle_search_memories(search_req)

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/handlers/config_builders.py

@@ -105,6 +105,7 @@ def build_chat_llm_config() -> list[dict[str, Any]]:
                 }
             ),
             "support_models": cfg.get("support_models", None),
+            "extra_body": cfg.get("extra_body", None),
         }
         for cfg in configs
     ]

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/handlers/memory_handler.py

@@ -202,6 +202,20 @@ def handle_get_memory(memory_id: str, naive_mem_cube: NaiveMemCube) -> GetMemory
     # Get the data from whichever memory source succeeded
     data = (memory or pref).model_dump() if (memory or pref) else None
 
+    if data is not None:
+        # For each returned item, tackle with metadata.info project_id /
+        # operation / manager_user_id
+        metadata = data.get("metadata", None)
+        if metadata is not None and isinstance(metadata, dict):
+            info = metadata.get("info", None)
+            if info is not None and isinstance(info, dict):
+                for key in ("project_id", "operation", "manager_user_id"):
+                    if key not in info:
+                        continue
+                    value = info.pop(key)
+                    if key not in metadata:
+                        metadata[key] = value
+
     return GetMemoryResponse(
         message="Memory retrieved successfully"
         if data
@@ -241,6 +255,25 @@ def handle_get_memory_by_ids(
             except Exception:
                 continue
 
+    # For each returned item, tackle with metadata.info project_id /
+    # operation / manager_user_id
+    for item in memories:
+        if not isinstance(item, dict):
+            continue
+        metadata = item.get("metadata")
+        if not isinstance(metadata, dict):
+            continue
+        info = metadata.get("info")
+        if not isinstance(info, dict):
+            continue
+
+        for key in ("project_id", "operation", "manager_user_id"):
+            if key not in info:
+                continue
+            value = info.pop(key)
+            if key not in metadata:
+                metadata[key] = value
+
     return GetMemoryResponse(
         message="Memories retrieved successfully", code=200, data={"memories": memories}
     )
@@ -345,6 +378,25 @@ def handle_get_memories(
         )
         format_preferences = [format_memory_item(item, save_sources=False) for item in preferences]
 
+        # For each returned item, tackle with metadata.info project_id /
+        # operation / manager_user_id
+        for item in format_preferences:
+            if not isinstance(item, dict):
+                continue
+            metadata = item.get("metadata")
+            if not isinstance(metadata, dict):
+                continue
+            info = metadata.get("info")
+            if not isinstance(info, dict):
+                continue
+
+            for key in ("project_id", "operation", "manager_user_id"):
+                if key not in info:
+                    continue
+                value = info.pop(key)
+                if key not in metadata:
+                    metadata[key] = value
+
     results = post_process_pref_mem(
         results, format_preferences, get_mem_req.mem_cube_id, get_mem_req.include_preference
     )
@@ -412,6 +464,12 @@ def handle_get_memories_dashboard(
     get_mem_req: GetMemoryDashboardRequest, naive_mem_cube: NaiveMemCube
 ) -> GetMemoryResponse:
     results: dict[str, Any] = {"text_mem": [], "pref_mem": [], "tool_mem": [], "skill_mem": []}
+    # for statistics
+    total_text_nodes, total_tool_nodes, total_skill_nodes, total_preference_nodes = 0, 0, 0, 0
+    total_tool_nodes = 0
+    total_skill_nodes = 0
+    total_preference_nodes = 0
+
     text_memory_type = ["WorkingMemory", "LongTermMemory", "UserMemory", "OuterMemory"]
     text_memories_info = naive_mem_cube.text_mem.get_all(
         user_name=get_mem_req.mem_cube_id,
@@ -421,7 +479,7 @@ def handle_get_memories_dashboard(
         filter=get_mem_req.filter,
         memory_type=text_memory_type,
     )
-    text_memories, _ = text_memories_info["nodes"], text_memories_info["total_nodes"]
+    text_memories, total_text_nodes = text_memories_info["nodes"], text_memories_info["total_nodes"]
 
     # Group text memories by cube_id from metadata.user_name
     text_mem_by_cube: dict[str, list] = {}
@@ -453,7 +511,7 @@ def handle_get_memories_dashboard(
             filter=get_mem_req.filter,
             memory_type=["ToolSchemaMemory", "ToolTrajectoryMemory"],
         )
-        tool_memories, _ = (
+        tool_memories, total_tool_nodes = (
             tool_memories_info["nodes"],
             tool_memories_info["total_nodes"],
         )
@@ -488,7 +546,7 @@ def handle_get_memories_dashboard(
             filter=get_mem_req.filter,
             memory_type=["SkillMemory"],
         )
-        skill_memories, _ = (
+        skill_memories, total_skill_nodes = (
             skill_memories_info["nodes"],
             skill_memories_info["total_nodes"],
         )
@@ -515,7 +573,6 @@ def handle_get_memories_dashboard(
         ]
 
     preferences: list[TextualMemoryItem] = []
-    total_preference_nodes = 0
 
     format_preferences = []
     if get_mem_req.include_preference and naive_mem_cube.pref_mem is not None:
@@ -578,4 +635,13 @@ def handle_get_memories_dashboard(
         "skill_mem": results.get("skill_mem", []),
     }
 
+    # statistics
+    statistics = {
+        "total_text_nodes": total_text_nodes,
+        "total_tool_nodes": total_tool_nodes,
+        "total_skill_nodes": total_skill_nodes,
+        "total_preference_nodes": total_preference_nodes,
+    }
+    filtered_results["statistics"] = statistics
+
     return GetMemoryResponse(message="Memories retrieved successfully", data=filtered_results)

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/handlers/search_handler.py

@@ -120,11 +120,14 @@ class SearchHandler(BaseHandler):
                     if not isinstance(mem, dict):
                         continue
                     meta = mem.get("metadata", {})
-                    score = meta.get("relativity", 0.0) if isinstance(meta, dict) else 0.0
+                    if key == "text_mem":
+                        score = meta.get("relativity", 1.0) if isinstance(meta, dict) else 1.0
+                    else:
+                        score = meta.get("score", 1.0) if isinstance(meta, dict) else 1.0
                     try:
-                        score_val = float(score) if score is not None else 0.0
+                        score_val = float(score) if score is not None else 1.0
                     except (TypeError, ValueError):
-                        score_val = 0.0
+                        score_val = 1.0
                     if score_val >= relativity:
                         filtered.append(mem)
 
@@ -220,7 +223,11 @@ class SearchHandler(BaseHandler):
         # Flatten preference memories
         for bucket_idx, bucket in enumerate(pref_buckets):
             for mem in bucket.get("memories", []):
-                score = mem.get("metadata", {}).get("relativity", 0.0)
+                meta = mem.get("metadata", {})
+                if isinstance(meta, dict):
+                    score = meta.get("score", meta.get("relativity", 0.0))
+                else:
+                    score = 0.0
                 flat.append(
                     ("preference", bucket_idx, mem, float(score) if score is not None else 0.0)
                 )

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/product_models.py

@@ -98,6 +98,15 @@ class ChatRequest(BaseRequest):
     add_message_on_answer: bool = Field(True, description="Add dialogs to memory after chat")
     manager_user_id: str | None = Field(None, description="Manager User ID")
     project_id: str | None = Field(None, description="Project ID")
+    relativity: float = Field(
+        0.0,
+        ge=0,
+        description=(
+            "Relevance threshold for recalled memories. "
+            "Only memories with metadata.relativity >= relativity will be returned. "
+            "Use 0 to disable threshold filtering. Default: 0.3."
+        ),
+    )
 
     # ==== Filter conditions ====
     filter: dict[str, Any] | None = Field(
@@ -775,6 +784,15 @@ class APIChatCompleteRequest(BaseRequest):
     add_message_on_answer: bool = Field(True, description="Add dialogs to memory after chat")
     manager_user_id: str | None = Field(None, description="Manager User ID")
     project_id: str | None = Field(None, description="Project ID")
+    relativity: float = Field(
+        0.0,
+        ge=0,
+        description=(
+            "Relevance threshold for recalled memories. "
+            "Only memories with metadata.relativity >= relativity will be returned. "
+            "Use 0 to disable threshold filtering. Default: 0.3."
+        ),
+    )
 
     # ==== Filter conditions ====
     filter: dict[str, Any] | None = Field(

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/api/routers/server_router.py

@@ -53,7 +53,6 @@ from memos.api.product_models import (
     SuggestionResponse,
     TaskQueueResponse,
 )
-from memos.graph_dbs.polardb import PolarDBGraphDB
 from memos.log import get_logger
 from memos.mem_scheduler.base_scheduler import BaseScheduler
 from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
@@ -371,15 +370,6 @@ def feedback_memories(feedback_req: APIFeedbackRequest):
 )
 def get_user_names_by_memory_ids(request: GetUserNamesByMemoryIdsRequest):
     """Get user names by memory ids."""
-    if not isinstance(graph_db, PolarDBGraphDB):
-        raise HTTPException(
-            status_code=400,
-            detail=(
-                "graph_db must be an instance of PolarDBGraphDB to use "
-                "get_user_names_by_memory_ids"
-                f"current graph_db is: {graph_db.__class__.__name__}"
-            ),
-        )
     result = graph_db.get_user_names_by_memory_ids(memory_ids=request.memory_ids)
     if vector_db:
         prefs = []

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/configs/llm.py

@@ -116,6 +116,7 @@ class VLLMLLMConfig(BaseLLMConfig):
         default=False,
         description="Enable reasoning outputs from vLLM",
     )
+    extra_body: Any = Field(default=None, description="Extra options for API")
 
 
 class LLMConfigFactory(BaseConfig):

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/llms/vllm.py

@@ -111,13 +111,7 @@ class VLLMLLM(BaseLLM):
                 "temperature": kwargs.get("temperature", self.config.temperature),
                 "max_tokens": kwargs.get("max_tokens", self.config.max_tokens),
                 "top_p": kwargs.get("top_p", self.config.top_p),
-                "extra_body": {
-                    "chat_template_kwargs": {
-                        "enable_thinking": kwargs.get(
-                            "enable_thinking", self.config.enable_thinking
-                        )
-                    }
-                },
+                "extra_body": kwargs.get("extra_body", self.config.extra_body),
             }
             if kwargs.get("tools"):
                 completion_kwargs["tools"] = kwargs.get("tools")
@@ -175,13 +169,7 @@ class VLLMLLM(BaseLLM):
                 "max_tokens": kwargs.get("max_tokens", self.config.max_tokens),
                 "top_p": kwargs.get("top_p", self.config.top_p),
                 "stream": True,
-                "extra_body": {
-                    "chat_template_kwargs": {
-                        "enable_thinking": kwargs.get(
-                            "enable_thinking", self.config.enable_thinking
-                        )
-                    }
-                },
+                "extra_body": kwargs.get("extra_body", self.config.extra_body),
             }
 
             stream = self.client.chat.completions.create(**completion_kwargs)

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/mem_reader/multi_modal_struct.py

@@ -646,6 +646,7 @@ class MultiModalStructMemReader(SimpleStructMemReader):
 
         return None
 
+    @timed
     def _process_string_fine(
         self,
         fast_memory_items: list[TextualMemoryItem],
@@ -883,6 +884,7 @@ class MultiModalStructMemReader(SimpleStructMemReader):
             logger.error(f"[MultiModalFine] Error calling LLM for tool trajectory: {e}")
             return []
 
+    @timed
     def _process_tool_trajectory_fine(
         self, fast_memory_items: list[TextualMemoryItem], info: dict[str, Any], **kwargs
     ) -> list[TextualMemoryItem]:

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/mem_reader/read_multi_modal/multi_modal_parser.py

@@ -11,6 +11,7 @@ from memos.llms.base import BaseLLM
 from memos.log import get_logger
 from memos.memories.textual.item import SourceMessage, TextualMemoryItem
 from memos.types import MessagesType
+from memos.utils import timed
 
 from .assistant_parser import AssistantParser
 from .base import BaseMessageParser
@@ -120,6 +121,7 @@ class MultiModalParser:
         logger.warning(f"[MultiModalParser] Could not determine parser for message: {message}")
         return None
 
+    @timed
     def parse(
         self,
         message: MessagesType,
@@ -157,6 +159,7 @@ class MultiModalParser:
             logger.error(f"[MultiModalParser] Error parsing message: {e}")
             return []
 
+    @timed
     def parse_batch(
         self,
         messages: list[MessagesType],
@@ -182,6 +185,7 @@ class MultiModalParser:
             results.append(items)
         return results
 
+    @timed
     def process_transfer(
         self,
         source: SourceMessage,

{memoryos-2.0.5 → memoryos-2.0.6}/src/memos/mem_reader/read_skill_memory/process_skill_memory.py

@@ -36,6 +36,7 @@ from memos.templates.skill_mem_prompt import (
     TOOL_GENERATION_PROMPT,
 )
 from memos.types import MessageList
+from memos.utils import timed
 
 
 load_dotenv()
@@ -52,6 +53,9 @@ def _generate_content_by_llm(llm: BaseLLM, prompt_template: str, **kwargs) -> An
     try:
         prompt = prompt_template.format(**kwargs)
         response = llm.generate([{"role": "user", "content": prompt}])
+        if not response:
+            logger.warning("[PROCESS_SKILLS] LLM returned empty or invalid response")
+            return {} if "json" in prompt_template.lower() else ""
         if "json" in prompt_template.lower():
             response = response.replace("```json", "").replace("```", "").strip()
             return json.loads(response)
@@ -61,6 +65,7 @@ def _generate_content_by_llm(llm: BaseLLM, prompt_template: str, **kwargs) -> An
         return {} if "json" in prompt_template.lower() else ""
 
 
+@timed
 def _batch_extract_skills(
     task_chunks: dict[str, MessageList],
     related_memories_map: dict[str, list[TextualMemoryItem]],
@@ -94,6 +99,7 @@ def _batch_extract_skills(
     return results
 
 
+@timed
 def _batch_generate_skill_details(
     raw_skills_data: list[tuple[dict[str, Any], str, MessageList]],
     related_skill_memories_map: dict[str, list[TextualMemoryItem]],
@@ -436,6 +442,9 @@ def _extract_skill_memory_by_llm(
             skills_llm = os.getenv("SKILLS_LLM", None)
             llm_kwargs = {"model_name_or_path": skills_llm} if skills_llm else {}
             response_text = llm.generate(prompt, **llm_kwargs)
+            if not response_text:
+                logger.warning("[PROCESS_SKILLS] LLM returned empty or invalid response")
+                continue
             # Clean up response (remove Markdown code blocks if present)
             logger.info(f"[Skill Memory]: response_text {response_text}")
             response_text = response_text.strip()
@@ -561,6 +570,9 @@ def _extract_skill_memory_by_llm_md(
             skills_llm = os.getenv("SKILLS_LLM", None)
             llm_kwargs = {"model_name_or_path": skills_llm} if skills_llm else {}
             response_text = llm.generate(prompt, **llm_kwargs)
+            if not response_text:
+                logger.warning("[PROCESS_SKILLS] LLM returned empty or invalid response")
+                continue
             # Clean up response (remove Markdown code blocks if present)
             logger.info(f"[Skill Memory]: response_text {response_text}")
             response_text = response_text.strip()
@@ -641,26 +653,22 @@ def _rewrite_query(task_type: str, messages: MessageList, llm: BaseLLM, rewrite_
     )
     prompt = [{"role": "user", "content": prompt_content}]
 
-    # Call LLM to rewrite the query with retry logic
-    for attempt in range(3):
-        try:
-            response_text = llm.generate(prompt)
-            # Clean up response (remove any markdown formatting if present)
-            response_text = response_text.strip()
-            logger.info(f"[PROCESS_SKILLS] Rewritten query for task '{task_type}': {response_text}")
-            return response_text
-        except Exception as e:
+    # Call LLM to rewrite the query
+    try:
+        response_text = llm.generate(prompt)
+        # Clean up response (remove any markdown formatting if present)
+        if response_text and isinstance(response_text, str):
+            return response_text.strip()
+        else:
             logger.warning(
-                f"[PROCESS_SKILLS] LLM query rewrite failed (attempt {attempt + 1}): {e}"
+                "[PROCESS_SKILLS] LLM returned empty or invalid response, returning first message content"
             )
-            if attempt == 2:
-                logger.warning(
-                    "[PROCESS_SKILLS] LLM query rewrite failed after 3 retries, returning first message content"
-                )
-                return messages[0]["content"] if messages else ""
-
-    # Fallback (should not reach here due to return in exception handling)
-    return messages[0]["content"] if messages else ""
+            return messages[0]["content"] if messages else ""
+    except Exception as e:
+        logger.warning(
+            f"[PROCESS_SKILLS] LLM query rewrite failed: {e}, returning first message content"
+        )
+        return messages[0]["content"] if messages else ""
 
 
 @require_python_package(
@@ -751,6 +759,7 @@ def _delete_skills(
             logger.warning(f"Error deleting local file: {e}")
 
 
+@timed
 def _write_skills_to_file(
     skill_memory: dict[str, Any], info: dict[str, Any], skills_dir_config: dict[str, Any]
 ) -> str:
@@ -995,6 +1004,7 @@ def _get_skill_file_storage_location() -> str:
         return "LOCAL"
 
 
+@timed
 def process_skill_memory_fine(
     fast_memory_items: list[TextualMemoryItem],
     info: dict[str, Any],
@@ -1059,6 +1069,7 @@ def process_skill_memory_fine(
                 )
                 related_skill_memories_by_task[task_name] = []
 
+    @timed
     def _simple_extract():
         # simple extract skill memory, only one stage
         memories = []
@@ -1091,6 +1102,7 @@ def process_skill_memory_fine(
                     )
         return memories
 
+    @timed
     def _full_extract():
         # full extract skill memory, include two stage
         raw_extraction_results = _batch_extract_skills(