pycontextdb 0.4.3__tar.gz → 0.4.4__tar.gz

{pycontextdb-0.4.3 → pycontextdb-0.4.4}/CHANGELOG.md

@@ -2,6 +2,12 @@
 
 ## Unreleased
 
+## 0.4.4 — 2026-09-20
+
+- PostgreSQL sibling composition now uses an ordered user-plus-entity cache
+  index, so a scoped recall does not scan other users that share the same
+  entity key.
+
 ## 0.4.3 — 2026-09-20
 
 - High-cardinality Postgres recall now indexes cached IDs by user and entity,

{pycontextdb-0.4.3 → pycontextdb-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: pycontextdb
-Version: 0.4.3
+Version: 0.4.4
 Summary: The unified context layer for AI agents — memory you can act on without treating wishes as facts
 Project-URL: Homepage, https://github.com/atomsai/contextdb
 Project-URL: Documentation, https://contextdb.dev

{pycontextdb-0.4.3 → pycontextdb-0.4.4}/contextdb/__init__.py

@@ -41,7 +41,7 @@ from contextdb.core.policy import TrustPolicy
 from contextdb.pool import ContextDBPool
 from contextdb.utils.embeddings import EmbeddingProvider
 
-__version__ = "0.4.3"
+__version__ = "0.4.4"
 
 __all__ = [
     "Clock",

{pycontextdb-0.4.3 → pycontextdb-0.4.4}/contextdb/store/postgres_store.py

@@ -282,6 +282,10 @@ class PostgresStore(BaseStore):
         self._index_items: dict[str, MemoryItem] = {}
         self._index_ids_by_user: dict[str | None, set[str]] = {}
         self._index_ids_by_entity: dict[str, dict[str, None]] = {}
+        self._index_ids_by_user_entity: dict[
+            tuple[str | None, str],
+            dict[str, None],
+        ] = {}
         self._embedding_dim = embedding_dim
         self._embedding_model_id = embedding_model_id
         self._index_loaded = False
@@ -389,6 +393,10 @@ class PostgresStore(BaseStore):
         self._index_ids_by_user.setdefault(item.user_id, set()).add(item.id)
         if item.entity_key is not None:
             self._index_ids_by_entity.setdefault(item.entity_key, {})[item.id] = None
+            self._index_ids_by_user_entity.setdefault(
+                (item.user_id, item.entity_key),
+                {},
+            )[item.id] = None
 
     def _discard_index_item(self, memory_id: str) -> None:
         item = self._index_items.pop(memory_id, None)
@@ -405,11 +413,22 @@ class PostgresStore(BaseStore):
                 entity_ids.pop(memory_id, None)
                 if not entity_ids:
                     self._index_ids_by_entity.pop(item.entity_key, None)
+            scoped_entity_ids = self._index_ids_by_user_entity.get(
+                (item.user_id, item.entity_key)
+            )
+            if scoped_entity_ids is not None:
+                scoped_entity_ids.pop(memory_id, None)
+                if not scoped_entity_ids:
+                    self._index_ids_by_user_entity.pop(
+                        (item.user_id, item.entity_key),
+                        None,
+                    )
 
     def _clear_index_items(self) -> None:
         self._index_items.clear()
         self._index_ids_by_user.clear()
         self._index_ids_by_entity.clear()
+        self._index_ids_by_user_entity.clear()
 
     def _scope_sql(self, user_id: str | None = None) -> tuple[str, list[Any]]:
         clauses: list[str] = []
@@ -1101,8 +1120,17 @@ class PostgresStore(BaseStore):
             await self._ensure_index()
         excluded = exclude_ids or set()
         items: list[MemoryItem] = []
+        scope = self._resolve_scope(user_id)
         for entity_key in dict.fromkeys(entity_keys):
-            for memory_id in self._index_ids_by_entity.get(entity_key, {}):
+            entity_ids = (
+                self._index_ids_by_entity.get(entity_key, {})
+                if scope is None
+                else self._index_ids_by_user_entity.get(
+                    (scope, entity_key),
+                    {},
+                )
+            )
+            for memory_id in entity_ids:
                 item = self._index_items.get(memory_id)
                 if item is None:
                     continue

{pycontextdb-0.4.3 → pycontextdb-0.4.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pycontextdb"
-version = "0.4.3"
+version = "0.4.4"
 description = "The unified context layer for AI agents — memory you can act on without treating wishes as facts"
 readme = "README.md"
 license = "Apache-2.0"

{pycontextdb-0.4.3 → pycontextdb-0.4.4}/tests/unit/test_pg_sql.py

@@ -98,6 +98,16 @@ async def test_list_by_entities_filters_and_copies_only_the_requested_limit(
         )
         for index in range(20)
     ]
+    foreign = [
+        MemoryItem(
+            content=f"foreign-{index}",
+            entity_key="entity-a",
+            user_id="user-2",
+            tenant_id="tenant-1",
+            agent_id="agent-1",
+        )
+        for index in range(1_000)
+    ]
     future = MemoryItem(
         content="future",
         entity_key="entity-a",
@@ -106,13 +116,15 @@ async def test_list_by_entities_filters_and_copies_only_the_requested_limit(
         agent_id="agent-1",
         valid_from=now + timedelta(days=1),
     )
-    indexed = [*entity_b, future, *entity_a]
+    indexed = [*foreign, *entity_b, future, *entity_a]
     for item in indexed:
         store._cache_index_item(item)
     store._index_loaded = True
 
     original_model_copy = MemoryItem.model_copy
+    original_scope_allows = store._item_scope_allows
     copies = 0
+    scope_checks = 0
 
     def counted_model_copy(
         self: MemoryItem,
@@ -125,6 +137,13 @@ async def test_list_by_entities_filters_and_copies_only_the_requested_limit(
         return original_model_copy(self, update=update, deep=deep)
 
     monkeypatch.setattr(MemoryItem, "model_copy", counted_model_copy)
+
+    def counted_scope_allows(item: MemoryItem, user_id: str | None) -> bool:
+        nonlocal scope_checks
+        scope_checks += 1
+        return original_scope_allows(item, user_id)
+
+    monkeypatch.setattr(store, "_item_scope_allows", counted_scope_allows)
     results = await store.list_by_entities(
         ["entity-a", "entity-b"],
         user_id="user-1",
@@ -135,6 +154,7 @@ async def test_list_by_entities_filters_and_copies_only_the_requested_limit(
 
     assert len(results) == 5
     assert copies == 5
+    assert scope_checks == 7
     assert all(item.entity_key == "entity-a" for item in results)
     assert entity_a[0].id not in {item.id for item in results}
     results[0].metadata["nested"].append(99)
@@ -238,6 +258,12 @@ def test_postgres_secondary_indexes_follow_cache_lifecycle() -> None:
     }
     assert list(store._index_ids_by_entity["entity-a"]) == [first.id]
     assert list(store._index_ids_by_entity["entity-b"]) == [second.id]
+    assert list(store._index_ids_by_user_entity["user-1", "entity-a"]) == [
+        first.id
+    ]
+    assert list(store._index_ids_by_user_entity["user-2", "entity-b"]) == [
+        second.id
+    ]
 
     replacement = first.model_copy(
         update={"user_id": "user-3", "entity_key": "entity-c"}
@@ -245,18 +271,24 @@ def test_postgres_secondary_indexes_follow_cache_lifecycle() -> None:
     store._cache_index_item(replacement)
     assert "user-1" not in store._index_ids_by_user
     assert "entity-a" not in store._index_ids_by_entity
+    assert ("user-1", "entity-a") not in store._index_ids_by_user_entity
     assert store._index_ids_by_user["user-3"] == {first.id}
     assert list(store._index_ids_by_entity["entity-c"]) == [first.id]
+    assert list(store._index_ids_by_user_entity["user-3", "entity-c"]) == [
+        first.id
+    ]
 
     store._discard_index_item(first.id)
     assert "user-3" not in store._index_ids_by_user
     assert "entity-c" not in store._index_ids_by_entity
+    assert ("user-3", "entity-c") not in store._index_ids_by_user_entity
     assert first.id not in store._index_items
 
     store._clear_index_items()
     assert store._index_items == {}
     assert store._index_ids_by_user == {}
     assert store._index_ids_by_entity == {}
+    assert store._index_ids_by_user_entity == {}
 
 
 @pytest.mark.asyncio