f3fi$|dZddlmZddlZddlZddlZddlZGddejZGddeZ ddgZ y)z;Interface for a rate limiter and an in-memory rate limiter.) annotationsNcheZdZdZej ddddZej ddddZy)BaseRateLimiterayBase class for rate limiters. Usage of the base limiter is through the acquire and aacquire methods depending on whether running in a sync or async context. Implementations are free to add a timeout parameter to their initialize method to allow users to specify a timeout for acquiring the necessary tokens when using a blocking call. Current limitations: - Rate limiting information is not surfaced in tracing or callbacks. This means that the total time it takes to invoke a chat model will encompass both the time spent waiting for tokens and the time spent making the request. TblockingcyaoAttempt to acquire the necessary tokens for the rate limiter. This method blocks until the required tokens are available if `blocking` is set to `True`. If `blocking` is set to `False`, the method will immediately return the result of the attempt to acquire the tokens. Args: blocking: If `True`, the method will block until the tokens are available. If `False`, the method will return immediately with the result of the attempt. Returns: `True` if the tokens were successfully acquired, `False` otherwise. Nselfrs Z/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_core/rate_limiters.pyacquirezBaseRateLimiter.acquiresc Kywr r r s r aacquirezBaseRateLimiter.aacquire/ssNrboolreturnr)__name__ __module__ __qualname____doc__abcabstractmethodrrr rr rr sA  *.  $ 15  rrcTeZdZdZdddd d dZd dZdddd Zdddd Zy )InMemoryRateLimiteraAn in memory rate limiter based on a token bucket algorithm. This is an in memory rate limiter, so it cannot rate limit across different processes. The rate limiter only allows time-based rate limiting and does not take into account any information about the input or the output, so it cannot be used to rate limit based on the size of the request. It is thread safe and can be used in either a sync or async context. The in memory rate limiter is based on a token bucket. The bucket is filled with tokens at a given rate. Each request consumes a token. If there are not enough tokens in the bucket, the request is blocked until there are enough tokens. These tokens have nothing to do with LLM tokens. They are just a way to keep track of how many requests can be made at a given time. Current limitations: - The rate limiter is not designed to work across different processes. It is an in-memory rate limiter, but it is thread safe. - The rate limiter only supports time-based rate limiting. It does not take into account the size of the request or any other factors. Example: ```python import time from langchain_core.rate_limiters import InMemoryRateLimiter rate_limiter = InMemoryRateLimiter( requests_per_second=0.1, # <-- Can only make a request once every 10 seconds!! check_every_n_seconds=0.1, # Wake up every 100 ms to check whether allowed to make a request, max_bucket_size=10, # Controls the maximum burst size. ) from langchain_anthropic import ChatAnthropic model = ChatAnthropic( model_name="claude-sonnet-4-5-20250929", rate_limiter=rate_limiter ) for _ in range(5): tic = time.time() model.invoke("hello") toc = time.time() print(toc - tic) ``` g?)requests_per_secondcheck_every_n_secondsmax_bucket_sizec|||_d|_||_tj|_d|_||_y)aA rate limiter based on a token bucket. These tokens have nothing to do with LLM tokens. They are just a way to keep track of how many requests can be made at a given time. This rate limiter is designed to work in a threaded environment. It works by filling up a bucket with tokens at a given rate. Each request consumes a given number of tokens. If there are not enough tokens in the bucket, the request is blocked until there are enough tokens. Args: requests_per_second: The number of tokens to add per second to the bucket. The tokens represent "credit" that can be used to make requests. check_every_n_seconds: Check whether the tokens are available every this many seconds. Can be a float to represent fractions of a second. max_bucket_size: The maximum number of tokens that can be in the bucket. Must be at least `1`. Used to prevent bursts of requests. gN)ravailable_tokensr threadingLock _consume_locklastr)r rrr s r __init__zInMemoryRateLimiter.__init__xs<:$7 #.'^^-"& %:"rc|j5tj}|j||_||jz }||jzdk\r)|xj ||jzz c_||_t |j |j|_|j dk\r|xj dzc_ dddy dddy#1swYyxYw)aTry to consume a token. Returns: True means that the tokens were consumed, and the caller can proceed to make the request. A False means that the tokens were not consumed, and the caller should try again later. NrTF)r%time monotonicr&rr"minr )r nowelapseds r _consumezInMemoryRateLimiter._consumes   .."Cyy  DIIoG111Q6%%43K3K)KK% %((=(=t?S?S$TD !$$)%%*%)  ,-   sBr8sFA"  5 cgg5 pq/qj r