cs-models 0.0.706__py3-none-any.whl → 0.0.708__py3-none-any.whl

cs_models/resources/Figure/models.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    String,
+    Text,
+    Boolean,
+    ForeignKey,
+)
+
+from ...database import Base
+
+
+class FigureModel(Base):
+    __tablename__ = "figures"
+
+    id = Column(Integer, primary_key=True)
+    file_id = Column(
+        Integer,
+        ForeignKey('files.id'),
+        nullable=True,
+    )
+    source_type = Column(String(50), nullable=False)
+    source_id = Column(Integer, nullable=False)
+    figure_label = Column(String(50), nullable=True)
+    figure_caption = Column(Text, nullable=True)
+    figure_description = Column(Text, nullable=True)
+    figure_link = Column(String(255), nullable=True)
+    llm_processed = 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/Figure/schemas.py

@@ -0,0 +1,20 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class FigureResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    file_id = fields.Integer(required=True)
+    source_type = fields.String(required=True)
+    source_id = fields.Integer(required=True)
+    figure_label = fields.String(allow_none=True)
+    figure_caption = fields.String(allow_none=True)
+    figure_description = fields.String(allow_none=True)
+    figure_link = fields.String(allow_none=True)
+    llm_processed = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/UserDocumentHierarchy/models.py

@@ -16,6 +16,7 @@ class UserDocumentHierarchyModel(Base):
     __tablename__ = 'user_documents_hierarchy'
 
     id = Column(Integer, primary_key=True)
+    provider_id = Column(String(128), nullable=False, index=True)
     user_id = Column(String(128), nullable=False, index=True)
     type = Column(String(50), nullable=False)
     document_name = Column(String(255), nullable=True)

cs_models/resources/UserDocumentHierarchy/schemas.py

@@ -9,6 +9,7 @@ class UserDocumentHierarchyResourceSchema(Schema):
     not_blank = validate.Length(min=1, error='Field cannot be blank')
 
     id = fields.Integer(dump_only=True)
+    provider_id = fields.String(required=True)
     user_id = fields.String(required=True)
     type = fields.String(required=True)
     document_name = fields.String(allow_none=True)

{cs_models-0.0.706.dist-info → cs_models-0.0.708.dist-info}/METADATA

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

{cs_models-0.0.706.dist-info → cs_models-0.0.708.dist-info}/RECORD

@@ -341,6 +341,9 @@ cs_models/resources/FederalCaseRef/__init__.py,sha256=0I8MBmiT-2pNtJYM8XF-xVjlOG
 cs_models/resources/FederalCaseRef/models.py,sha256=SiaWm1dmdTA407y8vVzf2gkU6IHbvuU6ZnmBjt_hBug,771
 cs_models/resources/FederalCaseRef/schemas.py,sha256=CjDlOyiRC_hT2ScdR9J_ln9TSVag_--ba9WEhGj3pXI,590
 cs_models/resources/FederalCaseRef/test_federal_case_ref.py,sha256=XfeyHzwn0gGAB1QfDoRM1CFgCGfggJCWvt5A_xqn4qM,4115
+cs_models/resources/Figure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/Figure/models.py,sha256=CSjZQFrRn5jpAsrX4rrf-GWN6t3aR8ouI9yk7QRun64,1011
+cs_models/resources/Figure/schemas.py,sha256=S4KcUNmHwcuWLONWhiy8KrPDOgaNSCHRmHoniq2fRao,646
 cs_models/resources/File/__init__.py,sha256=eODbmnbxhnyka2FlPZbF_KmHGVpSakqgHYw8sPqb_hs,5815
 cs_models/resources/File/models.py,sha256=PMG9wkcIQHofZmAPJHx5LUNNFkXfy922sHWXRtIH6Ec,897
 cs_models/resources/File/schemas.py,sha256=RyOtOd21lC00tN_SAuKiTSEbye4ukAD_Jq_3KGA2Xlk,1166
@@ -978,8 +981,8 @@ cs_models/resources/UserDocumentCondition/__init__.py,sha256=47DEQpj8HBSa-_TImW-
 cs_models/resources/UserDocumentCondition/models.py,sha256=FZzNgxc4-zZ5TiuJYfxNjPvi5mZCunTc9sVOXBGYpqc,1007
 cs_models/resources/UserDocumentCondition/schemas.py,sha256=BeeRrfrLW9N3k0IsOhoi3CXXos4rNbJ_ka4Qqbj-Ddw,550
 cs_models/resources/UserDocumentHierarchy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/UserDocumentHierarchy/models.py,sha256=vNo15EfKsfO136Fc0hqXB6sjazJ8ksqTuOUHTVCSEHE,1359
-cs_models/resources/UserDocumentHierarchy/schemas.py,sha256=4nAwk-qLhMKQqc5-netIIJeHDztBDORz9iTdyqRKe7o,712
+cs_models/resources/UserDocumentHierarchy/models.py,sha256=qz-vxxGo5hOrDBxJ8c9nLJmbObJbovnZprkuHzgMl-c,1425
+cs_models/resources/UserDocumentHierarchy/schemas.py,sha256=2AqeGjErCd_mZxAPClphJa4jI4ELmLo0lfgPg-G6H8k,759
 cs_models/resources/UserDocumentIntervention/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserDocumentIntervention/models.py,sha256=XDW1UIMnsQoyszHeel_aaCGnd_XfWxIOJF-JJBn9IoM,1019
 cs_models/resources/UserDocumentIntervention/schemas.py,sha256=XKgWDYGiU7NTPqNHkqxWgbdz0wMwt8__SlxrCPhEqrU,556
@@ -1171,7 +1174,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.706.dist-info/METADATA,sha256=Y7K7LbJnUge1ORlvUeTr-msB_1VsTitsr-cJejtY6W0,792
-cs_models-0.0.706.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.706.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.706.dist-info/RECORD,,
+cs_models-0.0.708.dist-info/METADATA,sha256=BFwhleVKYXcBAhmPqRwpmaBgzN1R2tKgfwUgRFKUWHY,792
+cs_models-0.0.708.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.708.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.708.dist-info/RECORD,,