add import redis func

Author: peng3307165

modelcache/manager/vector_data/redis.py

@@ -6,12 +6,11 @@
 
 import numpy as np
 from modelcache.manager.vector_data.base import VectorBase, VectorData
-# from modelcache.utils import import_redis
+from modelcache.utils import import_redis
 # from modelcache.utils.log import gptcache_log
 
-# import_redis()
+import_redis()
 #
-# # pylint: disable=C0413
 # from redis.commands.search.indexDefinition import IndexDefinition, IndexType
 # from redis.commands.search.query import Query
 # from redis.commands.search.field import TagField, VectorField

modelcache/utils/__init__.py

@@ -69,3 +69,8 @@ def import_timm():
 
 def import_pillow():
     _check_library("PIL", package="pillow")
+
+
+def import_redis():
+    _check_library("redis")
+    _check_library("redis_om")