metergraph-core 0.2.0__tar.gz → 0.2.1__tar.gz

{metergraph_core-0.2.0/src/metergraph_core.egg-info → metergraph_core-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metergraph-core
-Version: 0.2.0
+Version: 0.2.1
 Summary: Reusable MeterGraph catalog and deterministic billing engine
 License-Expression: Apache-2.0
 Requires-Python: >=3.10

{metergraph_core-0.2.0 → metergraph_core-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "metergraph-core"
-version = "0.2.0"
+version = "0.2.1"
 description = "Reusable MeterGraph catalog and deterministic billing engine"
 readme = "README.md"
 license = "Apache-2.0"

{metergraph_core-0.2.0 → metergraph_core-0.2.1}/src/metergraph_core/catalog.py

@@ -69,6 +69,7 @@ class Alias:
     canonical_id: str
     pricing_channel: str
     rules: Mapping[str, Any]
+    publisher: str | None = None
 
 
 @dataclass(frozen=True, slots=True)
@@ -270,6 +271,34 @@ class CatalogSnapshot:
                     return price
         return None
 
+    def infer_direct_channel(self, model: Any) -> str | None:
+        """Return the sole direct billing channel known for ``model``.
+
+        This supports captures that preserve a model identity but omit the
+        source provider. It uses the catalog's canonical model publisher and
+        returns ``None`` whenever model identity or direct channel is
+        ambiguous.
+        """
+        if not isinstance(model, str) or not model.strip():
+            return None
+        model_key = model.strip().lower()
+        canonical_ids = {
+            alias.canonical_id
+            for (known_model, _channel), alias in self._deployment_aliases.items()
+            if known_model == model_key
+        }
+        if len(canonical_ids) != 1:
+            return None
+        channels: set[str] = set()
+        for canonical_id in canonical_ids:
+            for alias in self._deployment_aliases.values():
+                if alias.canonical_id != canonical_id:
+                    continue
+                channel = direct_channel_for_provider(alias.publisher)
+                if channel is not None and (canonical_id, channel) in self._prices:
+                    channels.add(channel)
+        return next(iter(channels)) if len(channels) == 1 else None
+
     def resolve_price(
         self, *, model: Any, channel: Any, at: datetime
     ) -> ResolvedPrice | None:

{metergraph_core-0.2.0 → metergraph_core-0.2.1}/src/metergraph_core/loader.py

@@ -84,6 +84,10 @@ class LoadedCatalog:
             (_normalize_provider(provider), model_id), model_id
         )
 
+    def infer_direct_channel(self, model: Any) -> str | None:
+        """Infer a direct pricing channel only when the catalog is unambiguous."""
+        return self.snapshot.infer_direct_channel(model)
+
     def price(
         self,
         *,
@@ -150,6 +154,7 @@ def parse_catalog(
         canonical = str(entry.get("canonical_id") or "")
         if not canonical:
             raise CatalogError("model entry missing canonical_id")
+        publisher = str(entry.get("publisher") or "").strip().lower() or None
         for alias in entry.get("aliases") or []:
             provider = str(alias.get("provider") or "").lower()
             name = str(alias.get("alias") or "").lower()
@@ -167,6 +172,7 @@ def parse_catalog(
                     canonical_id=canonical,
                     pricing_channel=channel,
                     rules=_freeze(alias.get("rules") or {}),
+                    publisher=publisher,
                 )
         seen_windows: list[tuple[str, str, datetime, datetime | None]] = []
         for price in entry.get("prices") or []:

{metergraph_core-0.2.0 → metergraph_core-0.2.1/src/metergraph_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metergraph-core
-Version: 0.2.0
+Version: 0.2.1
 Summary: Reusable MeterGraph catalog and deterministic billing engine
 License-Expression: Apache-2.0
 Requires-Python: >=3.10