o § iÅ ã@sðddlmZddlmZddlmZddlmZm Z ddl m Z m Z ddl m Z ddlmZmZmZed ƒZd Ze d d d œd+dd„ƒZe d d dœd,dd„ƒZd d d dœd-dd„Ze dd œd.d&d'„ƒZe dd œd/d(d'„ƒZd d dd)œd0d*d'„Zd S)1é)Ú annotations)Údatetime)Úversion)ÚAnyÚoverloadé)Ú data_snapshotÚtypes)ÚUsage)Ú UpdatePricesÚwait_prices_updated_asyncÚwait_prices_updated_syncÚ genai_prices)r Ú calc_pricer r r Ú __version__N)Ú provider_idÚgenai_request_timestampÚusageútypes.AbstractUsageÚ model_refÚstrrútypes.ProviderID | str | Nonerúdatetime | NoneÚreturnútypes.PriceCalculationcCódS©N©)rrrrrrú[/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/genai_prices/__init__.pyrór)Úprovider_api_urlrr ú str | NonecCrrr)rrr rrrrrr)rr rcCst ¡ |||||¡S)aÔCalculate the price of an LLM API call. Either `provider_id` or `provider_api_url` should be provided, but not both. If neither are provided, we try to find the most suitable provider based on the model reference. Args: usage: The usage to calculate the price for. model_ref: A reference to the model used, this method will try to match this to a specific model. provider_id: The ID of the provider to calculate the price for. provider_api_url: The API URL of the provider to calculate the price for. genai_request_timestamp: The timestamp of the request to the GenAI service, use `None` to use the current time. Returns: The price calculation details. )rÚ get_snapshotÚcalc)rrrr rrrrr#sÚdefault)Ú api_flavorÚ response_datarútypes.ProviderID | strr%útypes.ExtractedUsagecCrrr)r&rr%rrrÚ extract_usage=ór)cCrrr)r&r r%rrrr)Cr*)rr r%cCst ¡ ||||¡S)aSExtract usage information from a response. One of `provider_id` or `provider_api_url` is required. Args: response_data: The response data to extract usage information from. provider: The provider to extract usage information for. provider_id: The ID of the provider to extract usage information for. provider_api_url: The API URL of the provider to extract usage information for. api_flavor: The API flavor of the provider to extract usage information for. Returns: The extracted usage information, model ref and provider used. )rr"r))r&rr r%rrrr)Is) rrrrrrrrrr) rrrrr r!rrrr) rrrrrrr r!rrrr)r&rrr'r%rrr()r&rr rr%rrr() r&rrrr r!r%rrr()Ú __future__rÚ _annotationsrÚimportlib.metadatarÚ_metadata_versionÚtypingrrÚrr r Ú update_pricesr r r rÚ__all__rr)rrrrÚs>   û û úÿÿû