cs-models 0.0.648__py3-none-any.whl → 0.0.650__py3-none-any.whl

cs_models/resources/ExplorerColumn/models.py

@@ -0,0 +1,33 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    Text,
+    Boolean,
+    String,
+)
+
+from ...database import Base
+
+
+class ExplorerColumnModel(Base):
+    __tablename__ = "explorer_columns"
+
+    id = Column(Integer, primary_key=True)
+    artifact_id = Column(
+        String(50),
+        nullable=False,
+        index=True
+    )
+    prompt = Column(String(128), nullable=False, index=True)
+    answer = Column(Text, nullable=True)
+    is_completed = 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/ExplorerColumn/schemas.py

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

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.648.dist-info → cs_models-0.0.650.dist-info}/METADATA

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

{cs_models-0.0.648.dist-info → cs_models-0.0.650.dist-info}/RECORD

@@ -283,6 +283,9 @@ cs_models/resources/EmptyCompany/schemas.py,sha256=U6fjKheQMxosVsS14C3Vz7zVNHHso
 cs_models/resources/Event/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Event/models.py,sha256=wkRLT1tkFjXId8Ga1BiThwflXK5lRaCFqEf_guIwWM0,1227
 cs_models/resources/Event/schemas.py,sha256=63C0x7i-JUu5Q0aAZeNBd1O0_XZ7cahM3KTpkLCbCM8,1690
+cs_models/resources/ExplorerColumn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/ExplorerColumn/models.py,sha256=t448jSUDUNpN3MPlA1-nf2H-K9KVfMkKIV7ruB-DSks,799
+cs_models/resources/ExplorerColumn/schemas.py,sha256=0qsNjOCkilZNa_Ly083dzp43h72MrAFM39dp2s8CnmY,442
 cs_models/resources/FDALabel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/FDALabel/models.py,sha256=eocy19GuoNjFZwLXvNuES3sLrNUnHo3kAC7A4k5hxxs,1123
 cs_models/resources/FDALabel/schemas.py,sha256=ZIDxBVsJ-Ha7ao7EVbfcGT2cukn73fy4ZiUhMaXWKk8,693
@@ -817,6 +820,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
@@ -1060,7 +1066,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.648.dist-info/METADATA,sha256=ygx8gCFHxXf4Vd01p1izSVA_tJG6EF6ktllAUckhKWU,792
-cs_models-0.0.648.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.648.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.648.dist-info/RECORD,,
+cs_models-0.0.650.dist-info/METADATA,sha256=dw_g1zQeB-4UtBhdZK9OW9P5jxq7ibHxaVW_X6uENqg,792
+cs_models-0.0.650.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.650.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.650.dist-info/RECORD,,