k i dZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZmZmZddlmZddlmZdd lmZdd lmZd d lmZeje Z!e d eZ"Gddee e"Z#dS)zAsync Redis middleware base class. This module provides the async base class for Redis middleware that is compatible with LangChain's AgentMiddleware protocol. N)abstractmethod)Any AwaitableCallableGenericOptionalTypeVarUnion)AgentMiddleware)ModelCallResult ModelRequest ModelResponse) ToolMessage)ToolCallRequest)Command)Redis)MiddlewareConfigAsyncRedisClientType)boundc ZeZdZUdZeed<eed<eed<eed<eed<ej ed<ded d fd Z e dd Z dd Z ddZddZdeedeedeed d fdZdedeegeefd efdZdedeegeeeeffd eeeffdZd S)AsyncRedisMiddlewareaAbstract base class for async Redis middleware. This class provides common functionality for all async Redis-based middleware: - Async Redis client lifecycle management - Lazy initialization with double-checked locking - Graceful degradation on Redis errors - Async context manager support - Default pass-through implementations for model/tool wrapping Subclasses must implement: - _setup_async(): Called once during initialization to set up resources Example: ```python class MyAsyncMiddleware(AsyncRedisMiddleware): async def _setup_async(self) -> None: # Initialize resources self._cache = SemanticCache(redis_client=self._redis) config = MiddlewareConfig(redis_url="redis://localhost:6379") async with MyAsyncMiddleware(config) as middleware: result = await middleware.awrap_model_call(request, handler) ``` _redis_config _owns_client_graceful_degradation _initialized _init_lockconfigreturnNc&||_|j|_d|_t j|_|j|j|_d|_ dS|j .|j pi}tj |j fi||_d|_ dStd)zInitialize the async middleware. Args: config: Middleware configuration with Redis connection details. Raises: ValueError: If neither redis_url nor redis_client is provided. FNTz1Either redis_url or redis_client must be provided)rgraceful_degradationrrasyncioLockr redis_clientrr redis_urlconnection_args AsyncRedisfrom_url ValueError)selfrr's C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/middleware/redis/aio.py__init__zAsyncRedisMiddleware.__init__?s %+%@"!!,..   * -DK %D      )$4:O$-f.>RR/RRDK $D   PQQ Qc KdS)zSet up middleware resources asynchronously. Called once during lazy initialization. Subclasses should override this to initialize caches, indices, or other resources. Nr+s r, _setup_asyncz!AsyncRedisMiddleware._setup_asyncXs  r.cK|jrdS|j4d{V|js!|d{Vd|_ dddd{VdS#1d{VswxYwYdS)zvEnsure middleware is initialized (async-safe). Uses double-checked locking pattern for async safety. NT)rrr2r1s r,_ensure_initialized_asyncz.AsyncRedisMiddleware._ensure_initialized_asyncas    F? ) ) ) ) ) ) ) )$ )'')))))))))$(!( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s)A A #A cK|jrgt|drY |jd{VdS#t$r(}t d|Yd}~dSd}~wwxYwdSdS)z7Close the Redis connection if owned by this middleware.rNz"Error closing async Redis client: )rhasattrraclose Exceptionloggerwarning)r+es r,r7zAsyncRedisMiddleware.aclosens   Ix!8!8 I Ik((*********** I I IGAGGHHHHHHHHH I I I I Is< A.A))A.*AsyncRedisMiddleware[AsyncRedisClientType]c K|S)zEnter async context manager.r0r1s r, __aenter__zAsyncRedisMiddleware.__aenter__vs  r.exc_typeexc_valexc_tbc>K|d{VdS)z/Exit async context manager and close resources.N)r7)r+r?r@rAs r, __aexit__zAsyncRedisMiddleware.__aexit__zs.kkmmr.requesthandlerc(K||d{VS)aWrap a model call with middleware logic. This method is part of the LangChain AgentMiddleware protocol. Default implementation passes through to the handler. Subclasses can override to add caching, logging, etc. Args: request: The model request (typically contains messages). handler: The async function to call the model. Returns: The model response (ModelResponse or AIMessage). Nr0r+rDrEs r,awrap_model_callz%AsyncRedisMiddleware.awrap_model_calls*$WW%%%%%%%%%r.c(K||d{VS)aWrap a tool call with middleware logic. This method is part of the LangChain AgentMiddleware protocol. Default implementation passes through to the handler. Subclasses can override to add caching, logging, etc. Args: request: The tool call request. handler: The async function to execute the tool. Returns: The tool result message or command. Nr0rGs r,awrap_tool_callz$AsyncRedisMiddleware.awrap_tool_calls*(WW%%%%%%%%%r.)r N)r r<) __name__ __module__ __qualname____doc__r__annotations__rboolr#r$r-rr2r4r7r>rtype BaseExceptionrrCr rrrr rHrr LangChainToolMessagerrJr0r.r,rrs2 !    R/RDRRRR2   ^  ) ) ) )IIII4.-(   &&<.)M*BBC&  &&&&(& &  y/CW/L)MN N & #W, - &&&&&&r.r)$rNr#loggingabcrtypingrrrrrr r langchain.agents.middlewarer !langchain.agents.middleware.typesr r rlangchain_core.messagesrrSlanggraph.prebuilt.tool_noderlanggraph.typesr redis.asynciorr(typesr getLoggerrKr9rrr0r.r,r_sz NNNNNNNNNNNNNNNNNN777777 HGGGGG888888######------######  8 $ $w5ZHHHM&M&M&M&M&?G4H,IM&M&M&M&M&r.