diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py index ac07c915..f7f11285 100644 --- a/lightrag/kg/redis_impl.py +++ b/lightrag/kg/redis_impl.py @@ -1,4 +1,5 @@ import os +import logging from typing import Any, final, Union from dataclasses import dataclass import pipmaster as pm @@ -50,7 +51,7 @@ redis_retry = retry( | retry_if_exception_type(TimeoutError) | retry_if_exception_type(RedisError) ), - before_sleep=before_sleep_log(logger, "WARNING"), + before_sleep=before_sleep_log(logger, logging.WARNING), )