k iNHpdZddlZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z mZmZddlmZddlmZddlmZdd lmZdd lmZd d lmZd d lmZd dlm Z m!Z!d dl"m#Z#d dl$m%Z%d dl&m'Z'm(Z(m)Z)m*Z*m+Z+ej,e-Z.dedefdZ/Gdde Z0 d*de(deedee1defdZ2 d*dee(deedee1de0fdZ3 d+deedee1d e1d!ee4d"e1d#ee4d$eee1d%eee1d&e5de0fd'Z6Gd(d)Z7dS),a$Middleware composition and factory functions. This module provides utilities for composing multiple middleware together and creating middleware stacks that share Redis connections with checkpointers and stores. Compatible with LangChain's AgentMiddleware protocol for use with create_agent. N)Any AwaitableCallableListOptionalSequenceUnion)AgentMiddleware)ModelCallResult ModelRequest ModelResponse) AIMessage) ToolMessage)ToolCallRequest)Command)Redis)AsyncRedisMiddleware)ConversationMemoryMiddleware)SemanticCacheMiddleware_strip_content_ids)SemanticRouterMiddleware)ToolResultCacheMiddleware)ConversationMemoryConfigMiddlewareConfigTypeSemanticCacheConfigSemanticRouterConfigToolCacheConfigrequestreturnct|tr|d}nt|dd}|s|Sg}d}|D]}t|trt |j}||ju}|jo7|jddiko*t|j dhk}t|j }|s|s|r<|jdrddini} | || id}d}| ||s|St|tri|d|iS||S) aStrip provider-specific IDs from AIMessages before sending to LLM. This is a safety net for stale checkpoint data: messages stored before the cache-deserialization fix may still carry provider IDs (rs_, msg_ prefixes) in content blocks, additional_kwargs, or response_metadata. Cleaning them here prevents "Duplicate item found" errors from the OpenAI Responses API. messagesNFcachedT)contentadditional_kwargsresponse_metadata)update)r") isinstancedictgetgetattrrrr$r%setkeysboolr& model_copyappendoverride) rr"cleanedchangedmsg new_contentcontent_changedhas_extra_kwargs has_metadata new_kwargss C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/middleware/redis/composition.py_sanitize_requestr;(s'4  6;;z**7J55 GG c9 % % ,S[99K).sanitized_handlers2 !23!7!788888888 8r<mwhc4dtdtffd }|S)NrGr c@K|d{VSrI)awrap_model_callrGrLrKs r:wrapperzGMiddlewareStack.awrap_model_call..make_wrapper..wrappers/!#!4!4S!!.make_wrappers<=|========r<)r?r r reversedrrrr )rCrrErJcurrent_handler middlewarerSs ` r:rOz MiddlewareStack.awrap_model_calls   * )))))))) )  9 9/ 9 9 9 9 9 9   #4#455 H HJ ( \NIm,DDE <.)M*BBC    +l:GGOO$_W---------r<c K|js||d{VS|}t|jD]}dtdttgt t ttffdttgt t ttfffd}|||}||d{VS)aWrap a tool call through all middleware. This method is part of the LangChain AgentMiddleware protocol. Args: request: The tool call request. handler: The final handler to execute the tool. Returns: The tool result. NrKrLr cXdtdtttfffd }|S)NrGr c@K|d{VSrI)awrap_tool_callrPs r:rQzFMiddlewareStack.awrap_tool_call..make_wrapper..wrappers1"$!3!3C!;!;;;;;;;;r<)rr LangChainToolMessagerrRs`` r:rSz5MiddlewareStack.awrap_tool_call..make_wrappersI<(</89<<<<<<< r<) r?rTrrrrr r[r)rCrrErUrVrSs r:rZzMiddlewareStack.awrap_tool_calls$  * )))))))) )""4#455 H HJ ( $%y7KW7T1U'VV   !9U3G3P-Q#RR    +l:GGOO$_W---------r<cK|jD]Q} |d{V#t$r'}td|Yd}~Jd}~wwxYwdS)z"Close all middleware in the stack.NzError closing middleware: )r?aclose Exceptionloggerwarning)rCrVes r:r]zMiddlewareStack.acloses+ A AJ A '')))))))))) A A A?A??@@@@@@@@ A A As( AAAc K|S)zEnter async context manager.)rCs r: __aenter__zMiddlewareStack.__aenter__s  r<exc_typeexc_valexc_tbc>K|d{VdS)z4Exit async context manager and close all middleware.N)r])rCrerfrgs r: __aexit__zMiddlewareStack.__aexit__s.kkmmr<)r N)r r>)__name__ __module__ __qualname____doc__rr__annotations__rrDr rrr r rOrr r[rrZr]rdrtype BaseExceptionrrircr<r:r>r>_sn@+,,,,.H-A$B.t....,.,.<.)M*BBC,.  ,.,.,.,.\+. +.  y/CW/L)MN N +. #W, - +.+.+.+.ZAAAA4.-(   r<r>config redis_client redis_urlc|||_d|_n|||_d|_t|trt |St|t rt |St|trt|St|trt|Stdt|)a/Create a middleware instance from a config object. Args: config: The middleware configuration. redis_client: Optional Redis client to use (overrides config). redis_url: Optional Redis URL to use (overrides config). Returns: The created middleware instance. NzUnknown config type: ) rrrsr(rrrrrrrr ValueErrorro)rqrrrss r:_create_middleware_from_configrvs *  $"&-.. A&v... FO , ,A(000 F0 1 1A'/// F4 5 5A+F333?f??@@@r<configsc|g}|D])}t|||}||*t|S)a?Create a middleware stack from configuration objects. All middlewares will share the same Redis connection. Args: configs: List of middleware configuration objects. redis_client: Optional Redis client to use for all middlewares. redis_url: Optional Redis URL to use for all middlewares. Returns: A MiddlewareStack with the configured middlewares. Example: ```python from langgraph.middleware.redis import ( from_configs, SemanticCacheConfig, ToolCacheConfig, ) stack = from_configs( redis_url="redis://localhost:6379", configs=[ SemanticCacheConfig(ttl_seconds=3600), ToolCacheConfig(cacheable_tools=["search"]), ], ) ``` )rrrs)rvr0r>)rwrrrsr@rqrVs r: from_configsrys[DK''3     :&&&& ; ' ''r<llmcache toolcache皙?semantic_cache_namesemantic_cache_ttltool_cache_nametool_cache_ttlcacheable_toolsexcluded_toolsdistance_thresholdc vt|||t|||||pgg} t| ||S)atCreate a middleware stack with semantic and tool caching. A convenience function for the common pattern of caching both LLM responses and tool results. Args: redis_client: Optional Redis client to use (deprecated, use redis_url). redis_url: Redis URL to use. semantic_cache_name: Name for the semantic cache index. semantic_cache_ttl: TTL in seconds for semantic cache entries. tool_cache_name: Name for the tool cache index. tool_cache_ttl: TTL in seconds for tool cache entries. cacheable_tools: List of tool names to cache (None = all). excluded_tools: List of tool names to never cache. distance_threshold: Distance threshold for cache hits. Returns: A MiddlewareStack with semantic and tool caching. Example: ```python from langgraph.middleware.redis import create_caching_stack stack = create_caching_stack( redis_url="redis://localhost:6379", semantic_cache_ttl=3600, tool_cache_ttl=7200, cacheable_tools=["search", "calculate"], ) ``` )name ttl_secondsr)rrrrr)rwrrrs)rrry) rrrsr}r~rrrrrrws r:create_caching_stackrJsqV $*1    &1+)/R    +G !   r<creZdZdZededeedefdZ ededeedefdZ dS) IntegratedRedisMiddlewareaFactory for creating middleware that shares Redis connections. This class provides static methods to create middleware stacks that connect to the same Redis server as existing checkpointers or stores. Note: The redisvl library components used by middleware require synchronous Redis connections. While the saver/store may use async clients, middleware will create their own sync connections to the same Redis URL. saverrwr ct|ddpt|dd}|kt|dd}|Xt|dd}|Et|di}|dd}|d d }d |d |}|td g}|D]4} || _d| _t | } || 5t|S)aCreate middleware stack connecting to same Redis as a checkpoint saver. Note: Middleware creates its own sync Redis connection to the same server. The redisvl library components require sync connections internally. Args: saver: A RedisSaver or AsyncRedisSaver instance. configs: List of middleware configurations. Returns: A MiddlewareStack connecting to the same Redis server. Example: ```python from langgraph.checkpoint.redis import AsyncRedisSaver from langgraph.middleware.redis import ( IntegratedRedisMiddleware, SemanticCacheConfig, ) saver = AsyncRedisSaver(redis_url="redis://localhost:6379") await saver.asetup() stack = IntegratedRedisMiddleware.from_saver( saver, [SemanticCacheConfig(ttl_seconds=3600)], ) ``` _redis_urlNrs_redisconnection_poolconnection_kwargshost localhostportredis://:z[Could not determine Redis URL from saver. Please provide a redis_url in middleware configs.r+r*rursrrrvr0r>) rrwrsrrpoolrrrr@rqrVs r: from_saverz$IntegratedRedisMiddleware.from_saver-FE<66 ' ;; ;   "5(D99L'|->EE#(/6I2(N(N%,00EED,00>>D 84 8 8$ 8 8I  D    + +F(F "&F 7??J   z * * * *{+++r<storect|ddpt|dd}|kt|dd}|Xt|dd}|Et|di}|dd}|d d }d |d |}|td g}|D]4} || _d| _t | } || 5t|S)aCreate middleware stack connecting to same Redis as a store. Note: Middleware creates its own sync Redis connection to the same server. The redisvl library components require sync connections internally. Args: store: A RedisStore or AsyncRedisStore instance. configs: List of middleware configurations. Returns: A MiddlewareStack connecting to the same Redis server. Example: ```python from langgraph.store.redis import AsyncRedisStore from langgraph.middleware.redis import ( IntegratedRedisMiddleware, SemanticCacheConfig, ) store = AsyncRedisStore(conn=redis_client) await store.asetup() stack = IntegratedRedisMiddleware.from_store( store, [SemanticCacheConfig(ttl_seconds=3600)], ) ``` rNrsrrrrrrrrrz[Could not determine Redis URL from store. Please provide a redis_url in middleware configs.r) rrwrsrrrrrrr@rqrVs r: from_storez$IntegratedRedisMiddleware.from_storerr<N) rjrkrlrm staticmethodrrrr>rrrcr<r:rrsA,A,./A, A,A,A,\A,FA,A,./A, A,A,A,\A,A,A,r<r)NN) NNrzNr{NNNr|)8rmloggingtypingrrrrrrr langchain.agents.middlewarer !langchain.agents.middleware.typesr r r langchain_core.messagesrrr[langgraph.prebuilt.tool_noderlanggraph.typesr redis.asyncior AsyncRedisaiorconversation_memoryrsemantic_cacherrsemantic_routerr tool_cachertypesrrrrr getLoggerrjr_r;r>strrvryintfloatrrrcr<r:rsLLLLLLLLLLLLLLLLLL777777 .-----GGGGGG888888######------%%%%%%======GGGGGGGG555555111111  8 $ $42|42 42424242nYYYYYoYYY|*.# A A  A:& A} A A A A AJ*.#)()( * +)(:&)(})( )()()()(Z*.#)(,&$(+/*. #==:&=}==! =  = SM =d3i(=T#Y'======@Q,Q,Q,Q,Q,Q,Q,Q,Q,Q,r<