Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ dependencies = [
"rich>=14.0.0",
"python-dotenv>=1.1.0",
"watchfiles>=1.1.0",
"numpy>=1.23.2",
]
license = "Apache-2.0"
urls = { Homepage = "https://cocoindex.io/" }
Expand Down
15 changes: 6 additions & 9 deletions python/cocoindex/functions.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,17 @@
"""All builtin functions."""

from typing import Annotated, Any, TYPE_CHECKING, Literal
import numpy as np
from numpy.typing import NDArray
import dataclasses
from typing import Annotated, Any, Literal

from .typing import Float32, Vector, TypeAttr
from . import op, llm
import numpy as np
from numpy.typing import NDArray

# Libraries that are heavy to import. Lazily import them later.
if TYPE_CHECKING:
import sentence_transformers
from . import llm, op
from .typing import TypeAttr, Vector

# Check if sentence_transformers is available
try:
import sentence_transformers
import sentence_transformers # type: ignore

_SENTENCE_TRANSFORMERS_AVAILABLE = True
except ImportError:
Expand Down