From 5fb7beae13b56fbb9be7ef08fb7e168ee3c8c985 Mon Sep 17 00:00:00 2001 From: chenjh3 Date: Thu, 13 Feb 2025 14:00:06 +0800 Subject: [PATCH] fix hint --- api/core/app/features/rate_limiting/rate_limit.py | 2 +- api/extensions/ext_redis.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/api/core/app/features/rate_limiting/rate_limit.py b/api/core/app/features/rate_limiting/rate_limit.py index df63db63cc..f53fa4506f 100644 --- a/api/core/app/features/rate_limiting/rate_limit.py +++ b/api/core/app/features/rate_limiting/rate_limit.py @@ -2,10 +2,10 @@ import logging import time import uuid from collections.abc import Generator, Mapping -from configs import dify_config from datetime import timedelta from typing import Any, Optional, Union +from configs import dify_config from core.errors.error import AppInvokeQuotaExceededError from extensions.ext_redis import redis_client diff --git a/api/extensions/ext_redis.py b/api/extensions/ext_redis.py index f8d4c21826..89342f6c33 100644 --- a/api/extensions/ext_redis.py +++ b/api/extensions/ext_redis.py @@ -1,5 +1,5 @@ -from typing import Any, Union, Callable, TypeVar, Generic import functools +from typing import Any, Callable, Generic, TypeVar, Union import redis from redis.cluster import ClusterNode, RedisCluster @@ -125,7 +125,8 @@ def init_app(app: DifyApp): for node in dify_config.REDIS_CLUSTERS.split(",") ] # FIXME: mypy error here, try to figure out how to fix it - redis_client.initialize(RedisCluster(startup_nodes=nodes, password=dify_config.REDIS_CLUSTERS_PASSWORD), prefix=dify_config.REDIS_KEY_PREFIX) # type: ignore + redis_client.initialize(RedisCluster(startup_nodes=nodes, password=dify_config.REDIS_CLUSTERS_PASSWORD), + prefix=dify_config.REDIS_KEY_PREFIX) # type: ignore else: redis_params.update( {