new: add late interaction multimodal to package init (#471)

qdrant · Feb 12, 2025 · de4ecb4 · de4ecb4
1 parent b9d6051
commit de4ecb4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/fastembed/__init__.py b/fastembed/__init__.py
@@ -2,6 +2,7 @@
 
 from fastembed.image import ImageEmbedding
 from fastembed.late_interaction import LateInteractionTextEmbedding
+from fastembed.late_interaction_multimodal import LateInteractionMultimodalEmbedding
 from fastembed.sparse import SparseEmbedding, SparseTextEmbedding
 from fastembed.text import TextEmbedding
 
@@ -17,4 +18,5 @@
     "SparseEmbedding",
     "ImageEmbedding",
     "LateInteractionTextEmbedding",
+    "LateInteractionMultimodalEmbedding",
 ]
diff --git a/tests/test_late_interaction_multimodal.py b/tests/test_late_interaction_multimodal.py
@@ -3,7 +3,7 @@
 from PIL import Image
 import numpy as np
 
-from fastembed.late_interaction_multimodal import LateInteractionMultimodalEmbedding
+from fastembed import LateInteractionMultimodalEmbedding
 from tests.config import TEST_MISC_DIR