added valkey sentinel support

Author: amirreza8002

scheduler/connection_types.py

@@ -1,19 +1,23 @@
 from typing import Union, Dict, Tuple, Type
 
 import redis
-import valkey
+try:
+    import valkey
+except ImportError:
+    valkey = redis
 
 from scheduler.settings import Broker
 
 ConnectionErrorType = Union[redis.ConnectionError, valkey.ConnectionError]
-ConnectionType = Union[redis.Redis, valkey.Valkey]
+ConnectionType = Union[redis.Redis, valkey.Redis]
 PipelineType = Union[redis.client.Pipeline, valkey.client.Pipeline]
 RedisSentinel = redis.sentinel.Sentinel
+ValkeySentinel = valkey.sentinel.Sentinel
 
 BrokerConnectionClass: Dict[Tuple[Broker, bool], Type] = {
     # Map of (Broker, Strict flag) => Connection Class
     (Broker.REDIS, False): redis.Redis,
-    (Broker.VALKEY, False): valkey.Valkey,
+    (Broker.VALKEY, False): valkey.Redis,
     (Broker.REDIS, True): redis.StrictRedis,
     (Broker.VALKEY, True): valkey.StrictValkey,
 }