hjxdl 0.1.70__py3-none-any.whl → 0.1.72__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.70'
-__version_tuple__ = version_tuple = (0, 1, 70)
+__version__ = version = '0.1.72'
+__version_tuple__ = version_tuple = (0, 1, 72)

hdl/utils/llm/vis.py

@@ -68,6 +68,20 @@ class ImgHandler:
         num_vec_dim: int = None,
         load_model: bool = True,
     ) -> None:
+        """Initializes the class with the provided parameters.
+
+        Args:
+            model_path (str): Path to the model file.
+            db_host (str): Hostname of the database.
+            db_port (int): Port number of the database.
+            model_name (str, optional): Name of the model. Defaults to None.
+            device (str, optional): Device to run the model on. Defaults to "cpu".
+            num_vec_dim (int, optional): Number of vector dimensions. Defaults to None.
+            load_model (bool, optional): Whether to load the model. Defaults to True.
+
+        Returns:
+            None
+        """
 
         self.device = torch.device(device)
         self.model_path = model_path
@@ -333,25 +347,22 @@ class ImgHandler:
             f"{num_docs} documents indexed with {indexing_failures} failures"
         )
 
-    def img_search(
+    def emb_search(
         self,
-        img,
+        emb_query,
         num_max: int = 3,
-        extra_params: dict = None
+        extra_params: dict = None,
     ):
-        """Search for similar images in the database based on the input image.
+        """Search for similar embeddings in the database.
 
         Args:
-            img: Input image to search for similar images.
-            num_max: Maximum number of similar images to return (default is 3).
-            extra_params: Additional parameters to include in the search query (default is None).
+            emb_query (str): The embedding query to search for.
+            num_max (int, optional): The maximum number of results to return. Defaults to 3.
+            extra_params (dict, optional): Extra parameters to include in the search query. Defaults to None.
 
         Returns:
-            List of tuples containing the ID and JSON data of similar images found in the database.
+            list: A list of tuples containing the document ID and JSON data for each result.
         """
-        emb_query = self.get_img_features(
-            [img], to_numpy=True
-        ).astype(np.float32)[0].tobytes()
         query = (
             Query(
                 f"(*)=>[KNN {str(num_max)} @vector $query_vector AS vector_score]"
@@ -379,5 +390,31 @@ class ImgHandler:
         ]
         return results
 
+    def img_search(
+        self,
+        img,
+        num_max: int = 3,
+        extra_params: dict = None
+    ):
+        """Search for similar images in the database based on the input image.
+
+        Args:
+            img: Input image to search for similar images.
+            num_max: Maximum number of similar images to return (default is 3).
+            extra_params: Additional parameters to include in the search query (default is None).
+
+        Returns:
+            List of tuples containing the ID and JSON data of similar images found in the database.
+        """
+        emb_query = self.get_img_features(
+            [img], to_numpy=True
+        ).astype(np.float32)[0].tobytes()
+        results = self.emb_search(
+            emb_query=emb_query,
+            num_max=num_max,
+            extra_params=extra_params
+        )
+        return results
+
 
 

{hjxdl-0.1.70.dist-info → hjxdl-0.1.72.dist-info}/METADATA

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

{hjxdl-0.1.70.dist-info → hjxdl-0.1.72.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=AYQNyn783xY6psbpqi7TYW2XK3FmuvjMGND0tFVkkvI,413
+hdl/_version.py,sha256=q-ohSPg6hCd49ZCMCxd2sZVRKz5Qcktg4XZBuNdmSsI,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
@@ -131,12 +131,12 @@ hdl/utils/llm/chat.py,sha256=sk7Lw5Oa30k-l2fnJknkMmTc5zkBeEKsR981aeFhH5s,11907
 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=8rzJMMmVpMibUvfvGfjwFwWE5xrhZM1pjkwbe5fgmNQ,12352
+hdl/utils/llm/vis.py,sha256=h9zzDP9ADczxvejUlmLGWNjcdRiMS6FZ_PE29oD0SYc,13727
 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.1.70.dist-info/METADATA,sha256=A7kOeOkBy4YNzlalYPnyGE3yCkKoy8xcautLlDF0pkk,903
-hjxdl-0.1.70.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-hjxdl-0.1.70.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.70.dist-info/RECORD,,
+hjxdl-0.1.72.dist-info/METADATA,sha256=wf1HP93Pswqr4_il3n7dWz38gDdkOPEIluEpQdZSAOY,903
+hjxdl-0.1.72.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+hjxdl-0.1.72.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.72.dist-info/RECORD,,