corvic-engine 0.3.0rc86__cp38-abi3-win_amd64.whl → 0.3.0rc87__cp38-abi3-win_amd64.whl

corvic/emodel/_space.py

@@ -37,13 +37,15 @@ _DEFAULT_CONCAT_SEPARATOR = " "
 
 
 embedding_model_proto_to_name: Final[dict[embedding_models_pb2.Model, str]] = {
-    embedding_models_pb2.MODEL_CUSTOM: "random",
+    embedding_models_pb2.MODEL_CUSTOM: system.RandomTextEmbedder.model_name(),
     embedding_models_pb2.MODEL_SENTENCE_TRANSFORMER: "text-embedding-004",
     embedding_models_pb2.MODEL_GCP_TEXT_EMBEDDING_004: "text-embedding-004",
     embedding_models_pb2.MODEL_OPENAI_TEXT_EMBEDDING_3_SMALL: "text-embedding-3-small",
     embedding_models_pb2.MODEL_OPENAI_TEXT_EMBEDDING_3_LARGE: "text-embedding-3-large",
-    embedding_models_pb2.MODEL_IDENTITY: "identity",
+    embedding_models_pb2.MODEL_IDENTITY: system.IdentityTextEmbedder.model_name(),
     embedding_models_pb2.MODEL_UNSPECIFIED: "",
+    embedding_models_pb2.MODEL_CLIP: system.ClipText.model_name(),
+    embedding_models_pb2.MODEL_SIGLIP2: system.SigLIP2Text.model_name(),
 }
 name_to_proto_embedding_model = {
     name: model for model, name in embedding_model_proto_to_name.items()

corvic/system/_embedder.py

@@ -39,6 +39,9 @@ class EmbedTextResult:
 class TextEmbedder(Protocol):
     """Use a model to embed text."""
 
+    @classmethod
+    def model_name(cls) -> str: ...
+
     async def aembed(
         self,
         context: EmbedTextContext,
@@ -86,8 +89,12 @@ class LoadedModels:
 class HFModelText(TextEmbedder):
     """Generic text/image embedder from hugging face models."""
 
+    @classmethod
     @abc.abstractmethod
-    def _load_models(self) -> LoadedModels: ...
+    def model_revision(cls) -> str: ...
+
+    @abc.abstractmethod
+    def load_models(self) -> LoadedModels: ...
 
     def embed(
         self, context: EmbedTextContext
@@ -98,7 +105,7 @@ class HFModelText(TextEmbedder):
             case 32:
                 coord_dtype = pl.Float32()
 
-        models = self._load_models()
+        models = self.load_models()
         model = models.model
         processor = models.processor
         model.eval()  # type: ignore[reportAttributeAccess]
@@ -152,7 +159,15 @@ class ClipText(HFModelText):
     overcoming several major challenges in computer vision.
     """
 
-    def _load_models(self):
+    @classmethod
+    def model_name(cls) -> str:
+        return "openai/clip-vit-base-patch32"
+
+    @classmethod
+    def model_revision(cls) -> str:
+        return "5812e510083bb2d23fa43778a39ac065d205ed4d"
+
+    def load_models(self):
         from transformers.models.clip import (
             CLIPModel,
             CLIPProcessor,
@@ -161,15 +176,15 @@ class ClipText(HFModelText):
         model = cast(
             AutoModel,
             CLIPModel.from_pretrained(  # pyright: ignore[reportUnknownMemberType]
-                pretrained_model_name_or_path="openai/clip-vit-base-patch32",
-                revision="5812e510083bb2d23fa43778a39ac065d205ed4d",
+                pretrained_model_name_or_path=self.model_name(),
+                revision=self.model_revision(),
             ),
         )
         processor = cast(
             AutoProcessor,
             CLIPProcessor.from_pretrained(  # pyright: ignore[reportUnknownMemberType]
-                pretrained_model_name_or_path="openai/clip-vit-base-patch32",
-                revision="5812e510083bb2d23fa43778a39ac065d205ed4d",
+                pretrained_model_name_or_path=self.model_name(),
+                revision=self.model_revision(),
                 use_fast=False,
             ),
         )
@@ -179,23 +194,31 @@ class ClipText(HFModelText):
 class SigLIP2Text(HFModelText):
     """SigLIP2 text/image embedder."""
 
-    def _load_models(self):
+    @classmethod
+    def model_name(cls) -> str:
+        return "google/siglip2-base-patch16-512"
+
+    @classmethod
+    def model_revision(cls) -> str:
+        return "a89f5c5093f902bf39d3cd4d81d2c09867f0724b"
+
+    def load_models(self):
         from transformers.models.auto.modeling_auto import AutoModel
         from transformers.models.auto.processing_auto import AutoProcessor
 
         model = cast(
             AutoModel,
             AutoModel.from_pretrained(  # pyright: ignore[reportUnknownMemberType]
-                pretrained_model_name_or_path="google/siglip2-base-patch16-512",
-                revision="a89f5c5093f902bf39d3cd4d81d2c09867f0724b",
+                pretrained_model_name_or_path=self.model_name(),
+                revision=self.model_revision(),
                 device_map="auto",
             ),
         )
         processor = cast(
             AutoProcessor,
             AutoProcessor.from_pretrained(  # pyright: ignore[reportUnknownMemberType]
-                pretrained_model_name_or_path="google/siglip2-base-patch16-512",
-                revision="a89f5c5093f902bf39d3cd4d81d2c09867f0724b",
+                pretrained_model_name_or_path=self.model_name(),
+                revision=self.model_revision(),
                 use_fast=True,
             ),
         )

corvic/system/_text_embedder.py

@@ -18,6 +18,10 @@ class RandomTextEmbedder(TextEmbedder):
     Useful for testing.
     """
 
+    @classmethod
+    def model_name(cls) -> str:
+        return "random"
+
     def embed(
         self, context: EmbedTextContext
     ) -> Ok[EmbedTextResult] | InvalidArgumentError | InternalError:
@@ -62,6 +66,10 @@ class IdentityTextEmbedder(TextEmbedder):
     - The resulting list is truncated or padded to match the expected vector length.
     """
 
+    @classmethod
+    def model_name(cls) -> str:
+        return "identity"
+
     def _text_to_embedding(
         self, text: str, vector_length: int, *, normalization: bool = False
     ) -> list[float]:

{corvic_engine-0.3.0rc86.dist-info → corvic_engine-0.3.0rc87.dist-info}/METADATA

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

{corvic_engine-0.3.0rc86.dist-info → corvic_engine-0.3.0rc87.dist-info}/RECORD

@@ -22,10 +22,10 @@ corvic/emodel/_proto_orm_convert.py,sha256=F5Q6Aa111muD7qkxkIaBRE_wzbJxJSAb8mwGd
 corvic/emodel/_resource.py,sha256=fovtKflZURa2o5Q-M9uoo9Ta_iIHO-sWknfK5iYyUMM,8508
 corvic/emodel/_room.py,sha256=D5V9YSo9Xcfnng-dACWAmpEBgUAOUVFS7Jj2bCsSXtM,2877
 corvic/emodel/_source.py,sha256=Fy6G7NlGEz3G8asBPW8b0wzumUirPzVSYH4aUw80Yd8,9795
-corvic/emodel/_space.py,sha256=3RAhKFuymGGR9XP35Raxom--GzVOBM_7CYYQ8Fx0h2k,38545
+corvic/emodel/_space.py,sha256=tzHHoVxIt_t13X2KAfZsjKJcgenhXOx_Igz-uM_BvYM,38747
 corvic/emodel/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/engine/__init__.py,sha256=XL4Vg7rNcBi29ccVelpeFizR9oJtGYXDn84W9zok9d4,975
-corvic/engine/_native.pyd,sha256=ahrw_mE1kyNZPiJutq0wumAlKmtv9YuB4pUePT6ACrc,438272
+corvic/engine/_native.pyd,sha256=l-rTCgL--9bObrMS0n8cqa0AhPf-3ZWdGFZNeWcBQr4,438272
 corvic/engine/_native.pyi,sha256=KYMPtvXqHZ-jMgZohLf4se3rr-rBpCihmjANcr6s8ag,1390
 corvic/engine/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/eorm/__init__.py,sha256=b4dFnu4fW7wj3Y0SMNVXOp8KoKOp_HAL4GyDN-S8fOY,13704
@@ -71,10 +71,10 @@ corvic/sql/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/system/__init__.py,sha256=PlsjTRklExrPA83n-PM5CqVmNcIQdIRhFHn3tZ0qh1Q,2765
 corvic/system/_column_encoding.py,sha256=feSWIv4vKstVq-aavWPk53YucUiq7rZvuyofqTicXBE,7574
 corvic/system/_dimension_reduction.py,sha256=2tg5SIHY4P480DJQj6PSjW1VgAJCAVJAH8D3BY-ZYXA,2964
-corvic/system/_embedder.py,sha256=5_dXBxfeXnjuCRexTLhGWVpZ9OW9CLZDcYzxZMHgiD0,6653
+corvic/system/_embedder.py,sha256=CTXeVAng-PTmqCki4SqOEhCIZh2ha9_5qVAx71Ik0so,7085
 corvic/system/_image_embedder.py,sha256=O2hOEtSIoQRrf_xoRnPpSNGn_-aGZC-X1DPfuWykWPU,13003
 corvic/system/_planner.py,sha256=ecL-HW8PVz5eWJ1Ktf-RAD2IdZkHu3GuBtXdqElo4ts,8210
-corvic/system/_text_embedder.py,sha256=NDi--3_tzwIWVImjhFWmp8dHmydGGXNu6GYH8qODsIc,4000
+corvic/system/_text_embedder.py,sha256=vH5ah_KCOdCTjiX0M5ZBkFM_KyLf9ep58cJ6r6Plpw8,4158
 corvic/system/client.py,sha256=JcA-fPraqDkl9f8BiClS0qeGY6wzKcEDPymutWrJo54,812
 corvic/system/in_memory_executor.py,sha256=4Wyi5pTCBVlsXJFansx1LT7OmfubzN7Hg6j1Uw8_8wo,66410
 corvic/system/op_graph_executor.py,sha256=tSKro-yb_y1_sgajZluM-6FCvDqO1oUPsiWw2DRxyMQ,3641
@@ -98,9 +98,9 @@ corvic/version/__init__.py,sha256=JlkRLvKXsu3zIxhdynO_0Ub5NfQOvGjfwCRkNnaOu9U,11
 corvic/version/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/well_known_types/__init__.py,sha256=Btbeqieik2AcmijeOXeqBptzueBpgNitvH9J5VNm12w,1289
 corvic/well_known_types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-corvic_engine-0.3.0rc86.dist-info/METADATA,sha256=mDZiK415Qn0OeeMzRMEP-dy-nnFKgg4_xCOp9C2uvUA,1879
-corvic_engine-0.3.0rc86.dist-info/WHEEL,sha256=U8GAVPGuk_i_fNfXAys1dgH7mS1u7uA5UI-r13T9uEs,94
-corvic_engine-0.3.0rc86.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
+corvic_engine-0.3.0rc87.dist-info/METADATA,sha256=DBk6PPLIH5tSz7_JxQQ_IhqKVsXGgOtuFD2Z8_iZ-j0,1879
+corvic_engine-0.3.0rc87.dist-info/WHEEL,sha256=U8GAVPGuk_i_fNfXAys1dgH7mS1u7uA5UI-r13T9uEs,94
+corvic_engine-0.3.0rc87.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
 corvic_generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/algorithm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/algorithm/graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -111,8 +111,8 @@ corvic_generated/algorithm/graph/v1/graph_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3M
 corvic_generated/algorithm/graph/v1/graph_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/embedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/embedding/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-corvic_generated/embedding/v1/models_pb2.py,sha256=i0gLpcb4LLBE11C4MIMFgWaM0fdwq9rFnYFVOmuWHiY,4481
-corvic_generated/embedding/v1/models_pb2.pyi,sha256=NzQ1TuhJ_IKp2xYjdf1qxu9VgwyMza7WPcKMneI-Ycg,4351
+corvic_generated/embedding/v1/models_pb2.py,sha256=vQp2zsXPcrhTJSef_BoitpZMz6BZtzt2YoRhjapbdU0,4544
+corvic_generated/embedding/v1/models_pb2.pyi,sha256=lHNZ5wIfckzgW8tGukm8D2r-qFW7qGYxQRDHEt_g4yo,4463
 corvic_generated/embedding/v1/models_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
 corvic_generated/embedding/v1/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/feature/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -211,4 +211,4 @@ corvic_generated/status/v1/service_pb2.py,sha256=CKXPX2ahq8O4cFhPpt6wo6l--6VZcgj
 corvic_generated/status/v1/service_pb2.pyi,sha256=iXLR2FOKQJpBgvBzpD2kVwcYOCksP2aRwK4JYaI9CBw,558
 corvic_generated/status/v1/service_pb2_grpc.py,sha256=y-a5ldrphWlNJW-yKswyjNmXokK4-5bbEEfczjagJHo,2736
 corvic_generated/status/v1/service_pb2_grpc.pyi,sha256=OoAnaZ64FD0UTzPoRhYvQU8ecoilhHj3ySjSfHbVDaU,1501
-corvic_engine-0.3.0rc86.dist-info/RECORD,,
+corvic_engine-0.3.0rc87.dist-info/RECORD,,

corvic_generated/embedding/v1/models_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n corvic/embedding/v1/models.proto\x12\x13\x63orvic.embedding.v1\"R\n\nParameters\x12\x30\n\x05model\x18\x01 \x01(\x0e\x32\x1a.corvic.embedding.v1.ModelR\x05model\x12\x12\n\x04ndim\x18\x02 \x01(\x05R\x04ndim\"\xf4\x01\n\x19\x43olumnEmbeddingParameters\x12q\n\x11\x63olumn_parameters\x18\x01 \x03(\x0b\x32\x44.corvic.embedding.v1.ColumnEmbeddingParameters.ColumnParametersEntryR\x10\x63olumnParameters\x1a\x64\n\x15\x43olumnParametersEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.corvic.embedding.v1.ParametersR\x05value:\x02\x38\x01\"\x8f\x01\n\x1e\x43oncatStringAndEmbedParameters\x12!\n\x0c\x63olumn_names\x18\x01 \x03(\tR\x0b\x63olumnNames\x12J\n\x10model_parameters\x18\x02 \x01(\x0b\x32\x1f.corvic.embedding.v1.ParametersR\x0fmodelParameters\"\x89\x01\n\x18\x43oncatAndEmbedParameters\x12!\n\x0c\x63olumn_names\x18\x01 \x03(\tR\x0b\x63olumnNames\x12J\n\x10model_parameters\x18\x02 \x01(\x0b\x32\x1f.corvic.embedding.v1.ParametersR\x0fmodelParameters\".\n\x18\x45mbedAndConcatParameters\x12\x12\n\x04ndim\x18\x01 \x01(\x05R\x04ndim\"a\n\x14ImageModelParameters\x12\x35\n\x05model\x18\x01 \x01(\x0e\x32\x1f.corvic.embedding.v1.ImageModelR\x05model\x12\x12\n\x04ndim\x18\x02 \x01(\x05R\x04ndim\"\x8d\x01\n\x14\x45mbedImageParameters\x12\x1f\n\x0b\x63olumn_name\x18\x01 \x01(\tR\ncolumnName\x12T\n\x10model_parameters\x18\x02 \x01(\x0b\x32).corvic.embedding.v1.ImageModelParametersR\x0fmodelParameters*\xdc\x01\n\x05Model\x12\x15\n\x11MODEL_UNSPECIFIED\x10\x00\x12\"\n\x1aMODEL_SENTENCE_TRANSFORMER\x10\x01\x1a\x02\x08\x01\x12\'\n#MODEL_OPENAI_TEXT_EMBEDDING_3_SMALL\x10\x03\x12\'\n#MODEL_OPENAI_TEXT_EMBEDDING_3_LARGE\x10\x04\x12 \n\x1cMODEL_GCP_TEXT_EMBEDDING_004\x10\x05\x12\x10\n\x0cMODEL_CUSTOM\x10\x02\x12\x12\n\x0eMODEL_IDENTITY\x10\x06*\x8a\x01\n\nImageModel\x12\x1b\n\x17IMAGE_MODEL_UNSPECIFIED\x10\x00\x12\x14\n\x10IMAGE_MODEL_CLIP\x10\x01\x12\x16\n\x12IMAGE_MODEL_CUSTOM\x10\x02\x12\x18\n\x14IMAGE_MODEL_IDENTITY\x10\x03\x12\x17\n\x13IMAGE_MODEL_SIGLIP2\x10\x04\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n corvic/embedding/v1/models.proto\x12\x13\x63orvic.embedding.v1\"R\n\nParameters\x12\x30\n\x05model\x18\x01 \x01(\x0e\x32\x1a.corvic.embedding.v1.ModelR\x05model\x12\x12\n\x04ndim\x18\x02 \x01(\x05R\x04ndim\"\xf4\x01\n\x19\x43olumnEmbeddingParameters\x12q\n\x11\x63olumn_parameters\x18\x01 \x03(\x0b\x32\x44.corvic.embedding.v1.ColumnEmbeddingParameters.ColumnParametersEntryR\x10\x63olumnParameters\x1a\x64\n\x15\x43olumnParametersEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.corvic.embedding.v1.ParametersR\x05value:\x02\x38\x01\"\x8f\x01\n\x1e\x43oncatStringAndEmbedParameters\x12!\n\x0c\x63olumn_names\x18\x01 \x03(\tR\x0b\x63olumnNames\x12J\n\x10model_parameters\x18\x02 \x01(\x0b\x32\x1f.corvic.embedding.v1.ParametersR\x0fmodelParameters\"\x89\x01\n\x18\x43oncatAndEmbedParameters\x12!\n\x0c\x63olumn_names\x18\x01 \x03(\tR\x0b\x63olumnNames\x12J\n\x10model_parameters\x18\x02 \x01(\x0b\x32\x1f.corvic.embedding.v1.ParametersR\x0fmodelParameters\".\n\x18\x45mbedAndConcatParameters\x12\x12\n\x04ndim\x18\x01 \x01(\x05R\x04ndim\"a\n\x14ImageModelParameters\x12\x35\n\x05model\x18\x01 \x01(\x0e\x32\x1f.corvic.embedding.v1.ImageModelR\x05model\x12\x12\n\x04ndim\x18\x02 \x01(\x05R\x04ndim\"\x8d\x01\n\x14\x45mbedImageParameters\x12\x1f\n\x0b\x63olumn_name\x18\x01 \x01(\tR\ncolumnName\x12T\n\x10model_parameters\x18\x02 \x01(\x0b\x32).corvic.embedding.v1.ImageModelParametersR\x0fmodelParameters*\xff\x01\n\x05Model\x12\x15\n\x11MODEL_UNSPECIFIED\x10\x00\x12\"\n\x1aMODEL_SENTENCE_TRANSFORMER\x10\x01\x1a\x02\x08\x01\x12\'\n#MODEL_OPENAI_TEXT_EMBEDDING_3_SMALL\x10\x03\x12\'\n#MODEL_OPENAI_TEXT_EMBEDDING_3_LARGE\x10\x04\x12 \n\x1cMODEL_GCP_TEXT_EMBEDDING_004\x10\x05\x12\x10\n\x0cMODEL_CUSTOM\x10\x02\x12\x12\n\x0eMODEL_IDENTITY\x10\x06\x12\x0e\n\nMODEL_CLIP\x10\x07\x12\x11\n\rMODEL_SIGLIP2\x10\x08*\x8a\x01\n\nImageModel\x12\x1b\n\x17IMAGE_MODEL_UNSPECIFIED\x10\x00\x12\x14\n\x10IMAGE_MODEL_CLIP\x10\x01\x12\x16\n\x12IMAGE_MODEL_CUSTOM\x10\x02\x12\x18\n\x14IMAGE_MODEL_IDENTITY\x10\x03\x12\x17\n\x13IMAGE_MODEL_SIGLIP2\x10\x04\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -26,9 +26,9 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_COLUMNEMBEDDINGPARAMETERS_COLUMNPARAMETERSENTRY']._options = None
   _globals['_COLUMNEMBEDDINGPARAMETERS_COLUMNPARAMETERSENTRY']._serialized_options = b'8\001'
   _globals['_MODEL']._serialized_start=966
-  _globals['_MODEL']._serialized_end=1186
-  _globals['_IMAGEMODEL']._serialized_start=1189
-  _globals['_IMAGEMODEL']._serialized_end=1327
+  _globals['_MODEL']._serialized_end=1221
+  _globals['_IMAGEMODEL']._serialized_start=1224
+  _globals['_IMAGEMODEL']._serialized_end=1362
   _globals['_PARAMETERS']._serialized_start=57
   _globals['_PARAMETERS']._serialized_end=139
   _globals['_COLUMNEMBEDDINGPARAMETERS']._serialized_start=142

corvic_generated/embedding/v1/models_pb2.pyi

@@ -15,6 +15,8 @@ class Model(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     MODEL_GCP_TEXT_EMBEDDING_004: _ClassVar[Model]
     MODEL_CUSTOM: _ClassVar[Model]
     MODEL_IDENTITY: _ClassVar[Model]
+    MODEL_CLIP: _ClassVar[Model]
+    MODEL_SIGLIP2: _ClassVar[Model]
 
 class ImageModel(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
@@ -30,6 +32,8 @@ MODEL_OPENAI_TEXT_EMBEDDING_3_LARGE: Model
 MODEL_GCP_TEXT_EMBEDDING_004: Model
 MODEL_CUSTOM: Model
 MODEL_IDENTITY: Model
+MODEL_CLIP: Model
+MODEL_SIGLIP2: Model
 IMAGE_MODEL_UNSPECIFIED: ImageModel
 IMAGE_MODEL_CLIP: ImageModel
 IMAGE_MODEL_CUSTOM: ImageModel