cocoindex 0.1.66__cp313-cp313-macosx_11_0_arm64.whl → 0.1.67__cp313-cp313-macosx_11_0_arm64.whl

cocoindex/functions.py

@@ -9,14 +9,6 @@ from numpy.typing import NDArray
 from . import llm, op
 from .typing import TypeAttr, Vector
 
-# Check if sentence_transformers is available
-try:
-    import sentence_transformers  # type: ignore
-
-    _SENTENCE_TRANSFORMERS_AVAILABLE = True
-except ImportError:
-    _SENTENCE_TRANSFORMERS_AVAILABLE = False
-
 
 class ParseJson(op.FunctionSpec):
     """Parse a text into a JSON object."""
@@ -79,18 +71,19 @@ class SentenceTransformerEmbedExecutor:
     """Executor for SentenceTransformerEmbed."""
 
     spec: SentenceTransformerEmbed
-    _model: "sentence_transformers.SentenceTransformer"
+    _model: Any | None = None
 
     def analyze(self, text: Any) -> type:
-        if not _SENTENCE_TRANSFORMERS_AVAILABLE:
+        try:
+            # Only import sentence_transformers locally when it's needed, as its import is very slow.
+            import sentence_transformers  # pylint: disable=import-outside-toplevel
+        except ImportError as e:
             raise ImportError(
                 "sentence_transformers is required for SentenceTransformerEmbed function. "
                 "Install it with one of these commands:\n"
                 "  pip install 'cocoindex[embeddings]'\n"
                 "  pip install sentence-transformers"
-            )
-
-        import sentence_transformers  # pylint: disable=import-outside-toplevel
+            ) from e
 
         args = self.spec.args or {}
         self._model = sentence_transformers.SentenceTransformer(self.spec.model, **args)
@@ -102,5 +95,6 @@ class SentenceTransformerEmbedExecutor:
         return result
 
     def __call__(self, text: str) -> NDArray[np.float32]:
+        assert self._model is not None
         result: NDArray[np.float32] = self._model.encode(text, convert_to_numpy=True)
         return result

{cocoindex-0.1.66.dist-info → cocoindex-0.1.67.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cocoindex
-Version: 0.1.66
+Version: 0.1.67
 Requires-Dist: click>=8.1.8
 Requires-Dist: rich>=14.0.0
 Requires-Dist: python-dotenv>=1.1.0
@@ -10,9 +10,11 @@ Requires-Dist: pytest ; extra == 'test'
 Requires-Dist: ruff ; extra == 'dev'
 Requires-Dist: pre-commit ; extra == 'dev'
 Requires-Dist: sentence-transformers>=3.3.1 ; extra == 'embeddings'
+Requires-Dist: cocoindex[embeddings] ; extra == 'all'
 Provides-Extra: test
 Provides-Extra: dev
 Provides-Extra: embeddings
+Provides-Extra: all
 License-File: LICENSE
 Summary: With CocoIndex, users declare the transformation, CocoIndex creates & maintains an index, and keeps the derived index up to date based on source update, with minimal computation and changes.
 Author-email: CocoIndex <cocoindex.io@gmail.com>

{cocoindex-0.1.66.dist-info → cocoindex-0.1.67.dist-info}/RECORD

@@ -1,14 +1,14 @@
-cocoindex-0.1.66.dist-info/METADATA,sha256=ZwdUYvaVBt5qmZ7pQqJOB6FoszGiwbEerJyeI52aSBE,10098
-cocoindex-0.1.66.dist-info/WHEEL,sha256=ZlnHGE4Mo7ps1v16zvcAfrDnL3ds3Klm21sZZ52w3vY,104
-cocoindex-0.1.66.dist-info/entry_points.txt,sha256=_NretjYVzBdNTn7dK-zgwr7YfG2afz1u1uSE-5bZXF8,46
-cocoindex-0.1.66.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+cocoindex-0.1.67.dist-info/METADATA,sha256=SJIeiMwnbmOKiJSenH-DSoPatDM3bh6SD7B08GdI5Xs,10172
+cocoindex-0.1.67.dist-info/WHEEL,sha256=ZlnHGE4Mo7ps1v16zvcAfrDnL3ds3Klm21sZZ52w3vY,104
+cocoindex-0.1.67.dist-info/entry_points.txt,sha256=_NretjYVzBdNTn7dK-zgwr7YfG2afz1u1uSE-5bZXF8,46
+cocoindex-0.1.67.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 cocoindex/__init__.py,sha256=nr-W-LX0e9EX6ZVMbf0MRF0EhtntqADOdwc9F2nwfWw,2004
-cocoindex/_engine.cpython-313-darwin.so,sha256=RSVwNx7kbT3cIbY-htck5st9U_6B9CfWe1iZV8xdKhQ,61479184
+cocoindex/_engine.cpython-313-darwin.so,sha256=ADERPLopodjUBwfyjAXUqTXkgP0KDUCAq-tHeFquXXY,61479024
 cocoindex/auth_registry.py,sha256=1XqO7ibjmBBd8i11XSJTvTgdz8p1ptW-ZpuSgo_5zzk,716
 cocoindex/cli.py,sha256=-gp639JSyQN6YjnhGqCakIzYoSSqXxQMbxbkcYGP0QY,22359
 cocoindex/convert.py,sha256=RYfRUungabr-dHakG4k2kDvYambxHFljAmTuPQeQths,13117
 cocoindex/flow.py,sha256=MFPtfJBVTjQ56d7vUn2LvtY30Vg4q2rY6nqvjjJL1kQ,35085
-cocoindex/functions.py,sha256=EvRnd2Y3TIeEueYjOmproBG5RinlbqK6ym1Hnrmnm-0,3251
+cocoindex/functions.py,sha256=Ih1rtaTvZzQ2wucCOSpzSUu2-eu0mgtBVi1mh9M-Buw,3162
 cocoindex/index.py,sha256=j93B9jEvvLXHtpzKWL88SY6wCGEoPgpsQhEGHlyYGFg,540
 cocoindex/lib.py,sha256=f--9dAYd84CZosbDZqNW0oGbBLsY3dXiUTR1VrfQ_QY,817
 cocoindex/llm.py,sha256=WxmWUbNcf9HOCM5xkbDeFs9lF67M3mr810B7deDDc-8,673
@@ -25,4 +25,4 @@ cocoindex/tests/test_optional_database.py,sha256=snAmkNa6wtOSaxoZE1HgjvL5v_ylitt
 cocoindex/tests/test_typing.py,sha256=NB4nUzoumOF_wGFa4D2Xf6d0bUVtOiSXyb78M1pYSG4,14827
 cocoindex/typing.py,sha256=MO9HkrNpargvMPvpkd7jgSu2R-21KE_NaB9-WI4YOZA,13241
 cocoindex/utils.py,sha256=hUhX-XV6XGCtJSEIpBOuDv6VvqImwPlgBxztBTw7u0U,598
-cocoindex-0.1.66.dist-info/RECORD,,
+cocoindex-0.1.67.dist-info/RECORD,,