hy-memory 2.0.5.dev1__tar.gz → 2.0.5.dev3__tar.gz

{hy_memory-2.0.5.dev1/hy_memory.egg-info → hy_memory-2.0.5.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hy-memory
-Version: 2.0.5.dev1
+Version: 2.0.5.dev3
 Summary: HY Memory - Industrial-grade dual-system cognitive memory framework
 Author: alvinfei
 License-Expression: Apache-2.0

{hy_memory-2.0.5.dev1 → hy_memory-2.0.5.dev3}/hy_memory/client.py

@@ -2936,6 +2936,49 @@ class HyMemoryClient:
             include_runtime_diag=True,
         )
 
+    def import_memory_exact(
+        self,
+        content: str,
+        *,
+        memory_id: str,
+        user_id: str,
+        agent_id: str = "default_agent",
+        session_id: str = "default_session",
+        source: str = "default",
+        layer: Any = "l2_fact",
+        status: Any = "active",
+        owner: Optional[str] = None,
+        tags: Optional[List[str]] = None,
+        supersedes: Optional[List[str]] = None,
+        superseded_by: Optional[List[str]] = None,
+        is_latest: bool = True,
+        memory_at: Optional[datetime] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+        request_id: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """Import one existing semantic memory without extraction/reconcile."""
+        return self._loop_thread.run(
+            self.async_import_memory_exact(
+                content,
+                memory_id=memory_id,
+                user_id=user_id,
+                agent_id=agent_id,
+                session_id=session_id,
+                source=source,
+                layer=layer,
+                status=status,
+                owner=owner,
+                tags=tags,
+                supersedes=supersedes,
+                superseded_by=superseded_by,
+                is_latest=is_latest,
+                memory_at=memory_at,
+                metadata=metadata,
+                request_id=request_id,
+            ),
+            include_runtime_diag=True,
+        )
+
     def get_raw(self, raw_id: str) -> Optional[Dict[str, Any]]:
         """Read one SDK raw record through the explicit raw API."""
         return self._loop_thread.run(self.async_get_raw(raw_id))
@@ -3223,6 +3266,157 @@ class HyMemoryClient:
             raise RuntimeError("raw import changed the requested identity")
         return {"success": True, "raw_id": node_id, "request_id": request_id}
 
+    @_ensure_internal_loop
+    async def async_import_memory_exact(
+        self,
+        content: str,
+        *,
+        memory_id: str,
+        user_id: str,
+        agent_id: str = "default_agent",
+        session_id: str = "default_session",
+        source: str = "default",
+        layer: Any = "l2_fact",
+        status: Any = "active",
+        owner: Optional[str] = None,
+        tags: Optional[List[str]] = None,
+        supersedes: Optional[List[str]] = None,
+        superseded_by: Optional[List[str]] = None,
+        is_latest: bool = True,
+        memory_at: Optional[datetime] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+        request_id: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """Import a canonical memory node exactly, with no model invocation."""
+        from .models.memory import MemoryLayer, MemoryStatus
+        from .utils.source_filter import normalize_source
+
+        if not content or not memory_id or not user_id:
+            raise ValueError("content, memory_id and user_id are required")
+        normalized_layer = (
+            layer if isinstance(layer, MemoryLayer) else MemoryLayer.from_string(str(layer))
+        )
+        if normalized_layer == MemoryLayer.L1_RAW:
+            raise ValueError("use import_raw for L1_RAW")
+        normalized_status = (
+            status if isinstance(status, MemoryStatus) else MemoryStatus(str(status).lower())
+        )
+        request_id = request_id or str(uuid.uuid4())
+        replay = await self._claim_write_outcome(
+            request_id=request_id,
+            user_id=user_id,
+            agent_id=agent_id,
+            session_id=session_id,
+            operation="import_memory_exact",
+        )
+        if replay is not None:
+            return replay
+        try:
+            result = await self._async_import_memory_exact_impl(
+                content,
+                memory_id=memory_id,
+                user_id=user_id,
+                agent_id=agent_id,
+                session_id=session_id,
+                source=normalize_source(source),
+                layer=normalized_layer,
+                status=normalized_status,
+                owner=owner,
+                tags=tags or [],
+                supersedes=supersedes or [],
+                superseded_by=superseded_by or [],
+                is_latest=is_latest,
+                memory_at=memory_at,
+                metadata=metadata,
+                request_id=request_id,
+            )
+        except BaseException:
+            await self._mark_write_outcome_unknown(request_id)
+            raise
+        await self._finish_write_outcome(request_id, "succeeded", result)
+        return result
+
+    async def _async_import_memory_exact_impl(
+        self,
+        content: str,
+        *,
+        memory_id: str,
+        user_id: str,
+        agent_id: str,
+        session_id: str,
+        source: str,
+        layer,
+        status,
+        owner: Optional[str],
+        tags: List[str],
+        supersedes: List[str],
+        superseded_by: List[str],
+        is_latest: bool,
+        memory_at: Optional[datetime],
+        metadata: Optional[Dict[str, Any]],
+        request_id: str,
+    ) -> Dict[str, Any]:
+        from .models.memory import MemoryNode, SourceType
+
+        existing = await self._vector_store.get_by_id(memory_id)
+        if existing is not None:
+            identity = (
+                existing.content == content,
+                existing.user_id == user_id,
+                existing.agent_id == agent_id,
+                existing.session_id == session_id,
+                existing.source == source,
+                existing.layer == layer,
+                existing.status == status,
+                existing.owner == owner,
+                list(existing.tags or []) == tags,
+                list(existing.supersedes or []) == supersedes,
+                list(existing.superseded_by or []) == superseded_by,
+                existing.is_latest == is_latest,
+            )
+            if not all(identity):
+                raise ValueError("memory_id is already bound to different memory content")
+            return {
+                "success": True,
+                "memory_id": memory_id,
+                "memory_ids": [memory_id],
+                "request_id": request_id,
+                "replayed": True,
+            }
+
+        custom = dict(metadata or {})
+        custom["request_id"] = request_id
+        custom["input_kind"] = "memory_exact_import"
+        embedding = await self._embed_service.embed(content)
+        node = MemoryNode(
+            node_id=memory_id,
+            user_id=user_id,
+            agent_id=agent_id,
+            session_id=session_id,
+            source=source,
+            layer=layer,
+            content=content,
+            source_type=SourceType.EXPLICIT,
+            status=status,
+            owner=owner,
+            tags=list(tags),
+            supersedes=list(supersedes),
+            superseded_by=list(superseded_by),
+            is_latest=is_latest,
+            embedding=embedding,
+            memory_at=memory_at,
+            custom=custom,
+        )
+        stored_id = await self._vector_store.upsert(node)
+        if stored_id != memory_id:
+            raise RuntimeError("exact memory import changed the requested identity")
+        return {
+            "success": True,
+            "memory_id": memory_id,
+            "memory_ids": [memory_id],
+            "request_id": request_id,
+        }
+
     @_ensure_internal_loop
     async def async_get_raw(self, raw_id: str) -> Optional[Dict[str, Any]]:
         from .models.memory import MemoryLayer
@@ -3275,6 +3469,66 @@ class HyMemoryClient:
             "offset": offset,
         }
 
+    async def _search_explicit_raw_layer(
+        self,
+        query: str,
+        *,
+        user_ids: List[str],
+        agent_ids: List[str],
+        session_ids: List[str],
+        include_sources: List[str],
+        exclude_sources: List[str],
+        limit: int,
+        min_score: float,
+        created_after: Optional[float],
+        request_id: str,
+        t0: float,
+        query_embedding: Optional[List[float]],
+    ) -> Dict[str, Any]:
+        """Implement `search(..., layers=['l1_raw'])` without a second API."""
+        from .models.memory import MemoryLayer, MemoryStatus
+
+        embedding = query_embedding or await self._embed_service.embed(query)
+        fetch = min(max(limit * (10 if session_ids else 2), limit), 10_000)
+        hits = await self._vector_store.search(
+            query_embedding=embedding,
+            user_ids=user_ids,
+            agent_ids=agent_ids,
+            include_sources=include_sources,
+            exclude_sources=exclude_sources,
+            layers=[MemoryLayer.L1_RAW],
+            limit=fetch,
+            score_threshold=min_score,
+            status_filter=[MemoryStatus.ACTIVE],
+            only_latest=False,
+            created_after=created_after,
+        )
+        sessions = set(session_ids)
+        rows: List[Dict[str, Any]] = []
+        for hit in hits:
+            node = hit.get("node")
+            if node is None or node.layer != MemoryLayer.L1_RAW:
+                continue
+            if sessions and node.session_id not in sessions:
+                continue
+            row = self._raw_node_to_public(node)
+            rows.append(
+                {
+                    "memory_id": row.pop("raw_id"),
+                    **row,
+                    "layer": MemoryLayer.L1_RAW.value,
+                    "status": MemoryStatus.ACTIVE.value,
+                    "score": round(float(hit.get("score", 0.0)), 4),
+                }
+            )
+            if len(rows) >= limit:
+                break
+        return {
+            "request_id": request_id,
+            "memories": {"schema": [], "proactive": [], "normal": rows},
+            "elapsed_ms": round((time.perf_counter() - t0) * 1000, 2),
+        }
+
     @staticmethod
     def _public_search_views(
         result_memories: List[Dict[str, Any]],
@@ -3935,6 +4189,22 @@ class HyMemoryClient:
         effective_search_layers = [MemoryLayer.L1_RAW] if lite_mode else layers
         effective_schema_limit = 0 if lite_mode else schema_limit
         effective_intention_limit = 0 if lite_mode else intention_limit
+        normalized_search_layers = self._normalize_layers(effective_search_layers)
+        if normalized_search_layers == [MemoryLayer.L1_RAW]:
+            return await self._search_explicit_raw_layer(
+                effective_target_query,
+                user_ids=effective_user_ids,
+                agent_ids=effective_agent_ids,
+                session_ids=effective_session_ids,
+                include_sources=effective_include_sources,
+                exclude_sources=effective_exclude_sources,
+                limit=limit,
+                min_score=min_score,
+                created_after=effective_created_after,
+                request_id=request_id,
+                t0=t0,
+                query_embedding=query_embedding,
+            )
 
         read_req = ReadRequest(
             query=effective_target_query,
@@ -3955,7 +4225,7 @@ class HyMemoryClient:
             layers=(
                 [
                     layer.value
-                    for layer in self._normalize_layers(effective_search_layers)
+                    for layer in normalized_search_layers
                 ]
                 if effective_search_layers else None
             ),
@@ -3980,7 +4250,7 @@ class HyMemoryClient:
         _wanted_layers = (
             {
                 layer.value
-                for layer in self._normalize_layers(effective_search_layers)
+                for layer in normalized_search_layers
             }
             if effective_search_layers else None
         )

{hy_memory-2.0.5.dev1 → hy_memory-2.0.5.dev3/hy_memory.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hy-memory
-Version: 2.0.5.dev1
+Version: 2.0.5.dev3
 Summary: HY Memory - Industrial-grade dual-system cognitive memory framework
 Author: alvinfei
 License-Expression: Apache-2.0

{hy_memory-2.0.5.dev1 → hy_memory-2.0.5.dev3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hy-memory"
-version = "2.0.5.dev1"
+version = "2.0.5.dev3"
 description = "HY Memory - Industrial-grade dual-system cognitive memory framework"
 readme = "README.md"
 requires-python = ">=3.9"