"jdSrSSKJr SSKrSSKJr SSKJr SSKJr SSK J r SSK J r S S K Jr S S KJr \(aSS KJr S S KJr S SKJr S SKJrJr SSjrSSSS.SSjjrSSS.SSjjrSSSS.SSjjrSSSS.SSjjrSSjrg)zShared [genai-prices](https://github.com/pydantic/genai-prices) helpers: best-effort cost calculation, the provider lookup order used for usage extraction, and the context window lookup.) annotationsN)Iterator)datetime) TYPE_CHECKING) calc_price get_snapshot)CostCalculationFailedWarning) UserError)PriceCalculation) ModelResponse) AbstractModel) RequestUsageRunUsagec[5 g)zLoad genai-prices' deferred data snapshot at `Model` construction, keeping the one-time cost off the event loop. See https://github.com/pydantic/pydantic-ai/issues/7405. NrV/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/_genai_prices.pypreload_pricing_datars Nr)provider_api_url provider_idprovider_fallbackc#^# SU4US4US44Hup4U(d U(dMX44v M g7f)aYield `(provider_id, provider_api_url)` genai-prices lookup references, most specific first. The API URL identifies a provider more precisely than a name (e.g. several providers reselling the same models), so it's tried first, then the provider ID, then the fallback ID; references with nothing to match on are skipped. Shared by `RequestUsage.extract` and `lookup_context_window` so the lookup order is defined once. Nr)rrr candidate_id candidate_urls riter_provider_referencesr s?*./?(@;PTBUXikoWp'q# ==- -(rs- -)r provider_namecB[U[5(aUnO$URURp#URn[XS9H-upE[5RUSXES9upgURs $ g![ [ 4a SnNOf=f![a MWf=f)aLook up a model's context window in [genai-prices](https://github.com/pydantic/genai-prices) data. Takes a model instance, whose `model_name`, `system`, and `base_url` are matched on, or a model name together with the provider name and/or API URL to match on. Returns the context window recorded for the model under the first provider reference that knows it, or `None` if none does or no context window is recorded. N)rr)providerrr) isinstancestr model_namesystembase_urlAttributeErrorr rr find_provider_model LookupErrorcontext_window)modelrrr#rr_ model_infos rlookup_context_windowr-2s% $)$4$4ellM $$~~  (@)(#  (N>>T|?MA ((((  * $ $  $   s# A8 B8B  B  BB)rrgenai_request_timestampcbU(a [UUUUS9$[UUUUS9$![a Nf=f)a;Price a usage object with [genai-prices](https://github.com/pydantic/genai-prices), propagating its errors. Tries matching on `provider_api_url` first as it's more specific, then falls back to `provider_name`. Only `ModelResponse.cost()` wants this behaviour; everything internal goes through `best_effort_price`. )rr.)rr.)rr()usager#rrr.s rcalculate_price_for_usager1VsT !1(?    ! 7      s ! ..cU(dg[UUUUUS9$![[4a g[a=n[R "S[ U5RSU3[SS9 SnAgSnAff=f)aPrice a usage object, degrading any failure to `None`; pricing must never fail a run. A missing model name (e.g. a synthetic response from a capability) leaves nothing to look up. `genai-prices` raises `LookupError` for providers/models it doesn't know about (including `test` and `function` models) and `ValueError` for usage it can't price (e.g. cache token counts that imply a negative uncached remainder); both are expected. Anything else is unexpected and surfaces as a `CostCalculationFailedWarning` rather than being raised. Nr#rrr.zFailed to get cost: z: ) stacklevel) r1r( ValueError Exceptionwarningswarntype__name__r )r0r#rrr.es rbest_effort_pricer=ws} ( !-'$;     $  ,T!W-=-=,>bDFbopqs A/ A/3A**A/cURRc`[URURURUR UR S9=nbURURlggg)a3Fill `response.usage.cost` with a best-effort price if it's still unset. An already-set cost is never overwritten, so a provider-reported cost could take precedence in future; no model sets one today. If pricing data is unavailable the cost stays `None`, distinguishing "unknown" from a genuine zero cost. Nr3)r0costr=r# provider_urlr timestamp total_price)responseprices rfill_response_costrEsv #&#..!)!6!6&44(0(:(:  E $//  $r)returnNone)r str | NonerrHrrHrFz'Iterator[tuple[str | None, str | None]])r*zAbstractModel | strrrHrrHrFz int | None) r0RequestUsage | RunUsager#r"rrHrrHr.datetime | NonerFr ) r0rIr#rHrrHrrHr.rJrFzPriceCalculation | None)rCrrFrG) __doc__ __future__rr8collections.abcrrtypingr genai_pricesrgenai_prices.data_snapshotr _warningsr exceptionsr genai_prices.typesr messagesrmodels._abstractrr0rrrrr-r1r=rErrrrVs?A"$ #33!3'/-$("$( . .." . - .*$( $ ! !!! !  !P$( $/3  "!    - J$( $/3 ! "!!! !  ! - !!H0r