cs-models 0.0.558__py3-none-any.whl → 0.0.559__py3-none-any.whl

cs_models/resources/LLMQueue/models.py

@@ -0,0 +1,38 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    Boolean,
+    String,
+    Text,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class LLMQueueModel(Base):
+    __tablename__ = 'llm_queue'
+
+    id = Column(Integer, primary_key=True)
+    source_type = Column(String(50), nullable=False)
+    source_id = Column(
+        Integer,
+        nullable=False,
+    )
+    llm_task_type = Column(String(50), nullable=False)
+    processed = Column(
+        Boolean,
+        nullable=True,
+    )
+    error = Column(Text, nullable=True)
+    submitted = Column(Boolean, nullable=True)
+    submitted_date = Column(DateTime, nullable=True)
+    historical = 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/LLMQueue/schemas.py

@@ -0,0 +1,20 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class LLMQueueResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    source_type = fields.String(required=True)
+    source_id = fields.Integer(required=True)
+    llm_task_type = fields.String(required=True)
+    processed = fields.Boolean(allow_none=True)
+    error = fields.String(allow_none=True)
+    submitted = fields.Boolean(allow_none=True)
+    submitted_date = fields.DateTime(allow_none=True)
+    historical = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

{cs_models-0.0.558.dist-info → cs_models-0.0.559.dist-info}/METADATA

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

{cs_models-0.0.558.dist-info → cs_models-0.0.559.dist-info}/RECORD

@@ -391,6 +391,9 @@ cs_models/resources/InvestorBucket/schemas.py,sha256=gwPUdJtWWi9-BliGkBuVu3QVAG0
 cs_models/resources/LLMCache/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/LLMCache/models.py,sha256=Std5BPOsMFvGArvzIbIs5Mscwg1vtsuaTlTJbDzIC34,776
 cs_models/resources/LLMCache/schemas.py,sha256=7RTxC4ZqNyd3c2r_NqbG-d6QoIaOMS6U4-Bcw7py4pE,483
+cs_models/resources/LLMQueue/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/LLMQueue/models.py,sha256=zlnidRfeLOYBrqLkNAeerdMb3b_Nh2_qWJ18W-VleUU,975
+cs_models/resources/LLMQueue/schemas.py,sha256=k7J-38xB4ENk9kMz7f3lUXPSNbkgBWOn64iYpGnQYok,650
 cs_models/resources/LatestPipelineProduct/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/LatestPipelineProduct/models.py,sha256=UZKXNTaDpAsr13miUx_lHkrwhbTYehpcB0TwLgAz--k,988
 cs_models/resources/LatestPipelineProduct/schemas.py,sha256=_yTbddbRov58yE8MdiZk4_VBK92EYQU9VXPUMw7bf18,1476
@@ -907,7 +910,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.558.dist-info/METADATA,sha256=UWL_ToyvJ3zkIjWIk3w7w6ozrySrlPkHaQKJqm3Lu_g,791
-cs_models-0.0.558.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.558.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.558.dist-info/RECORD,,
+cs_models-0.0.559.dist-info/METADATA,sha256=IRTirUNaK5jNaoZEvzD_aon8dJpIT8WEB6U7hNpsEXk,791
+cs_models-0.0.559.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.559.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.559.dist-info/RECORD,,