gTtddlmZddlmZddlmZddlmZdZeGddZ eGd d Z y ) ) annotations)copy) dataclass)UsageLimitExceeded)Usage UsageLimitscxeZdZUdZdZded< dZded< dZded< dZded < dZ d ed < dd dd Z ddZ y)raLLM usage associated with a request or run. Responsibility for calculating usage is on the model; PydanticAI simply sums the usage information across requests. You'll need to look up the documentation of the model you're using to convert usage to monetary costs. rintrequestsN int | Nonerequest_tokensresponse_tokens total_tokenszdict[str, int] | Nonedetails)r c|xj|z c_dD]7}t||}t||}|| t|||xsd|xsdz9|jrd|jxsi|_|jj D]1\}}|jj |d|z|j|<3yy)zIncrement the usage in place. Args: incr_usage: The usage to increment by. requests: The number of requests to increment by in addition to `incr_usage.requests`. )r rrrNr)r getattrsetattrritemsget)self incr_usager f self_value other_valuekeyvalues F/var/www/openai/venv/lib/python3.12/site-packages/pydantic_ai/usage.pyincrz Usage.incrs ! PA q)J!*a0K%)@a*/k6FQ!GH Q   <<-2DL(00668 U$(LL$4$4S!$t|}|j||S)zAdd two Usages together. This is provided so it's trivial to sum usage information from multiple requests and runs. )rr)rother new_usages r__add__z Usage.__add__2s J ur )rrr r returnNone)r"rr%r) __name__ __module__ __qualname____doc__r __annotations__rrrrrr$r rrr sYHc1!%NJ%-"&OZ&.#L*#n%)G ")29:E&r rcjeZdZUdZdZded< dZded< dZded< dZded< d d Z d d Z d d Z y)r aELimits on model usage. The request count is tracked by pydantic_ai, and the request limit is checked before each request to the model. Token counts are provided in responses from the model, and the token limits are checked after each response. Each of the limits can be set to `None` to disable that limit. 2r request_limitNrequest_tokens_limitresponse_tokens_limittotal_tokens_limitchtd|j|j|jfDS)axReturns `True` if this instance places any limits on token counts. If this returns `False`, the `check_tokens` method will never raise an error. This is useful because if we have token limits, we need to check them after receiving each streamed message. If there are no limits, we can skip that processing in the streaming response iterator. c3$K|]}|du yw)Nr,).0limits r z/UsageLimits.has_token_limits..Ws i  is)anyr0r1r2)rs rhas_token_limitszUsageLimits.has_token_limitsOs8 33T5O5OQUQhQhi   r c\|j}||j|k\rtd|yy)z[Raises a `UsageLimitExceeded` exception if the next request would exceed the request_limit.Nz3The next request would exceed the request_limit of )r/r r)rusager/s rcheck_before_requestz UsageLimits.check_before_request\s<**  $=)H$'Z[hZi%jk k*I $r c|jxsd}|j+||jkDrtd|jd|d|jxsd}|j+||jkDrtd|jd|d|j xsd}|j ,||j kDrtd|j d |dyy) zURaises a `UsageLimitExceeded` exception if the usage exceeds any of the token limits.rNz%Exceeded the request_tokens_limit of z (request_tokens=)z&Exceeded the response_tokens_limit of z (response_tokens=z#Exceeded the total_tokens_limit of z (total_tokens=)rr0rrr1rr2)rr;rrrs r check_tokenszUsageLimits.check_tokensbs--2  $ $ 0^dF_F_5_$78Q8Q7RRdUcTeefg  //41  % % 1oHbHb6b$89S9S8TTgWfVhhij )).Q  " " .<$BYBY3Y$'J4KbKbJccsfrettu%vw w4Z .r )r%bool)r;rr%r&) r'r(r)r*r/r+r0r1r2r9r<r?r,r rr r <sQ!#M:">'+*+H(,:,K%) )R  l xr r N) __future__r _annotationsr dataclassesr exceptionsr__all__rr r,r rrFsL2!*  -- -` 5x5x 5xr