hjxdl 0.2.25__py3-none-any.whl → 0.2.27__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.25'
-__version_tuple__ = version_tuple = (0, 2, 25)
+__version__ = version = '0.2.27'
+__version_tuple__ = version_tuple = (0, 2, 27)

hdl/utils/database_tools/connect.py

@@ -28,14 +28,11 @@ def connect_by_infofile(info_file: str):
 #     conn.close()
 
 def conn_redis(
-    host: str,
-    port: int
+    **redis_args
 ):
     import redis
     client = redis.Redis(
-        host=host,
-        port=port,
-        decode_responses=True
+        **redis_args
     )
     res = client.ping()
     print(res)

hdl/utils/llm/vis.py

@@ -18,15 +18,9 @@ from hdl.jupyfuncs.show.pbar import tqdm
 from redis.commands.search.query import Query
 
 
-from ..database_tools.connect import conn_redis
+# from ..database_tools.connect import conn_redis
 
 
-__all__ = [
-    "ImgHandler",
-    "imgfile_to_base64",
-    "imgbase64_to_pilimg"
-]
-
 HF_HUB_PREFIX = "hf-hub:"
 
 import requests
@@ -133,8 +127,6 @@ class ImgHandler:
     def __init__(
         self,
         model_path,
-        db_host = "127.0.0.1",
-        db_port = 8888,
         conn=None,
         model_name: str = None,
         device: str = "cpu",
@@ -160,9 +152,7 @@ class ImgHandler:
         self.model_path = model_path
         self.model_name = model_name
 
-        self.db_host = db_host
-        self.db_port = db_port
-        self._db_conn = None
+        self.db_conn = conn
         self.num_vec_dim = num_vec_dim
         self.pic_idx_name = "idx:pic_idx"
         if load_model:
@@ -205,17 +195,6 @@ class ImgHandler:
             HF_HUB_PREFIX + self.model_path
         )
 
-    @property
-    def db_conn(self):
-        """Establishes a connection to Redis server if not already connected.
-
-            Returns:
-                Redis connection object: A connection to the Redis server.
-        """
-        if self._db_conn is None:
-            self._db_conn = conn_redis(self.db_host, self.db_port)
-        return self._db_conn
-
     def get_img_features(
         self,
         images,

{hjxdl-0.2.25.dist-info → hjxdl-0.2.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hjxdl
-Version: 0.2.25
+Version: 0.2.27
 Summary: A collection of functions for Jupyter notebooks
 Home-page: https://github.com/huluxiaohuowa/hdl
 Author: Jianxing Hu

{hjxdl-0.2.25.dist-info → hjxdl-0.2.27.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=UmsAAHFuqYBzGCBQpGNQPt7zCgBErX1ViQbYB7gdPPg,413
+hdl/_version.py,sha256=WGPiv7rOFEL9zbeoRoeVA9Vyz2mZKhrzn7WKR7gHcL8,413
 hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
 hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -118,7 +118,7 @@ hdl/utils/chemical_tools/__init__.py,sha256=_QRNtVx0ieZZgSEsHndPFKm6XU3WXfRb7GYq
 hdl/utils/chemical_tools/query_info.py,sha256=wyQXwKSY_gBGVUNvYggHpYBtOLAtpYKq3PN5wqDb7Co,4204
 hdl/utils/chemical_tools/sdf.py,sha256=71PEqU0H885L6IeGHEa6n7ZLZThvMsZOVLuFG2wnoyM,542
 hdl/utils/database_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/database_tools/connect.py,sha256=f9ZH7ohyEzznnq42WEfhQBRzhIPNVYu9CHBUMTYTUGE,934
+hdl/utils/database_tools/connect.py,sha256=xCacGucKxlQUXs6AsNddpeECvdqT1180V1ZWqHrUdNA,875
 hdl/utils/database_tools/datetime.py,sha256=xqE2xNiOpADzX-R8_bM0bioJRF3Ay9Jp1CAG6dy6uVI,1202
 hdl/utils/database_tools/web.py,sha256=vZJYWA02QbyUUMGLDX710cexn7RlfWeHa8YwK3UsDZ0,1934
 hdl/utils/desc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -132,13 +132,13 @@ hdl/utils/llm/chatgr.py,sha256=GO2G7g6YybduA5VCUuGjvEsJfC_6L7rycSnPeHMcxyM,2820
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
 hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
-hdl/utils/llm/vis.py,sha256=skeoBgFNes8swPBeWIsn3oKZ4Jv4Cl5GGrxsRDNupac,16854
+hdl/utils/llm/vis.py,sha256=WYCZpWSXC9-ExMCRJNM5Kb2e9igDF5cGthHUYD91nQ0,16299
 hdl/utils/llm/visrag.py,sha256=_PuKtmQIXD5bnmXwDWhTLdzOhgC42JiqdMNb1uKA7n8,9190
 hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
 hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
-hjxdl-0.2.25.dist-info/METADATA,sha256=i8tbHzSQf9nal7CU2P7Whe3AChbtkBHJPUIIKQaVqQo,836
-hjxdl-0.2.25.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-hjxdl-0.2.25.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.2.25.dist-info/RECORD,,
+hjxdl-0.2.27.dist-info/METADATA,sha256=FAlKe4yNZV4alDJB5bAghBnns3eB6dDLY9w-yJ-4b08,836
+hjxdl-0.2.27.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+hjxdl-0.2.27.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.2.27.dist-info/RECORD,,