k7iSrSSKrSSKJrJr SSKJr SSKJr SSK r SSK J r SSK J r SS KJrJrJr "S S \ 5r"S S 5r"SS5r"SS\5r"SS\5rg)zCRate limiting middleware for protecting FastMCP servers from abuse.N) defaultdictdeque)Callable)Any)McpError) ErrorData)CallNext MiddlewareMiddlewareContextc8^\rSrSrSrSS\4U4SjjjrSrU=r$)RateLimitErrorz)Error raised when rate limit is exceeded.messagec4>[TU][SUS95 g)Ni)coder)super__init__r)selfr __class__s a/home/james-whalen/.local/lib/python3.13/site-packages/fastmcp/server/middleware/rate_limiting.pyrRateLimitError.__init__s @A)zRate limit exceeded) __name__ __module__ __qualname____firstlineno____doc__strr__static_attributes__ __classcell__)rs@rrrs3BBBrrc@\rSrSrSrS\S\4SjrS S\S\4Sjjr S r g ) TokenBucketRateLimiterz.Token bucket implementation for rate limiting.capacity refill_ratecXlX lXl[R"5Ul[ R "5Ulg)z~Initialize 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 rrTokenBucketRateLimiter.__init__s/! & 99;ZZ\ rr)returnc# URIShvN [R"5nX R- n[URUR X0R --5UlX lUR U:a'U=R U-slSSS5IShvN gSSS5IShvN gNNN !,IShvN (df  g=f7f)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 rconsumeTokenBucketRateLimiter.consume&s:::))+C,,,GdmmT[[7EUEU;U-UVDK" {{f$ v% :::::::scCB>CBC C&C'C,C- C8C9CCCC C CC)r.r&r+r'r)N)r ) rrrrrintfloatrboolr5r!rrrr$r$s08 " "5 "Crr$c8\rSrSrSrS\S\4SjrS\4SjrSr g ) SlidingWindowRateLimiter=z+Sliding window rate limiter implementation. max_requestswindow_secondscnXlX l[5Ul[R "5Ulg)zInitialize sliding window rate limiter. Args: max_requests: Maximum requests allowed in the time window window_seconds: Time window in seconds N)r=r>rrequestsr,r-r.)rr=r>s rr!SlidingWindowRateLimiter.__init__@s&), ZZ\ rr0cF# URIShvN [R"5nXR- nUR(aSURSU:a@URR 5 UR(aURSU:aM@[ UR5UR :a-URRU5 SSS5IShvN gSSS5IShvN gNNN !,IShvN (df  g=f7f)zCheck if a request is allowed.NrTF)r.r*r>r@popleftlenr=append)rr3cutoffs r is_allowed#SlidingWindowRateLimiter.is_allowedLs:::))+C...F--DMM!$4v$= %%'--DMM!$4v$=4==!D$5$55 $$S):::::::siD!DD!BD?D D!)D*D!/D0 D!;D<D!D!D!D D  DD!)r.r=r@r>N) rrrrrr7rr9rGr!rrrr;r;=s$5 "S "# " $ rr;c ~\rSrSrSrSS\S\S-S\\/\ 4S-S\ 4Sjjr S \S \ 4S jr S \S \ S \4S jrSrg)RateLimitingMiddleware\aMiddleware 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^UTlU=(d [US-5TlUTlUTl[ U4Sj5TlTR(a&[TRTR5Tlgg)anInitialize 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. If None, uses global limiting global_limit: If True, apply limit globally; if False, per-client cD>[TRTR5$N)r$rMrLrsr1RateLimitingMiddleware.__init__..s*##T%A%ArN) rLr7rMrNrOrlimitersr$global_limiter)rrLrMrNrOs` rrRateLimitingMiddleware.__init__qsw(?$,P4Ka4O0P*([TRTR5$rS)r;r=r>rTsrrU>SlidingWindowRateLimitingMiddleware.__init__..s,T->->@S@STrN)r=r>rNrrW)rr=rmrNs` rr,SlidingWindowRateLimitingMiddleware.__init__s4),r1*>I T>  rrZr0cHUR(aURU5$gr\r^r_s rr`:SlidingWindowRateLimitingMiddleware._get_client_identifierrbrrcc# URU5nURUnUR5IShvN nU(d+[SURSUR S-SU35eU"U5IShvN $NFN7f)z/Apply sliding window rate limiting to requests.NzRate limit exceeded: z requests per roz minutes for client: )r`rWrGrr=r>)rrZrcrfrgres rrh.SlidingWindowRateLimitingMiddleware.on_requests//8 -- ***,, '(9(9':.&&",--B9+O  w'''-(s"4BA=AB8A?9B?B)rNrWr=r>)r N)rrrrrr7rr r rr`r rrhr!rrrrkrkss. CG     !2 3S 89D@  ,.?C ((9 (h (SV (rrk)rr* collectionsrrcollections.abcrtypingrr,mcpr mcp.typesr middlewarer r r rr$r;rJrkrrrr}saI *$ ??BXB$$N>K(ZK(\=(*=(r