o i@sUddlmZddlZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZd Zdad ed <dddZe dddZdddZeGdd d ZdddZdS) ) annotationsN) dataclassfield)datetime timedeltatimezone)cache)Any)types) DataSnapshotset_custom_snapshotDataSnapshot | None_custom_snapshotreturnr cCstdurtStSN)r_bundled_snapshotrr`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/genai_prices/data_snapshot.py get_snapshotsrcCsddlm}t|ddS)Nr  providersF)rfrom_auto_update)datarr rrrrrs  rsnapshotcCs|adSr)r)rrrrr sr c@seZdZUded<ded<edddZded <eejdZd ed <d.ddZ d/ddZ   d0d1d$d%Z d2d)d*Z d3d,d-Z dS)4r list[types.Provider]rboolrcCsiSrrrrrr(szDataSnapshot.)default_factoryzPdict[tuple[str | None, str | None, str], tuple[types.Provider, types.ModelInfo]] _lookup_cacher timestampttlrrcCs|j|tkS)zPCheck if the snapshot is "active" (e.g. hasn't expired) based on a time to live.)r rnow)selfr!rrractive,szDataSnapshot.activeusagetypes.AbstractUsage model_refstr provider_id str | Noneprovider_api_urlgenai_request_timestampdatetime | Nonetypes.PriceCalculationcCsH|ptjtjd}||d||\}}|j||||jr |jdSddS)z(Calculate the price for the given usage.)tzN)r,auto_update_timestamp)rr"rutcfind_provider_model calc_pricerr )r#r%r'r)r+r,providermodelrrrcalc0s  zDataSnapshot.calcNdefault response_datar types.ProviderID | str | None api_flavortypes.ExtractedUsagec Cs`|d||}|j||d\}}|dur|||dd\}} nd} t|| ||jr-|jSdS)N)r:) find_provider extract_usager2r ExtractedUsagerr ) r#r8r)r+r:r4r'r%_r5rrrr=Cs zDataSnapshot.extract_usager4types.Provider | None&tuple[types.Provider, types.ModelInfo]cCs|}|r|j|jd|f}r|Sn|j|||f}r"|S||||}||}r>||f|j|||f<}|Std|d|j)z[Find the provider and model for the given model reference and optional provider identifier.Nz$Unable to find model with model_ref=z in )lowerrgetidr< find_model LookupError)r#r'r4r)r+provider_modelr5retrrrr2Rsz DataSnapshot.find_provider_modeltypes.ProvidercCs|durt|j|}r|Std||dur1|jD] }t|j|r)|Sqtd||rH|jD]}|jdurG|j|rG|Sq6td|)Nz$Unable to find provider provider_id=z)Unable to find provider provider_api_url=z,Unable to find provider with model matching )find_provider_by_idrrFrematch api_pattern model_matchis_match)r#r'r)r+r4rrrr<js   zDataSnapshot.find_provider)r!rrr) r%r&r'r(r)r*r+r*r,r-rr.)NNr7) r8r r)r9r+r*r:r(rr;) r'r(r4r@r)r*r+r*rrA)r'r*r)r*r+r*rrI)__name__ __module__ __qualname____annotations__rrrr"r r$r6r=r2r<rrrrr #s    rrr)r(r@cCsP|}|D] }|j|kr|Sq|D]}|jr%|j|r%|SqdS)zFind a provider by matching against provider_match logic. Args: providers: List of available providers provider_id: The provider ID to match Returns: The matching provider or None N)rBstriprDprovider_matchrO)rr)normalized_provider_idr4rrrrJs  rJ)rr )rr)rrr)r(rr@) __future__r _annotationsrK dataclassesrrrrr functoolsrtypingr r __all__rrSrrr r rJrrrrs        _