k i*dZddlZddlZddlmZmZddlmZmZm Z m Z ddl m Z ddl mZejeZe de ZGd d eeeZdS) z(Base synchronous Redis middleware class.N)ABCabstractmethod)AnyGenericOptionalTypeVar)Redis)MiddlewareConfigRedisClientType)boundceZdZUdZeed<eed<eed<eed<eed<ej ed<ded d fd Z e dd Z dd Z ddZddZdeedeedeed d fdZd S)BaseRedisMiddlewarea0Abstract base class for synchronous Redis middleware. This class provides common functionality for all Redis-based middleware: - Redis client lifecycle management - Lazy initialization with thread-safe setup - Graceful degradation on Redis errors - Context manager support Subclasses must implement: - _setup_sync(): Called once during initialization to set up resources Example: ```python class MyMiddleware(BaseRedisMiddleware): def _setup_sync(self) -> None: # Initialize resources self._cache = SemanticCache(redis_client=self._redis) config = MiddlewareConfig(redis_url="redis://localhost:6379") with MyMiddleware(config) as middleware: # Use middleware pass ``` _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 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_degradationrr threadingLockr redis_clientrr redis_urlconnection_argsr from_url ValueError)selfrrs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/middleware/redis/base.py__init__zBaseRedisMiddleware.__init__2s %+%@"!#.**   * -DK %D      )$4:O.)9MM_MMDK $D   PQQ QcdS)zSet up middleware resources. Called once during lazy initialization. Subclasses should override this to initialize caches, indices, or other resources. Nr!s r" _setup_synczBaseRedisMiddleware._setup_syncKs  r$c|jrdS|j5|js|d|_ddddS#1swxYwYdS)zxEnsure middleware is initialized (thread-safe). Uses double-checked locking pattern for thread safety. NT)rrr(r's r"_ensure_initialized_syncz,BaseRedisMiddleware._ensure_initialized_syncTs    F _ ) )$ )  """$(! ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s#AAAc|jrat|drS |jdS#t$r(}t d|Yd}~dSd}~wwxYwdSdS)z7Close the Redis connection if owned by this middleware.rzError closing Redis client: N)rhasattrrclose Exceptionloggerwarning)r!es r"r-zBaseRedisMiddleware.closeas   Cx!8!8 C C !!##### C C CAaAABBBBBBBBB C C C C Cs4 A&A!!A&$BaseRedisMiddleware[RedisClientType]c|S)zEnter context manager.r&r's r" __enter__zBaseRedisMiddleware.__enter__is r$exc_typeexc_valexc_tbc.|dS)z)Exit context manager and close resources.N)r-)r!r5r6r7s r"__exit__zBaseRedisMiddleware.__exit__ms r$)rN)rr2)__name__ __module__ __qualname____doc__r __annotations__r boolrrr#rr(r*r-r4rtype BaseExceptionrr9r&r$r"rrs<2  R/RDRRRR2   ^  ) ) ) )CCCC4.-(   r$r)r=loggingrabcrrtypingrrrrredisr typesr getLoggerr:r/r rr&r$r"rHs..########222222222222######  8 $ $'+5999ccccc#w7cccccr$