cs-models 0.0.659__py3-none-any.whl → 0.0.661__py3-none-any.whl

cs_models/resources/AssistantCommand/models.py

@@ -17,6 +17,7 @@ class AssistantCommandTypeEnum(enum.Enum):
     SUMMARIZE = "SUMMARIZE"
     QUESTION_AND_ANSWER = "QUESTION_AND_ANSWER"
     TABLE_GENERATION = "TABLE_GENERATION"
+    CHART_GENERATION = "CHART_GENERATION"
 
 
 class AssistantCommandStatusEnum(enum.Enum):

cs_models/resources/AssistantCommandChart/models.py

@@ -0,0 +1,37 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Boolean,
+    Text,
+)
+
+from ...database import Base
+
+
+class AssistantCommandChartModel(Base):
+    __tablename__ = "assistant_command_charts"
+
+    id = Column(Integer, primary_key=True)
+    assistant_command_id = Column(
+        Integer,
+        ForeignKey('assistant_commands.id'),
+        nullable=False,
+    )
+    file_id = Column(
+        Integer,
+        ForeignKey('files.id'),
+        nullable=False,
+    )
+    chart_info = Column(Text, nullable=True)
+    is_deleted = 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/AssistantCommandChart/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class AssistantCommandChartResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    assistant_command_id = fields.Integer(required=True)
+    file_id = fields.Integer(required=True)
+    chart_info = fields.String(allow_none=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime()

{cs_models-0.0.659.dist-info → cs_models-0.0.661.dist-info}/METADATA

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

{cs_models-0.0.659.dist-info → cs_models-0.0.661.dist-info}/RECORD

@@ -27,11 +27,14 @@ cs_models/resources/Assignee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 cs_models/resources/Assignee/models.py,sha256=_f0_UwzF1gbOlU0t0l-9UIi0E4vaIS5VsHeOxc698SE,617
 cs_models/resources/Assignee/schemas.py,sha256=N4_upICR1Cd_B45LHpOly7FEwWhAB1fXUP4y60OiezE,269
 cs_models/resources/AssistantCommand/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/AssistantCommand/models.py,sha256=xP5gQSbdxZwXzbep9aGC4clinhQ92MTWW-4ngV8-xew,2014
+cs_models/resources/AssistantCommand/models.py,sha256=pcUVYIxI6EtbJphk1umNbdWQPt_Bul0kvUWNGqwO5Q4,2056
 cs_models/resources/AssistantCommand/schemas.py,sha256=7mS-WTHwPsu8Kd_0FLgJqYwrg0qVJUp7iioWqPT1ZBE,3695
 cs_models/resources/AssistantCommandArtifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AssistantCommandArtifact/models.py,sha256=SxSjZA_y2SGzXv7YW-19-rCTzRBcUiPf5JzMK1B-xRg,808
 cs_models/resources/AssistantCommandArtifact/schemas.py,sha256=fNOCH2oS0gUsGZtgpKeYPc4-xJhODw2HidHV3C12tGo,414
+cs_models/resources/AssistantCommandChart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/AssistantCommandChart/models.py,sha256=N4Mri-74Is82ZGOvPLwvpcNHH9cJrWwYwDZtfNUGnjM,892
+cs_models/resources/AssistantCommandChart/schemas.py,sha256=tD-ePZEKtZZOprpaxOZKhAfE-DVWmrTFTMWzwQ6tqfk,463
 cs_models/resources/AssistantCommandEdge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AssistantCommandEdge/models.py,sha256=3blYNPb6zH6ewfj-VK7z77bbtvO9Ly6os4NL2vh4wIg,1372
 cs_models/resources/AssistantCommandTable/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1087,7 +1090,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.659.dist-info/METADATA,sha256=MjgYiwWRgPUQbKD-tQNHlfZWd5TRhdGEN-iwwyTBQEo,792
-cs_models-0.0.659.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.659.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.659.dist-info/RECORD,,
+cs_models-0.0.661.dist-info/METADATA,sha256=IaenOEUGGXn83r7S7nrGr-8bFPTQxfXYAwo27-emk-8,792
+cs_models-0.0.661.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.661.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.661.dist-info/RECORD,,