hjxdl 0.2.21__py3-none-any.whl → 0.2.23__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.21'
-__version_tuple__ = version_tuple = (0, 2, 21)
+__version__ = version = '0.2.23'
+__version_tuple__ = version_tuple = (0, 2, 23)

hdl/utils/llm/vis.py

@@ -3,6 +3,7 @@ import json
 import base64
 from io import BytesIO
 import requests
+import uuid
 
 import torch
 import numpy as np
@@ -372,12 +373,20 @@ class ImgHandler:
         for img_file, emb in tqdm(zip(sorted_imgs, img_feats)):
             # 初始化 Redis，先使用 img 文件名作为 Key 和 Value，后续再更新为图片特征向量
             # pipeline.json().set(img_file, "$", img_file)
+
+            if img_file.startswith("data:"):
+                img_data = img_file
+                img_idx = f"pic-{str(uuid.uuid4())}"
+            else:
+                img_data = imgfile_to_base64(img_file)
+                img_idx = f"pic-{img_file}"
+
             emb = emb.astype(np.float32).tolist()
             emb_json = {
                 "emb": emb,
-                "data": imgfile_to_base64(img_file)
+                "data": img_data
             }
-            pipeline.json().set(f"pic-{img_file}", "$", emb_json)
+            pipeline.json().set(img_idx, "$", emb_json)
             res = pipeline.execute()
             # print('redis set:', res)
 

{hjxdl-0.2.21.dist-info → hjxdl-0.2.23.dist-info}/METADATA

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

{hjxdl-0.2.21.dist-info → hjxdl-0.2.23.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=Ar3Tbs4Bhi_cZcBKtRv2aWiaZHLwaa5_3QayHWYxGhY,413
+hdl/_version.py,sha256=-hljAU21MXFeuL9fV1MtjsvUKcUWxJn4DlS0PdRjDcs,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
@@ -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=-6QvxSVzKqxLh_l0aYg2wN2G5HOiQvCpfp-jn9twXw0,16210
+hdl/utils/llm/vis.py,sha256=aeTVX-jKXJBnvQXaOWiHpI9Nc7IneSCqm_g7N4spJUc,16446
 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.21.dist-info/METADATA,sha256=9nC7NyKwWGFkRyiQYP8KAl5LXUzaUJU6E538x4d0Q-k,836
-hjxdl-0.2.21.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-hjxdl-0.2.21.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.2.21.dist-info/RECORD,,
+hjxdl-0.2.23.dist-info/METADATA,sha256=xFu32son2_PThYfRcnKAh5-Kc1bUoRgyAhoo9hsHixc,836
+hjxdl-0.2.23.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+hjxdl-0.2.23.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.2.23.dist-info/RECORD,,