kKjU~UdZddlZddlmZddlmZddlmZmZddl Z ddl Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZm Z m!Z!ddl"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0ddl1m2Z2e0e3Z4ee5d<dZ6dZ7dZ8dZ9Gdde2Z:Gdde2Z;Gdd e2Z<Gd!d"e2Z=Gd#d$e2Z>Gd%d&eZ?Gd'd(e?Z@Gd)d*e?ZAGd+d,e?ZBGd-d.e?ZCGd/d0e?ZDGd1d2e?ZEGd3d4e2ZFGd5d6e*ZGd7eHeIefdzd8eIfd9ZJd:eId8eIfd;ZKd8eIfd<ZLe9fd=e jjd>eId8eIfd?ZOe9fd=e jjd>eId8eIfd@ZQe9fd=e jjd>eId8eIfdAZSy)Bz"A middleware for response caching.N)Sequence)Logger)Any TypedDict)PydanticAdapter) AsyncKeyValue) MemoryStore)LimitSizeWrapper)StatisticsWrapper)KVStoreCollectionStatistics)Field) NotRequiredSelfoverride)MessagePrompt PromptResult)ResourceResourceContentResourceResult)get_access_token)CallNext MiddlewareMiddlewareContext)Tool ToolResult) get_logger)FastMCPBaseModel)nameloggerii,i __anonymous__cTeZdZUdZeezed<dZedzed<dZe ee fdzed<y)CachableResourceContentz1A wrapper for ResourceContent that can be cached.contentN mime_typemeta) __name__ __module__ __qualname____doc__strbytes__annotations__r%r&dictrr/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/middleware/caching.pyr#r#(s4; 5[ IsTz "&D$sCx.4 &r0r#creZdZUdZeeed<dZee e fdzed<de fdZ e dedefdZdefd Zy) CachableResourceResultz0A wrapper for ResourceResult that can be cached.contentsNr&returnc4t|jSNlenmodel_dump_jsonselfs r1get_sizezCachableResourceResult.get_size64'')**r0valuec ||jDcgc].}t|j|j|j0c}|jScc}wN)r$r%r&)r4r&)r4r#r$r%r&)clsr?items r1wrapzCachableResourceResult.wrap9sZ "NN +D( LLDNN+    s3Ac t|jDcgc].}t|j|j|j 0c}|j Scc}wrA)rr4rr$r%r&)r<rCs r1unwrapzCachableResourceResult.unwrapEsZ !MM *D  LLDNN*    s3A)r'r(r)r*listr#r-r&r.r+rintr= classmethodrrrDrFr/r0r1r3r30sb:*++"&D$sCx.4 &+#+    D        r0r3ceZdZUeej j ed<ee e fdzed<ee e fdzed<dZ e ed<e dedefd Zdefd Zy) CachableToolResultr$Nstructured_contentr&Fis_errorr?r5cj||j|j|j|jSNr$rLr&rMrP)rBr?s r1rDzCachableToolResult.wrapWs-MM$77^^   r0cpt|j|j|j|jSrO)rr$rLr&rMr;s r1rFzCachableToolResult.unwrap`s-LL#66]]   r0)r'r(r)rGmcptypes ContentBlockr-r.r+rrMboolrIrrrDrFr/r0r1rKrKQss #))(( ))S#X-- sCx.4 Hd      r0rKceZdZUdZeed<ejjejjzejjzejjzed<y)CachableMessagez)A wrapper for Message that can be cached.roler$N) r'r(r)r*r+r-rRrS TextContent ImageContent AudioContentEmbeddedResourcer/r0r1rWrWisT3 I  ))  ! ))  ! )) $ $ %r0rWceZdZUdZeeed<dZedzed<dZ e ee fdzed<de fdZ ededefd Zdefd Zy) CachablePromptResultz.A wrapper for PromptResult that can be cached.messagesN descriptionr&r5c4t|jSr7r8r;s r1r=zCachablePromptResult.get_size|r>r0r?c ||jDcgc]#}t|j|j%c}|j|j Scc}w)N)rXr$r_r`r&)r_rWrXr$r`r&)rBr?ms r1rDzCachablePromptResult.wrapsPINIWAQVVQYY?))   s(Ac t|jDcgc]#}t|j|j%c}|j |j Scc}w)N)r$rXrc)rr_rr$rXr`r&)r<rds r1rFzCachablePromptResult.unwrapsV&A 7&((   s(A)r'r(r)r*rGrWr-r`r+r&r.rrHr=rIrrrDrFr/r0r1r^r^uso8?##"Kt""&D$sCx.4 &+#+  $    r0r^c2eZdZUdZeeed<eeed<y)SharedMethodSettingsz!Shared config for a cache method.ttlenabledN)r'r(r)r*rrHr-rUr/r0r1rgrgs+ S   r0rgceZdZdZy)ListToolsSettings/Configuration options for Tool-related caching.Nr'r(r)r*r/r0r1rkrks9r0rkceZdZdZy)ListResourcesSettings3Configuration options for Resource-related caching.Nrmr/r0r1roro=r0roceZdZdZy)ListPromptsSettings1Configuration options for Prompt-related caching.Nrmr/r0r1rsrs;r0rsc>eZdZUdZeeeed<eeeed<y)CallToolSettingsrlincluded_toolsexcluded_toolsN)r'r(r)r*rrGr+r-r/r0r1rwrws#9S **S **r0rwceZdZdZy)ReadResourceSettingsrpNrmr/r0r1r{r{rqr0r{ceZdZdZy)GetPromptSettingsrtNrmr/r0r1r}r}rur0r}ceZdZUedZedzed<edZedzed<edZedzed<edZ edzed<edZ edzed<edZ edzed<y) ResponseCachingStatisticsN)default list_toolslist_resources list_prompts read_resource get_prompt call_tool) r'r(r)r rr r-rrrrrr/r0r1rrs5:45HJ+d2H9>t9LN/$6L7'<'> % " :%8%: # # <&:&< $ 6#4#6 !  4"2"4  ?Nkk:+?  GVkk>/G " CRkk<-C   ++5#3  !IXkk/,I  FUkk-+F r0context call_nextr5c nK|jjddur||d{St}|jj|d{x}r|S||d{}|Dcgc]e}t |j |j |j|j|j|j|j|jg}}|jj|||jjdtd{|S777cc}w7w)zList tools from the cache, if caching is enabled, and the result is in the cache. Otherwise, otherwise call the next middleware and store the result in the cache if caching is enabled.riFNkey)rtitler` parameters output_schema annotationsr&tagsrhrr?rh)rget_get_auth_partition_keyrrrrr`rrrr&rputFIVE_MINUTES_IN_SECONDS)r<rr cache_key cached_valuetoolstoolcachable_toolss r1 on_list_toolsz'ResponseCachingMiddleware.on_list_tools*s;  $ $ ( ( 3u <"7++ +02 !%!7!7!;!; !;!JJ J< J &/&8 8 &  YYjj ,,??"00 ,,YYYY   & $$(( ))--e5LM)   =,K!9 &  S)D5D(-D5D*D5-D,.D56A*D. AD5!D3"D5*D5,D5.D5c nK|jjddur||d{St}|jj|d{x}r|S||d{}|Dcgc]e}t |j |j |j|j|j|j|j|jg}}|jj|||jjdtd{|S777cc}w7w)zList resources from the cache, if caching is enabled, and the result is in the cache. Otherwise, otherwise call the next middleware and store the result in the cache if caching is enabled.riFNr)rrr`rr&r%rurirhr)rrrrrrrr`rr&r%rrrr)r<rrrr resourcesresourcecachable_resourcess r1on_list_resourcesz+ResponseCachingMiddleware.on_list_resourcesSs<  ( ( , ,Y 75 @"7++ +02 !%!;!;!?!?I!?!NN N< N .7.@(@ & . & ]]nn$00]]]]",,$00LL &  . ((,,$--11%9PQ-   "!=,O)A .  rc BK|jjddur||d{St}|jj|d{x}r|S||d{}|Dcgc]O}t |j |j |j|j|j|jQ}}|jj|||jjdtd{|S777cc}w7w)zList prompts from the cache, if caching is enabled, and the result is in the cache. Otherwise, otherwise call the next middleware and store the result in the cache if caching is enabled.riFNr)rrr`rr& argumentsrhr) rrrrrrrr`rr&rrr)r<rrrrpromptspromptcachable_promptss r1on_list_promptsz)ResponseCachingMiddleware.on_list_prompts|s+  & & * *9 5 >"7++ +02 !%!9!9!=!=)!=!LL L< L *3G*<$<" * " [[ll"..[[[[ **  "  * &&**"++//7NO+    9,M%= *  sS)DD-DDD-D.D6AD AD D DDDDcbK|jj}|jjddus|j |s||d{St |jt }|jj|d{x}r|jS||d{}tj|}|jj|||jjdt d{|jS777r7w) zCall a tool from the cache, if caching is enabled, and the result is in the cache. Otherwise, otherwise call the next middleware and store the result in the cache if caching is enabled.riF) tool_nameNmsgauth_keyrr?rhr) messagerrr_matches_tool_cache_settings_make_call_tool_cache_keyrrrFrKrDrONE_HOUR_IN_SECONDS)r<rrrrr tool_resultcachable_tool_results r1 on_call_toolz&ResponseCachingMiddleware.on_call_tools1OO((  # # ' '   ;;i;P"7++ +2*A*C "&!6!6!:!:y!:!II I< I&&( ((1'(:": 3E3J3J4K4 ##''&((,,U4GH(   $**,,), J#;  sJAD/D'AD/D)!D/8D+9AD/D-D/)D/+D/-D/cK|jjddur||d{St|jt }|j j|d{x}r|j S||d{}tj|}|j j|||jjdtd{|j S777q7w)zRead a resource from the cache, if caching is enabled, and the result is in the cache. Otherwise, otherwise call the next middleware and store the result in the cache if caching is enabled.riFNrrrhr) rr_make_read_resource_cache_keyrrrrFr3rDrrr<rrrrr?s r1on_read_resourcez*ResponseCachingMiddleware.on_read_resources  ' ' + +I 6% ?"7++ +6*A*C "&!:!:!>!>9!>!MM M< M&&( (&/&8 8-2259 ''++,,008KL,   ""$$',N!9 G)DC>AD.D/!DDAD)D*DDDDcK|jjddur||d{St|jt }|j j|d{x}r|j S||d{}tj|}|j j|||jjdtd{|j S777q7w)zGet a prompt from the cache, if caching is enabled, and the result is in the cache. Otherwise, otherwise call the next middleware and store the result in the cache if caching is enabled.riFNrrrhr) rr_make_get_prompt_cache_keyrrrrFr^rDrrrs r1 on_get_promptz'ResponseCachingMiddleware.on_get_prompts  $ $ ( ( 3u <"7++ +3*A*C "&!7!7!;!; !;!JJ J< J&&( ($-g$66+007 $$(())--e5HI)   ""$$%, K7 rrc|jjdx}r||vry|jjdx}r||vryy)z K.!5599:JK K> KN*r0c @t|jjjj d|jjjj d|jjjj d|jjjj d|jjjj d|jjjj dS)z!Get the statistics for the cache.rrrrrr)rrrrrr)rr statistics collectionsrr;s r1rz$ResponseCachingMiddleware.statisticss({{--99==lK;;11==AABRS//;;??O++00<<@@AQR{{--99==mLkk,,88<<\J   r0)+r'r(r)r*ONE_MB_IN_BYTESrrkrorsr{r}rwrHrrrrRrSListToolsRequestrrrrListResourcesRequestrrListPromptsRequestrrCallToolRequestParamsrrReadResourceRequestParamsrrGetPromptRequestParamsrrr+rUrrrr/r0r1rrs$/38<@D<@>B8<6:,X $t+X /5X "7!= X 3T9 X !5t ; X /5X -t3X X t&"399#=#=>&CII66FG& $ &&P&""399#A#AB&"CII::HX>%,,. ) 3 3 55r0cPt}|tSt|jS)afReturn a stable, hashed identifier for the current access token. Cache entries are partitioned by access token so that responses filtered by per-component authorization (e.g. `auth=require_scopes(...)`) are not leaked across users with different permissions. Unauthenticated callers (including STDIO) share a single anonymous partition. )rANONYMOUS_AUTH_KEYrtoken)rs r1rr-s&  E }!! 5;; ''r0rrc bt|d|jdt|jS)zKMake a cache key for a tool call using a stable hash of name and arguments.:rrrrrs r1rr</ hZq !4Fs}}4U3VW XXr0c6t|d|jS)z@Make a cache key for a resource read using a stable hash of URI.r)rrrs r1rrDs hZq 2 33r0c bt|d|jdt|jS)zLMake a cache key for a prompt get using a stable hash of name and arguments.rrrs r1rrLrr0)Tr*rcollections.abcrloggingrtypingrr mcp.typesrRrkey_value.aio.adapters.pydanticr!key_value.aio.protocols.key_valuerkey_value.aio.stores.memoryr !key_value.aio.wrappers.limit_sizer !key_value.aio.wrappers.statisticsr )key_value.aio.wrappers.statistics.wrapperr pydanticr typing_extensionsrrrfastmcp.prompts.baserrrfastmcp.resources.baserrrfastmcp.server.dependenciesr$fastmcp.server.middleware.middlewarerrrfastmcp.tools.baserrfastmcp.utilities.loggingrfastmcp.utilities.typesrr'r r-rrrrr#r3rKrWr^rgrkrorsrwr{r}rrr.r+rrrrSrrrrrrr/r0r1rs($!;;3>?99>>LL8XX/04**$'.' - B ) 0 &  + >9:,:>0><.<+++>/><,<H 0HX X v $sCx.4"7 C 63636 ( ( ;MY  ( (Y47YY?Q4  , ,48;44