visual-rag-toolkit 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

demo/__init__.py

@@ -7,4 +7,4 @@ A Streamlit-based UI for:
 - Interactive playground for visual search
 """
 
-__version__ = "0.1.4"
+__version__ = "0.1.5"

visual_rag/__init__.py

@@ -33,7 +33,7 @@ Each component works independently - use only what you need.
 
 import logging
 
-__version__ = "0.1.4"
+__version__ = "0.1.5"
 
 
 def setup_logging(level: str = "INFO", format: str = None) -> None:

visual_rag/cli/main.py

@@ -288,7 +288,7 @@ def cmd_search(args):
         )
 
     # Search
-    query_np = query_embedding.detach().cpu().numpy()
+    query_np = query_embedding.detach().cpu().float().numpy()  # .float() for BFloat16
     if args.strategy == "single_full":
         results = single_stage.search(
             query_embedding=query_np,

visual_rag/retrieval/multi_vector.py

@@ -157,9 +157,10 @@ class MultiVectorRetriever:
     ) -> List[Dict[str, Any]]:
         q = self.embedder.embed_query(query)
         if isinstance(q, torch.Tensor):
-            query_embedding = q.detach().cpu().numpy()
+            # .float() converts BFloat16 to Float32 (numpy doesn't support BFloat16)
+            query_embedding = q.detach().cpu().float().numpy()
         else:
-            query_embedding = np.asarray(q)
+            query_embedding = np.asarray(q, dtype=np.float32)
 
         return self.search_embedded(
             query_embedding=query_embedding,

visual_rag/retrieval/single_stage.py

@@ -129,5 +129,5 @@ class SingleStageRetriever:
         if isinstance(embedding, torch.Tensor):
             if embedding.dtype == torch.bfloat16:
                 return embedding.cpu().float().numpy()
-            return embedding.cpu().numpy()
+            return embedding.cpu().float().numpy()  # .float() for BFloat16 compatibility
         return np.array(embedding, dtype=np.float32)

visual_rag/retrieval/three_stage.py

@@ -51,7 +51,7 @@ class ThreeStageRetriever:
         if isinstance(embedding, torch.Tensor):
             if embedding.dtype == torch.bfloat16:
                 return embedding.cpu().float().numpy()
-            return embedding.cpu().numpy()
+            return embedding.cpu().float().numpy()  # .float() for BFloat16 compatibility
         return np.array(embedding, dtype=np.float32)
 
     def _infer_vector_is_multivector(self, vector_name: str) -> bool:

visual_rag/retrieval/two_stage.py

@@ -418,7 +418,7 @@ class TwoStageRetriever:
         if isinstance(embedding, torch.Tensor):
             if embedding.dtype == torch.bfloat16:
                 return embedding.cpu().float().numpy()
-            return embedding.cpu().numpy()
+            return embedding.cpu().float().numpy()  # .float() for BFloat16 compatibility
         return np.array(embedding, dtype=np.float32)
 
     def build_filter(

visual_rag/visualization/saliency.py

@@ -55,14 +55,14 @@ def generate_saliency_map(
     if hasattr(query_embedding, "numpy"):
         query_np = query_embedding.numpy()
     elif hasattr(query_embedding, "cpu"):
-        query_np = query_embedding.cpu().numpy()
+        query_np = query_embedding.cpu().float().numpy()  # .float() for BFloat16
     else:
         query_np = np.array(query_embedding, dtype=np.float32)
 
     if hasattr(doc_embedding, "numpy"):
         doc_np = doc_embedding.numpy()
     elif hasattr(doc_embedding, "cpu"):
-        doc_np = doc_embedding.cpu().numpy()
+        doc_np = doc_embedding.cpu().float().numpy()  # .float() for BFloat16
     else:
         doc_np = np.array(doc_embedding, dtype=np.float32)
 

{visual_rag_toolkit-0.1.4.dist-info → visual_rag_toolkit-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: visual-rag-toolkit
-Version: 0.1.4
+Version: 0.1.5
 Summary: End-to-end visual document retrieval with ColPali, featuring two-stage pooling for scalable search
 Project-URL: Homepage, https://github.com/Ara-Yeroyan/visual-rag-toolkit
 Project-URL: Documentation, https://github.com/Ara-Yeroyan/visual-rag-toolkit#readme

{visual_rag_toolkit-0.1.4.dist-info → visual_rag_toolkit-0.1.5.dist-info}/RECORD

@@ -11,7 +11,7 @@ benchmarks/vidore_tatdqa_test/dataset_loader.py,sha256=gCCneGAKWQm0WlJHLvGjoMrAb
 benchmarks/vidore_tatdqa_test/metrics.py,sha256=cLdYbRt5VcxInO1cN79ve6ZLP3kaSxRkdzRX3IbPPMs,1112
 benchmarks/vidore_tatdqa_test/run_qdrant.py,sha256=_PikeqIYpWPim-KEQOwvT-aqwYoAWASjqJVisi8PfQg,28681
 benchmarks/vidore_tatdqa_test/sweep_eval.py,sha256=d_kbyNTJ1LoFfIVnsZyiRO1nKyMqmRB5jEweZL6kYd4,12688
-demo/__init__.py,sha256=QtvjqZ94WpbJ915Xik5ZzYFDtK7APmsQK6m0-Sxe8uw,204
+demo/__init__.py,sha256=zxRkBWe4opMZXfJxNULuktXdZLuYsZFHeaijgxPe07w,204
 demo/app.py,sha256=nZbCz1mpRK-GZTgOHyz4m4AfgKFgsH-09JwXeL3d3ng,1405
 demo/commands.py,sha256=qxRE2x610yZvcjwEfSKiR9CyFonX-vRxFqQNJCUKfyA,13690
 demo/config.py,sha256=BNkV4NSEEMIV9e6Z-cxds2v247uVmTPCgL-M5ItPzMg,757
@@ -27,12 +27,12 @@ demo/ui/header.py,sha256=J2hXr_nNyg1H9rmrd-EGx3WUl7lYo-Ca30ptgzBCfBs,806
 demo/ui/playground.py,sha256=yRlWWzJgsc596vALn5f0PHhmhtJCMmfv61nYakW75GQ,13672
 demo/ui/sidebar.py,sha256=k7CZ5V7SxwbPOL9nQTWUct5y9mPY-YnLIeBgha7aIEE,8379
 demo/ui/upload.py,sha256=NVmd1-xRj_dfgq14x_eqcN6MjDAZ3hbPBD0YRUz8OOY,20508
-visual_rag/__init__.py,sha256=O8I4wQdQcQxAs7_SIPxnAzUOmRv5I9ezau8sZx9yqXQ,4384
+visual_rag/__init__.py,sha256=mOcAcA2rkPVA9GQ63CKFb1g3KatGqILsjfWFj4kkpus,4384
 visual_rag/config.py,sha256=qqSQk2lM5MiRji-6xQNGS2gSiXA4NgyJnCbgGx7uGJQ,7395
 visual_rag/demo_runner.py,sha256=ahIKYhHQSLOF_kWbj0u_29SLX4qTdyxb3Smt_cMMfzI,2763
 visual_rag/qdrant_admin.py,sha256=NNczko2S5-K3qATNUxgYn51hNWgWb6boheL7vlCQGpM,7055
 visual_rag/cli/__init__.py,sha256=WgBRXm0VACfLltvVlLcSs3FTM1uQ7Uuw3CVD4-zWZwc,46
-visual_rag/cli/main.py,sha256=QmpnQ0lbC6Q9lwxaSCDh6paEEzI78IPY1jwc3_9y7VI,21083
+visual_rag/cli/main.py,sha256=lw2xS96FtdzFnftModGcqRyJYlxFWA0U3aE3C247mBk,21116
 visual_rag/embedding/__init__.py,sha256=7QIENmxwRnwnUzsYKRY3VQTyF3HJkRiL1D7Au9XHF0w,682
 visual_rag/embedding/pooling.py,sha256=x8uY4VHbxEnsJRM2JeOkzPHDiwOkbi5NK4XW21U1hAc,11401
 visual_rag/embedding/visual_embedder.py,sha256=he9JpVHmo_szOiXCwtJdrCseGmf2y5Gi0UEFjwazzVY,23198
@@ -44,14 +44,14 @@ visual_rag/indexing/qdrant_indexer.py,sha256=E9uDwOL3A1nFJhFhJcI8V3JR8FL7nDahEYi
 visual_rag/preprocessing/__init__.py,sha256=rCzfBO0jaVKp6MpPRRused_4gasHfobAbG-139Y806E,121
 visual_rag/preprocessing/crop_empty.py,sha256=iHXITFkRlF40VPJ4k9d432RUAi_89BhAEvK4wOEn96Q,5211
 visual_rag/retrieval/__init__.py,sha256=J9pnbeB83Fqs9n4g3GcIp1VR9dnuyAlcsIDVsf0lSb8,601
-visual_rag/retrieval/multi_vector.py,sha256=KyJg1VAxODr5rY80vXmsjSD9ueRpay4QGCJizPKiiBw,7008
-visual_rag/retrieval/single_stage.py,sha256=Ba06V-KRSFSZm0xzbjFR3EBEWaQkDo7U_pWNx25W8H0,4425
-visual_rag/retrieval/three_stage.py,sha256=YC0CVEohxTT5zhilcQHI7nYAk08E5jC3zkQ3-rNdLMw,5951
-visual_rag/retrieval/two_stage.py,sha256=eUbyvdf0-3f0qIiAdU6Wmlos5hGrGNyXjuvc2-Be41M,16775
+visual_rag/retrieval/multi_vector.py,sha256=oQowl-Yv4cYDj2jroX1nv_uZl8sb0d5VKMmwxL-GlBA,7119
+visual_rag/retrieval/single_stage.py,sha256=-hljRha6kWZAr1IQJVSii4pqiIqk0cHhW68bUyYK41o,4472
+visual_rag/retrieval/three_stage.py,sha256=UDjH92V6ILV_w4bOeN0vCF3SdXYHMml9oN2gl3kfwWk,5998
+visual_rag/retrieval/two_stage.py,sha256=OTTH-MYtwK45RHYMF1esEeciXMTU8h7AQzH68BZVrYE,16822
 visual_rag/visualization/__init__.py,sha256=SITKNvBEseDp7F3K6UzLPA-6OQFqYfY5azS5nlDdihQ,447
-visual_rag/visualization/saliency.py,sha256=F3Plc18Sf3tzWcyncuaruTmENm1IfW5j9NFGEQR93cY,11248
-visual_rag_toolkit-0.1.4.dist-info/METADATA,sha256=2nnhqsq3P_OG8GRk48sz569VB1I-aaKcOjKXYl-OvJo,13460
-visual_rag_toolkit-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-visual_rag_toolkit-0.1.4.dist-info/entry_points.txt,sha256=6Tob1GPg_ILGELjYTPsAnNMZ1W0NS939nfI7xyW2DIY,102
-visual_rag_toolkit-0.1.4.dist-info/licenses/LICENSE,sha256=hEg_weKnHXJakQRR3sw2ygcZ101zCI00zMhBOPb3yfA,1069
-visual_rag_toolkit-0.1.4.dist-info/RECORD,,
+visual_rag/visualization/saliency.py,sha256=RuVcF1cPrmzdwfigqhP-e1aqzMr817F64aYucbwT79g,11314
+visual_rag_toolkit-0.1.5.dist-info/METADATA,sha256=66Z99JyOGp5QVb2eJR7U2zjF8Qd6nvRJanhezrbZ-3o,13460
+visual_rag_toolkit-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+visual_rag_toolkit-0.1.5.dist-info/entry_points.txt,sha256=6Tob1GPg_ILGELjYTPsAnNMZ1W0NS939nfI7xyW2DIY,102
+visual_rag_toolkit-0.1.5.dist-info/licenses/LICENSE,sha256=hEg_weKnHXJakQRR3sw2ygcZ101zCI00zMhBOPb3yfA,1069
+visual_rag_toolkit-0.1.5.dist-info/RECORD,,