cs-models 0.0.651__py3-none-any.whl → 0.0.653__py3-none-any.whl

cs_models/resources/ArtifactVectorIndex/models.py

@@ -0,0 +1,32 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    Boolean,
+    String,
+)
+
+from ...database import Base
+
+
+class ArtifactVectorIndexModel(Base):
+    __tablename__ = "artifact_vector_index"
+
+    id = Column(Integer, primary_key=True)
+    artifact_id = Column(
+        String(50),
+        nullable=False,
+        index=True,
+    )
+    embedding_created = Column(Boolean, nullable=True)
+    embedding_s3_key = Column(String(128), nullable=True)
+    vector_indexed = Column(Boolean, nullable=True)
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        # https://stackoverflow.com/questions/58776476/why-doesnt-freezegun-work-with-sqlalchemy-default-values
+        default=lambda: datetime.utcnow(),
+        onupdate=lambda: datetime.utcnow(),
+    )

cs_models/resources/ArtifactVectorIndex/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class ArtifactVectorIndexResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    artifact_id = fields.String(required=True)
+    embedding_created = fields.Boolean(allow_none=True)
+    embedding_s3_key = fields.String(allow_none=True)
+    vector_indexed = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/UserExplorerColumn/schemas.py

@@ -4,7 +4,7 @@ from marshmallow import (
 )
 
 
-class UserSavedSearchResourceSchema(Schema):
+class UserExplorerColumnResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     user_id = fields.String(required=True)
     artifact_ids = fields.String(required=True)

{cs_models-0.0.651.dist-info → cs_models-0.0.653.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cs-models
-Version: 0.0.651
+Version: 0.0.653
 Summary: MySQL db models
 Home-page: https://github.com/mindgram/cs-models
 Author: Shrey Verma

{cs_models-0.0.651.dist-info → cs_models-0.0.653.dist-info}/RECORD

@@ -17,6 +17,9 @@ cs_models/resources/ActionLog/schemas.py,sha256=z9tUH9Sg5oq3gC_ZyEhDM6NM1aJYv5b2
 cs_models/resources/AmountLicensingCollab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AmountLicensingCollab/models.py,sha256=-xCbvCGawvbAsrb2wKc8zB19Q4ZMCsWbciWKP5cJwE8,848
 cs_models/resources/AmountLicensingCollab/schemas.py,sha256=sqSIVGNEjWB0WlTrSUMUjCVqfHQvKGbCHQ9XJYuPyX8,452
+cs_models/resources/ArtifactVectorIndex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/ArtifactVectorIndex/models.py,sha256=P8pMWjeGz-sJMIM5ZPdxhas3zfbgq8rEHbz1xJEZ0Ms,813
+cs_models/resources/ArtifactVectorIndex/schemas.py,sha256=v5B4B24hwXSIAb8Fk4q8KKUhcAKhesCAL9Kz7zkVRlI,473
 cs_models/resources/Assignee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Assignee/models.py,sha256=_f0_UwzF1gbOlU0t0l-9UIi0E4vaIS5VsHeOxc698SE,617
 cs_models/resources/Assignee/schemas.py,sha256=N4_upICR1Cd_B45LHpOly7FEwWhAB1fXUP4y60OiezE,269
@@ -916,7 +919,7 @@ cs_models/resources/UserDocumentTarget/models.py,sha256=Wm0iA3lWKIeRSXbOiXzrp-WB
 cs_models/resources/UserDocumentTarget/schemas.py,sha256=xX8DuLwpRVxXjw36WPjJBWwpU1RaDqQShotrlgfUmKw,544
 cs_models/resources/UserExplorerColumn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserExplorerColumn/models.py,sha256=Fnwvqs31tqjd95ouCo0DScdLdLWRuT7J3uq95cwOUtg,811
-cs_models/resources/UserExplorerColumn/schemas.py,sha256=jO644hnn1-5dNHBVO0nZFWnNruH9AbLY7IrkQlYfLLg,414
+cs_models/resources/UserExplorerColumn/schemas.py,sha256=pfL8wGtzO4KjBTuLitwSkgbrty4jvaFZszEC361VIZY,417
 cs_models/resources/UserFile/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserFile/models.py,sha256=-eQ-6VqLl2SDiQ1mpuIxSwt8rnFDF0F-jALulgqZauY,744
 cs_models/resources/UserFile/schemas.py,sha256=UN2-NlmGSW5Xvkphs7vSQjJ0uWjGNXFi_eOlytO1jV8,317
@@ -1069,7 +1072,7 @@ cs_models/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/utils/alchemy.py,sha256=fhINGFn41owJ2DLXQKXAAtLqeZ1BRzD_qU0wPK_bsGQ,1598
 cs_models/utils/utils.py,sha256=bY623DuzycfPQiaOQT2AxfANeWfwr5w76dBuQ813-ns,3664
 cs_models/utils/profiling/__init__.py,sha256=N-73vb0M92C975fxgXyBCBjCPELl8Oh21ZY_-tzDnns,569
-cs_models-0.0.651.dist-info/METADATA,sha256=uY2u1X4QB_kpO9Q0w5xztykGJOwLB0mxctWJ-OGZY5s,792
-cs_models-0.0.651.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.651.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.651.dist-info/RECORD,,
+cs_models-0.0.653.dist-info/METADATA,sha256=3LjiaEfJiqlonxEjXkctJg7OeacjtdaROfl04DRpdcU,792
+cs_models-0.0.653.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.653.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.653.dist-info/RECORD,,