unitysvc-core 0.2.28__tar.gz → 0.2.29__tar.gz

{unitysvc_core-0.2.28/src/unitysvc_core.egg-info → unitysvc_core-0.2.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitysvc-core
-Version: 0.2.28
+Version: 0.2.29
 Summary: Shared data models and validation helpers for UnitySVC packages
 Author-email: Bo Peng <bo.peng@unitysvc.com>
 Maintainer-email: Bo Peng <bo.peng@unitysvc.com>

{unitysvc_core-0.2.28 → unitysvc_core-0.2.29}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "unitysvc-core"
-version = "0.2.28"
+version = "0.2.29"
 description = "Shared data models and validation helpers for UnitySVC packages"
 readme = "README.md"
 authors = [{ name = "Bo Peng", email = "bo.peng@unitysvc.com" }]

{unitysvc_core-0.2.28 → unitysvc_core-0.2.29}/src/unitysvc_core/models/pricing.py

@@ -342,6 +342,23 @@ def upstream_price_from_description(text: str | None) -> Decimal | None:
     return _blend(by_label["in"], by_label["out"]) * scale
 
 
+def _zero_rate(figure: EffectivePrice, description: str | None) -> EffectivePrice | None:
+    """A zero-rate price's effective price: free, or what the customer pays upstream.
+
+    A ``0`` with a description is a bring-your-own-key price: the customer pays
+    the upstream provider, at the rate the description states. A description
+    that states no parseable rate, or $0/$0 (what generators write when they
+    lack the provider's price), is unknown. A ``0`` with no description is
+    genuinely free.
+    """
+    if figure.amount != 0 or not description:
+        return figure
+    upstream = upstream_price_from_description(description)
+    if upstream is None or upstream == 0:
+        return None
+    return EffectivePrice(amount=upstream, unit=UNIT_TOKENS, source="upstream")
+
+
 # ============================================================================
 # Pricing Models - Discriminated Union for type-safe pricing validation
 # ============================================================================
@@ -475,14 +492,19 @@ class TokenPriceData(BasePriceData):
         return {"one_million_tokens": 1_000_000, "one_thousand_tokens": 1_000, "one_token": 1}[self.type]
 
     def effective_price(self) -> EffectivePrice | None:
-        """``(input + 4*output) / 5`` when priced separately, else ``price``; per 1M tokens."""
+        """``(input + 4*output) / 5`` when priced separately, else ``price``; per 1M tokens.
+
+        A zero rate with a description is a bring-your-own-key price: see
+        ``_zero_rate``.
+        """
         if self.input is not None and self.output is not None:
             per_unit = _blend(Decimal(self.input), Decimal(self.output))
         elif self.price is not None:
             per_unit = Decimal(self.price)
         else:
             return None
-        return _per_normalized_unit(self.type, per_unit)
+        figure = _per_normalized_unit(self.type, per_unit)
+        return None if figure is None else _zero_rate(figure, self.description)
 
     def calculate_cost(
         self,
@@ -931,19 +953,10 @@ class ConstantPriceData(BasePriceData):
     def effective_price(self) -> EffectivePrice | None:
         """``price`` per request.
 
-        A ``0`` with a description is a bring-your-own-key price: the customer
-        pays the upstream provider, at the rate the description states. A
-        description that states no parseable rate, or $0/$0 (what generators
-        write when they lack the provider's price), is unknown. A ``0`` with no
-        description is genuinely free.
+        A ``0`` with a description is a bring-your-own-key price: see
+        ``_zero_rate``.
         """
-        amount = Decimal(self.price)
-        if amount != 0 or not self.description:
-            return _rate(amount, UNIT_REQUEST)
-        upstream = upstream_price_from_description(self.description)
-        if upstream is None or upstream == 0:
-            return None
-        return EffectivePrice(amount=upstream, unit=UNIT_TOKENS, source="upstream")
+        return _zero_rate(_rate(Decimal(self.price), UNIT_REQUEST), self.description)
 
     def calculate_cost(
         self,

{unitysvc_core-0.2.28 → unitysvc_core-0.2.29/src/unitysvc_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitysvc-core
-Version: 0.2.28
+Version: 0.2.29
 Summary: Shared data models and validation helpers for UnitySVC packages
 Author-email: Bo Peng <bo.peng@unitysvc.com>
 Maintainer-email: Bo Peng <bo.peng@unitysvc.com>

{unitysvc_core-0.2.28 → unitysvc_core-0.2.29}/tests/test_effective_price.py

@@ -112,6 +112,33 @@ class TestByokConstant:
         assert ep(priced) == rate("0.02", "request")
 
 
+class TestByokZeroRateTokens:
+    """Generators also write BYOK as a zero token rate, e.g. Mistral models."""
+
+    MISTRAL = (
+        "Free ~ BYOK | usage is billed by Mistral AI directly at $1.5 / $7.5 / $0.15 per 1M input/output/cached tokens"
+    )
+
+    def test_zero_split_rate_reads_description(self) -> None:
+        byok = {"type": "one_million_tokens", "input": "0", "output": "0", "cached_input": "0"}
+        figure = ep({**byok, "description": self.MISTRAL})
+        assert figure == rate("6.3", "one_million_tokens", "upstream")
+
+    def test_zero_unified_rate_reads_description(self) -> None:
+        figure = ep({"type": "one_thousand_tokens", "price": "0", "description": HF_UPSTREAM})
+        assert figure == rate("3.8", "one_million_tokens", "upstream")
+
+    def test_zero_rate_without_description_is_free(self) -> None:
+        assert ep({"type": "one_million_tokens", "input": "0", "output": "0"}) == rate("0", "one_million_tokens")
+
+    def test_zero_rate_with_unparseable_description_is_unknown(self) -> None:
+        assert ep({"type": "one_million_tokens", "price": "0", "description": "Bring your own key"}) is None
+
+    def test_nonzero_rate_ignores_description(self) -> None:
+        priced = {**TOKENS, "description": self.MISTRAL}
+        assert ep(priced) == rate("1.8", "one_million_tokens")
+
+
 class TestUpstreamPriceFromDescription:
     @pytest.mark.parametrize(
         ("text", "expected"),