resolvekit 0.1.7__tar.gz → 0.1.8__tar.gz

{resolvekit-0.1.7 → resolvekit-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: resolvekit
-Version: 0.1.7
+Version: 0.1.8
 Summary: Entity and place resolution system that maps messy place/entity strings and codes to canonical entities
 Keywords: entity-resolution,geocoding,place-names,data-commons,iso-codes,offline,disambiguation,normalization
 Author: Jorge Rivera

{resolvekit-0.1.7 → resolvekit-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "resolvekit"
-version = "0.1.7"
+version = "0.1.8"
 description = "Entity and place resolution system that maps messy place/entity strings and codes to canonical entities"
 requires-python = ">=3.12"
 

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/api/containment_api.py

@@ -91,10 +91,12 @@ class ContainmentAPI:
                 break
             frontier = next_frontier
 
-        # Hydrate — drop entities the store can't resolve.
+        # Hydrate in one batch — bulk bypasses the per-entity SQLite LRU cache, which is
+        # acceptable here: within()'s result set is not reused across calls.
+        hydrated = self._runner.bulk_get_entities(result_ids)
         records: list[EntityRecord] = []
-        for eid in result_ids:
-            entity = self._runner.get_entity(eid)
+        for eid in result_ids:  # preserve BFS-collected order pre-sort
+            entity = hydrated.get(eid)  # None-on-missing: absent IDs skipped
             if entity is None:
                 continue
             if entity_type is not None and entity.entity_type not in entity_type:

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/engine/interfaces.py

@@ -132,6 +132,10 @@ class ResolverBackend(Protocol):
         """Look up a fully hydrated entity by ID."""
         ...
 
+    def bulk_get_entities(self, entity_ids: list[str]) -> dict[str, EntityRecord]:
+        """Fetch multiple entities by ID; missing IDs are omitted from the result."""
+        ...
+
     def lookup_code(
         self,
         system: str,

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/engine/multi_runner.py

@@ -192,6 +192,10 @@ class MultiPackRunner:
         """Fetch a fully hydrated entity record from any backing store."""
         return self._view.get_entity(entity_id)
 
+    def bulk_get_entities(self, entity_ids: list[str]) -> dict[str, EntityRecord]:
+        """Fetch multiple entities by ID; missing IDs are omitted from the result."""
+        return self._view.bulk_get_entities(entity_ids)
+
     def normalize_code_value(
         self, system: str, value: str, *, pack_filter: frozenset[str] | None = None
     ) -> str:

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/engine/runner.py

@@ -178,6 +178,10 @@ class PipelineRunner:
         """Fetch a fully hydrated entity record from the configured store."""
         return self._view.get_entity(entity_id)
 
+    def bulk_get_entities(self, entity_ids: list[str]) -> dict[str, EntityRecord]:
+        """Fetch multiple entities by ID; missing IDs are omitted from the result."""
+        return self._view.bulk_get_entities(entity_ids)
+
     def normalize_code_value(
         self, system: str, value: str, *, pack_filter: frozenset[str] | None = None
     ) -> str:

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/store/interface.py

@@ -176,6 +176,15 @@ class EntityStore(ABC):
         """
         return frozenset()
 
+    def relation_types(self) -> frozenset[str] | None:
+        """Return the distinct relation types stored, or None if unknown.
+
+        None means "this store cannot characterize its relation types, so callers
+        must not prune it" — the default. SQL backends override with a concrete set
+        (possibly empty, meaning known to hold no relations).
+        """
+        return None
+
     def close(self) -> None:  # noqa: B027
         """Release resources held by this store.
 

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/store/sqlite.py

@@ -494,6 +494,17 @@ class SQLiteEntityStore(EntityStore):
             rows = conn.execute("SELECT DISTINCT system FROM codes").fetchall()
             return frozenset(r["system"] for r in rows)
 
+    @override
+    def relation_types(self) -> frozenset[str]:
+        with self._connection() as conn:
+            try:
+                rows = conn.execute(
+                    "SELECT DISTINCT relation_type FROM relations"
+                ).fetchall()
+            except sqlite3.OperationalError:
+                return frozenset()  # legacy DB with no relations table
+            return frozenset(r["relation_type"] for r in rows)
+
     def lookup_name_exact(
         self, value_norm: str, name_kinds: set[str] | None = None
     ) -> list[str]:

{resolvekit-0.1.7 → resolvekit-0.1.8}/src/resolvekit/core/store/store_view.py

@@ -23,6 +23,9 @@ class StoreView:
 
     def __init__(self, stores: list[tuple[str | None, EntityStore]]) -> None:
         self._stores = stores
+        self._relation_type_index: list[frozenset[str] | None] = [
+            store.relation_types() for _, store in self._stores
+        ]
 
     # ------------------------------------------------------------------
     # Entity fetch — first non-None wins; no dedup needed
@@ -36,6 +39,18 @@ class StoreView:
                 return entity
         return None
 
+    def bulk_get_entities(self, entity_ids: list[str]) -> dict[str, EntityRecord]:
+        """Return entities for *entity_ids*, first store wins per ID; missing IDs omitted."""
+        result: dict[str, EntityRecord] = {}
+        remaining = list(entity_ids)
+        for _, store in self._stores:
+            if not remaining:
+                break
+            found = store.bulk_get_entities(remaining)
+            result.update(found)
+            remaining = [eid for eid in remaining if eid not in result]
+        return result
+
     # ------------------------------------------------------------------
     # Lookups — dedup by entity_id, preserve first-seen order
     # ------------------------------------------------------------------
@@ -118,7 +133,10 @@ class StoreView:
         """
         seen: set[str] = set()
         result: list[str] = []
-        for _, store in self._stores:
+        for i, (_, store) in enumerate(self._stores):
+            rv = self._relation_type_index[i]
+            if rv is not None and relation_type not in rv:
+                continue
             for eid in store.get_reverse_relations(
                 entity_id, relation_type, as_of=as_of
             ):
@@ -136,7 +154,10 @@ class StoreView:
     ) -> frozenset[str]:
         """Return target entity IDs for relations active on *as_of*, unioned."""
         result: set[str] = set()
-        for _, store in self._stores:
+        for i, (_, store) in enumerate(self._stores):
+            rv = self._relation_type_index[i]
+            if rv is not None and relation_type not in rv:
+                continue
             result.update(store.get_relations_as_of(entity_id, relation_type, as_of))
         return frozenset(result)