corvic-engine 0.3.0rc79__tar.gz → 0.3.0rc80__tar.gz

{corvic_engine-0.3.0rc79 → corvic_engine-0.3.0rc80}/Cargo.lock

@@ -16,7 +16,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.79"
+version = "0.3.0-rc.80"
 dependencies = [
  "numpy",
  "pyo3",
@@ -57,7 +57,7 @@ checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.79"
+version = "0.3.0-rc.80"
 dependencies = [
  "version_check",
 ]

{corvic_engine-0.3.0rc79 → corvic_engine-0.3.0rc80}/Cargo.toml

@@ -3,7 +3,7 @@ resolver = "2"
 members = ["python"]
 
 [workspace.package]
-version = "0.3.0-rc.79"
+version = "0.3.0-rc.80"
 edition = "2021"
 publish = false
 

{corvic_engine-0.3.0rc79 → corvic_engine-0.3.0rc80}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corvic-engine
-Version: 0.3.0rc79
+Version: 0.3.0rc80
 Classifier: Environment :: Console
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: Implementation :: CPython

{corvic_engine-0.3.0rc79 → corvic_engine-0.3.0rc80}/pyproject.toml

@@ -18,7 +18,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.79" # sync this with tool.poetry.version below
+version = "0.3.0-rc.80" # sync this with tool.poetry.version below
 
 # TODO(ddn): Pull dependencies from tool.poetry.dependencies. We use maturin as
 # the build system but poetry for the development environment.
@@ -72,7 +72,7 @@ packages = [
 ]
 description = ""
 authors = []
-version = "0.3.0-rc.79"
+version = "0.3.0-rc.80"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"

{corvic_engine-0.3.0rc79 → corvic_engine-0.3.0rc80}/python/corvic/orm/_soft_delete.py

@@ -81,6 +81,11 @@ class SoftDeleteMixin(sa_orm.MappedAsDataclass):
             return None
         return self._deleted_at.replace(tzinfo=UTC)
 
+    @deleted_at.inplace.expression
+    @classmethod
+    def _deleted_at_at_expression(cls):
+        return cls._deleted_at
+
     def reset_delete(self):
         self._deleted_at = None
 

{corvic_engine-0.3.0rc79 → corvic_engine-0.3.0rc80}/python/corvic/system/_image_embedder.py

@@ -180,17 +180,53 @@ class Clip(ImageEmbedder):
         )
 
 
+class SigLIP2(Clip):
+    """SigLIP2 image embedder."""
+
+    def _load_models(self):
+        from transformers.models.auto.modeling_auto import AutoModel
+        from transformers.models.auto.processing_auto import AutoProcessor
+        from transformers.models.clip import (
+            CLIPModel,
+            CLIPProcessor,
+        )
+
+        model = cast(
+            CLIPModel,
+            AutoModel.from_pretrained(  # pyright: ignore[reportUnknownMemberType]
+                pretrained_model_name_or_path="google/siglip2-base-patch16-512",
+                revision="a89f5c5093f902bf39d3cd4d81d2c09867f0724b",
+                device_map="auto",
+            ),
+        )
+        processor = cast(
+            CLIPProcessor,
+            AutoProcessor.from_pretrained(  # pyright: ignore[reportUnknownMemberType]
+                pretrained_model_name_or_path="google/siglip2-base-patch16-512",
+                revision="a89f5c5093f902bf39d3cd4d81d2c09867f0724b",
+            ),
+        )
+        return ClipModels(model=model, processor=processor)
+
+
 class CombinedImageEmbedder(ImageEmbedder):
     def __init__(self):
         self._clip_embedder = Clip()
+        self._siglip2_embedder = SigLIP2()
         self._random_embedder = RandomImageEmbedder()
 
     def embed(
         self, context: EmbedImageContext
     ) -> Ok[EmbedImageResult] | InvalidArgumentError | InternalError:
-        if context.model_name == "random":
-            return self._random_embedder.embed(context)
-        return self._clip_embedder.embed(context)
+        match context.model_name:
+            case "random":
+                return self._random_embedder.embed(context)
+            case "clip":
+                return self._clip_embedder.embed(context)
+            case "siglip2":
+                return self._siglip2_embedder.embed(context)
+            case _:
+                return InvalidArgumentError(f"Unknown model name {context.model_name}")
 
     async def aembed(
         self,