cs-models 0.0.647__py3-none-any.whl → 0.0.649__py3-none-any.whl

cs_models/resources/DocMarkdownTable/models.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Text,
+    Boolean,
+)
+
+from ...database import Base
+
+
+class DocMarkdownTableModel(Base):
+    __tablename__ = "doc_markdown_tables"
+
+    id = Column(Integer, primary_key=True)
+    artifact_id = Column(String(50), nullable=False, index=True)
+    markdown_table = Column(Text, nullable=False)
+    markdown_table_description = Column(Text, nullable=True)
+    has_table_description = 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/DocMarkdownTable/schemas.py

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

cs_models/resources/PubmedVectorIndex/models.py

@@ -0,0 +1,33 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Boolean,
+    String,
+)
+
+from ...database import Base
+
+
+class PubmedVectorIndexModel(Base):
+    __tablename__ = "pubmed_vector_index"
+
+    id = Column(Integer, primary_key=True)
+    pubmed_id = Column(
+        Integer,
+        ForeignKey('pubmed.id'),
+        nullable=False,
+    )
+    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/PubmedVectorIndex/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class PubmedVectorIndexResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    pubmed_id = fields.Integer(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-0.0.647.dist-info → cs_models-0.0.649.dist-info}/METADATA

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

{cs_models-0.0.647.dist-info → cs_models-0.0.649.dist-info}/RECORD

@@ -254,6 +254,9 @@ cs_models/resources/DocChunk/schemas.py,sha256=fDcASRgqpes9lM5hWd09QXl8dl5bzUrhw
 cs_models/resources/DocEmbedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/DocEmbedding/models.py,sha256=8nkbhp2GE-sZjtg6t6ABHGpRhw5W30BSB-77VvZ8h8k,760
 cs_models/resources/DocEmbedding/schemas.py,sha256=jaQCgkk-hj2HEaadOrRzkZqtwbuqw8bniZ-Vp85o88c,448
+cs_models/resources/DocMarkdownTable/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/DocMarkdownTable/models.py,sha256=Ft52Xt1RudmUrypSldJjgfX9OoztLg2wI_J7baROyOc,792
+cs_models/resources/DocMarkdownTable/schemas.py,sha256=cp4S3HSPFAlsjKfIMIfC5c_jwVOrwd-2I1U2PLvASeg,496
 cs_models/resources/DocMarkdownText/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/DocMarkdownText/models.py,sha256=BQqvn8G8F0SeyY3R_p6EmlBTbWStWYEKKIMP8dB9b84,691
 cs_models/resources/DocMarkdownText/schemas.py,sha256=k5URZhG-0BHpWZehV11LKpyMAkBcAFghIYpzOLISBe0,417
@@ -814,6 +817,9 @@ cs_models/resources/PubmedTarget/schemas.py,sha256=fY6t6JwzSgc95WVPXJUKJz5kYNZoL
 cs_models/resources/PubmedUpdateLedger/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/PubmedUpdateLedger/models.py,sha256=3roSR67RZbpxo5sQ-ZSfYItHInrJaT1lMbSKr95JjOw,744
 cs_models/resources/PubmedUpdateLedger/schemas.py,sha256=rlXKvQNfcBUjcsYZtxbd6xBrAQslNHD5lfaLKpL2rvw,445
+cs_models/resources/PubmedVectorIndex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/PubmedVectorIndex/models.py,sha256=PaCcSubMQijFEm7haiPT13-KMyz4fR0ConRUW1f1dzg,833
+cs_models/resources/PubmedVectorIndex/schemas.py,sha256=U1SFhceJoe2MypkoHp78-Gz__yg4B6F6wvCyMpmTfYk,470
 cs_models/resources/PurpleBookPatent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/PurpleBookPatent/models.py,sha256=PJ8xqjt8_OU75YjU9E1KMzyRFoW32K9izfu7LdeXZ-o,947
 cs_models/resources/PurpleBookPatent/schemas.py,sha256=x2oFaUPqXCJO0RxD-BOlaLAqFtNmZLKd1kiMo7jIXJU,573
@@ -1057,7 +1063,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.647.dist-info/METADATA,sha256=pJFTgrM0aak2TF5WP5P-0BS1ohz3QroGMFHS3E_ecy0,792
-cs_models-0.0.647.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.647.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.647.dist-info/RECORD,,
+cs_models-0.0.649.dist-info/METADATA,sha256=EYlap9wvn37UbYKMudIu0pRzNWvahUfexijr0wGAOpE,792
+cs_models-0.0.649.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.649.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.649.dist-info/RECORD,,