kKj!dZddlZddlZddlmZmZddlmZmZddl m Z m Z ddl Z ddl mZddlmZdd lmZmZmZGd d eZGd d ZGddZGddeZGddeZy)zCRate limiting middleware for protecting FastMCP servers from abuse.N) defaultdictdeque) AwaitableCallable)Anycast)McpError) ErrorData)CallNext MiddlewareMiddlewareContextc*eZdZdZddeffd ZxZS)RateLimitErrorz)Error raised when rate limit is exceeded.messagec:t|td|y)Ni)coder)super__init__r )selfr __class__s x/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/middleware/rate_limiting.pyrzRateLimitError.__init__s @A)zRate limit exceeded)__name__ __module__ __qualname____doc__strr __classcell__)rs@rrrs3BBBrrc2eZdZdZdedefdZd dedefdZy) TokenBucketRateLimiterz.Token bucket implementation for rate limiting.capacity refill_ratec||_||_||_tj|_t j |_y)zInitialize token bucket. Args: capacity: Maximum number of tokens in the bucket refill_rate: Tokens added per second N)r"r#tokenstime last_refillanyioLock_lock)rr"r#s rrzTokenBucketRateLimiter.__init__s5! & 99;ZZ\ rr%returncK|j4d{tj}||jz }t|j|j ||j zz|_||_|j |k\r'|xj |zc_ dddd{y dddd{y777 #1d{7swYyxYww)zTry to consume tokens from the bucket. Args: tokens: Number of tokens to consume Returns: True if tokens were available and consumed, False otherwise NTF)r*r&r'minr"r%r#)rr%nowelapseds rconsumezTokenBucketRateLimiter.consume's:::))+CD,,,GdmmT[[7TEUEU;U-UVDK"D {{f$ v% :::::::::scCB?CBC C'C(C-C. C9C:CCCC C CCN)r ) rrrrintfloatrboolr0rrr!r!s+8 " "5 "Crr!c,eZdZdZdedefdZdefdZy)SlidingWindowRateLimiterz+Sliding window rate limiter implementation. max_requestswindow_secondscp||_||_t|_t j |_y)zInitialize sliding window rate limiter. Args: max_requests: Maximum requests allowed in the time window window_seconds: Time window in seconds N)r7r8rrequestsr(r)r*)rr7r8s rrz!SlidingWindowRateLimiter.__init__As*), ZZ\ rr+c K|j4d{tj}||jz }|jrK|jd|kr9|jj |jr|jd|kr9t |j|j kr-|jj| dddd{y dddd{y777 #1d{7swYyxYww)zCheck if a request is allowed.NrTF)r*r&r8r:popleftlenr7append)rr.cutoffs r is_allowedz#SlidingWindowRateLimiter.is_allowedMs:::))+C4...F--DMM!$4v$= %%'--DMM!$4v$=4==!D$5$55 $$S):::::::::siDC3DA:C9>C9 DC5D!C9" D-C7.D5D7D9D ?D D DN)rrrrr1rr3r@r4rrr6r6>s$5 "S "# " $ rr6c eZdZdZ d dededzdeegefeege efzdzde fdZ ded efd Z ded e d efd Zy)RateLimitingMiddlewareaKMiddleware that implements rate limiting to prevent server abuse. Uses a token bucket algorithm by default, allowing for burst traffic while maintaining a sustainable long-term rate. Example: ```python from fastmcp.server.middleware.rate_limiting import RateLimitingMiddleware # Allow 10 requests per second with bursts up to 20 rate_limiter = RateLimitingMiddleware( max_requests_per_second=10, burst_capacity=20 ) mcp = FastMCP("MyServer") mcp.add_middleware(rate_limiter) ``` Nmax_requests_per_secondburst_capacity get_client_id global_limitc|_|xst|dz_|_|_t fd_jr&tjj_yy)aInitialize rate limiting middleware. Args: max_requests_per_second: Sustained requests per second allowed burst_capacity: Maximum burst capacity. If None, defaults to 2x max_requests_per_second get_client_id: Function to extract client ID from context. Can be sync or async. If None, uses global limiting global_limit: If True, apply limit globally; if False, per-client cDtjjSN)r!rDrCrsrz1RateLimitingMiddleware.__init__..s*##T%A%ArN) rCr1rDrErFrlimitersr!global_limiter)rrCrDrErFs` rrzRateLimitingMiddleware.__init__rsw$(?$,P4Ka4O0P*(SlidingWindowRateLimitingMiddleware.__init__..s,T->->@S@STrN)r7r8rErrM)rr7rbrEs` rrz,SlidingWindowRateLimitingMiddleware.__init__s4 ),r1*>I T>  rrOr+cK|jr@|j|}tj|rtt|d{S|Sy7 wrQrSrVs rrXz:SlidingWindowRateLimitingMiddleware._get_client_identifierrYrZr[cK|j|d{}|j|}|jd{}|s+td|jd|j dzd|||d{S7h7C7w)z/Apply sliding window rate limiting to requests.NzRate limit exceeded: z requests per rdz minutes for client: )rXrMr@rr7r8)rrOr[rWr^r]s rr_z.SlidingWindowRateLimitingMiddleware.on_requests55g>> -- ***,, '(9(9':.&&",--B9+O  w'''?-(s1BB&BBrnsfI */ ??BXB$$N>Q(ZQ(hC(*C(r