yomemoai-mcp 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

yomemoai_mcp/client.py

@@ -1,4 +1,5 @@
 import base64
+from cryptography.hazmat.primitives.ciphers.modes import GCM
 import json
 import logging
 import time
@@ -104,7 +105,7 @@ class MemoClient:
             tag = combined_data[-16:]
             ciphertext = combined_data[12:-16]
 
-            cipher = Cipher(algorithms.AES(aes_key), modes.GCM(
+            cipher = Cipher[GCM](algorithms.AES(aes_key), modes.GCM(
                 nonce, tag), backend=default_backend())
             decryptor = cipher.decryptor()
             return decryptor.update(ciphertext) + decryptor.finalize()
@@ -184,6 +185,10 @@ class MemoClient:
         resp.raise_for_status()
 
         memories = resp.json().get("data", [])
+        # Backend may return `data: null` when there are no memories; normalize to empty list.
+        if memories is None:
+            memories = []
+
         for m in memories:
             try:
                 decrypted = self.unpack_and_decrypt(m["content"])

yomemoai_mcp/server.py

@@ -1,10 +1,17 @@
 import asyncio
 import json
 import logging
+import os
 import sys
+from typing import Optional
+
 from mcp.server.fastmcp import FastMCP
 from .client import MemoClient, MemoRequestError
-import os
+
+if "--version" in sys.argv or "-version" in sys.argv:
+    from importlib.metadata import version
+    print(version("yomemoai-mcp"))
+    sys.exit(0)
 
 from dotenv import load_dotenv
 load_dotenv()
@@ -54,21 +61,27 @@ def _format_payload(payload: dict) -> dict:
 
 
 @mcp.tool()
-async def save_memory(content: str, handle: str = "general", description: str = "") -> str:
+async def save_memory(content: str, handle: str = "general", description: str = "", metadata: dict = {}) -> str:
     """
-    Store important information as a permanent memory. You should call this proactively when you
-    detect user preferences, important decisions, reusable logic, or context worth recalling later—
-    not only when the user explicitly says "remember" or "save". After saving successfully, reply with ✓.
-
-    :param content: The actual text/information to be remembered. Be concise but maintain context.
-    :param handle: A short, unique category or tag (e.g., 'work', 'personal', 'project-x'). Defaults to 'general'.
-    :param description: A brief, non-sensitive summary or tag for this memory (helps future identification and search).
+    Archives a high-density knowledge asset using the Semantic Fingerprint Protocol.
+    Proactively triggered by the AMP (Autonomous Persistence Trigger) during 'Moments of Truth' 
+    such as strategic decisions, fact updates, or logic finalization.
+
+    :param content: The actual high-density factual information, decision logic, or SOP. 
+                   Maintain context while stripping conversational noise.
+    :param handle: Categorical routing based on Layer ID (L1-L5) or specific project Name. 
+                  Helps in contextual retrieval (PRT).
+    :param description: A brief, non-sensitive summary or high-level tag for safe indexing and search. 
+                        STRICT CONSTRAINT: Must NOT contain specific transactional details, 
+                        PII, or sensitive affairs to prevent information leakage.
+    :param metadata: MANDATORY. Stores high-dimensional tag data under the 'semantic_fingerprint' key, 
+                    including ELAP metrics, ontology mode, and engineering VCS status.
     """
     logger.debug(
         f"save_memory called: handle={handle}, description={description}, content_length={len(content)}")
     try:
         result = client.add_memory(
-            content, handle=handle, description=description)
+            content, handle=handle, description=description,metadata=metadata)
         logger.debug(f"add_memory response: {result}")
 
         if "memory_id" not in result:
@@ -103,13 +116,17 @@ async def save_memory(content: str, handle: str = "general", description: str =
 
 
 @mcp.tool()
-async def load_memories(handle: str = None) -> str:
+async def load_memories(handle: Optional[str] = None) -> str:
     """
     Retrieve previously stored memories. Call this when the user asks what you remember, or when you
     need historical context (preferences, past decisions, project details) to answer accurately.
 
     :param handle: Optional filter. If the user specifies a category (e.g., 'work', 'project-x'),
                    use the relevant handle; otherwise omit to load across handles.
+
+    The output includes each memory's handle, idempotent key, and content so that callers can
+    later use the same idempotent key with save_memory to update existing memories instead of
+    creating duplicates.
     """
     logger.debug(f"load_memories called: handle={handle}")
     try:
@@ -123,9 +140,16 @@ async def load_memories(handle: str = None) -> str:
 
         output = ["### Retrieved Memories:"]
         for m in memories:
-            timestamp = m.get('created_at', 'N/A')
+            handle_value = m.get("handle")
+            idempotent_key = m.get("idempotent_key") or "N/A"
+            content = m.get("content")
+            metadata = m.get("metadata")
             output.append(
-                f"Handle: [{m.get('handle')}]\nContent: {m.get('content')}\n---"
+                "Handle: [{handle}]\nIdempotent Key: {key}\nContent: {content}\n---".format(
+                    handle=handle_value,
+                    key=idempotent_key,
+                    content=content,
+                    metadata=json.dumps(metadata, ensure_ascii=False) if metadata else "{}")
             )
         logger.info(f"Successfully loaded {len(memories)} memories")
         return "\n".join(output)

{yomemoai_mcp-0.1.2.dist-info → yomemoai_mcp-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yomemoai-mcp
-Version: 0.1.2
+Version: 0.1.4
 Summary: Model Context Protocol (MCP) server for YoMemoAI - enables AI assistants to save and retrieve encrypted memories
 Project-URL: Homepage, https://github.com/yomemoai/python-yomemo-mcp
 Project-URL: Documentation, https://github.com/yomemoai/python-yomemo-mcp#readme

yomemoai_mcp-0.1.4.dist-info/RECORD

@@ -0,0 +1,9 @@
+yomemoai_mcp/__init__.py,sha256=gDi0t9iKR423OZlwrrtwDaED3ck06XMe_y8mcxisBl0,95
+yomemoai_mcp/client.py,sha256=sXfHcU6wgUT7pDavsxU5HBAwqf9VSX1s3I4PXdzo12U,7177
+yomemoai_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+yomemoai_mcp/server.py,sha256=evZSzZuq7OaluF6OLUzsdK-jDMnPMHl9YSf5M78CMtg,6283
+yomemoai_mcp-0.1.4.dist-info/METADATA,sha256=owu--8SQKkQecwqZLfQayqm92OcC-r409iH4DleVi2I,7551
+yomemoai_mcp-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+yomemoai_mcp-0.1.4.dist-info/entry_points.txt,sha256=uV0479NK4ppOUV1T9VSwFTLi87F4deaRjbL8s43R8uI,100
+yomemoai_mcp-0.1.4.dist-info/licenses/LICENSE,sha256=UVt2r_C3iy7rNKJAj_EcLcmcZdXONxJ01oJNsIEqTfw,1066
+yomemoai_mcp-0.1.4.dist-info/RECORD,,

yomemoai_mcp-0.1.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-yomemoai_mcp/__init__.py,sha256=gDi0t9iKR423OZlwrrtwDaED3ck06XMe_y8mcxisBl0,95
-yomemoai_mcp/client.py,sha256=oZNfRW3ppfSRkWvG3ej-w56lUyfKsc6jOjvV_IY4JxM,6960
-yomemoai_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-yomemoai_mcp/server.py,sha256=K4-yj2KoBrcKFc9-C8v3NcG87sBpUDde1giHJamwho8,5093
-yomemoai_mcp-0.1.2.dist-info/METADATA,sha256=clSl9U0TcQlC602gJlxFZFfWz8epgWXxsHsKZV_2glg,7551
-yomemoai_mcp-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-yomemoai_mcp-0.1.2.dist-info/entry_points.txt,sha256=uV0479NK4ppOUV1T9VSwFTLi87F4deaRjbL8s43R8uI,100
-yomemoai_mcp-0.1.2.dist-info/licenses/LICENSE,sha256=UVt2r_C3iy7rNKJAj_EcLcmcZdXONxJ01oJNsIEqTfw,1066
-yomemoai_mcp-0.1.2.dist-info/RECORD,,