cs-models 0.0.557__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/resources/LicensingCollabOutbox/models.py

@@ -27,6 +27,7 @@ class LicensingCollabOutboxModel(Base):
     sentences = Column(Text, nullable=True)
     reviewed = Column(Boolean, nullable=True)
     historical = Column(Boolean, nullable=True)
+    llm_output = Column(Text, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/LicensingCollabOutbox/schemas.py

@@ -17,6 +17,7 @@ class LicensingCollabOutboxResourceSchema(Schema):
     milestones = fields.String(allow_none=True)
     sentences = fields.String(allow_none=True)
     reviewed = fields.String(allow_none=True)
+    llm_output = fields.String(allow_none=True)
     historical = fields.Boolean(allow_none=True)
     updated_at = fields.DateTime()
 

cs_models/resources/LicensingCollabSentences/models.py

@@ -6,6 +6,8 @@ from sqlalchemy import (
     DateTime,
     ForeignKey,
     Text,
+    String,
+    Boolean,
 )
 
 from ...database import Base
@@ -21,6 +23,8 @@ class LicensingCollabSentenceModel(Base):
         nullable=False,
     )
     text = Column(Text, nullable=False, index=True)
+    type = Column(String(50), nullable=True)
+    llm_output = Column(Boolean, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/LicensingCollabSentences/schemas.py

@@ -11,4 +11,6 @@ class LicensingCollabSentenceResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     licensing_collab_id = fields.Integer(required=True)
     text = fields.String(required=True, validate=not_blank)
+    type = fields.String(allow_none=True)
+    llm_output = fields.Boolean(allow_none=True)
     updated_at = fields.DateTime()

cs_models/resources/MergerOutbox/models.py

@@ -24,6 +24,7 @@ class MergerOutboxModel(Base):
     counsels = Column(Text, nullable=True)
     sentences = Column(Text, nullable=True)
     reviewed = Column(Boolean, nullable=True)
+    llm_output = Column(Text, nullable=True)
     historical = Column(Boolean, nullable=True)
     updated_at = Column(
         DateTime,

cs_models/resources/MergerOutbox/schemas.py

@@ -17,6 +17,7 @@ class MergerOutboxResourceSchema(Schema):
     counsels = fields.String(allow_none=True)
     sentences = fields.String(allow_none=True)
     reviewed = fields.Boolean(allow_none=True)
+    llm_output = fields.String(allow_none=True)
     historical = fields.Boolean(allow_none=True)
     updated_at = fields.DateTime()
 

cs_models/resources/MergerSentences/models.py

@@ -6,6 +6,8 @@ from sqlalchemy import (
     DateTime,
     ForeignKey,
     Text,
+    String,
+    Boolean,
 )
 
 from ...database import Base
@@ -21,6 +23,8 @@ class MergerSentenceModel(Base):
         nullable=False,
     )
     text = Column(Text, nullable=False, index=True)
+    type = Column(String(50), nullable=True)
+    llm_output = Column(Boolean, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/MergerSentences/schemas.py

@@ -11,4 +11,6 @@ class MergerSentenceResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     merger_id = fields.Integer(required=True)
     text = fields.String(required=True, validate=not_blank)
+    type = fields.String(allow_none=True)
+    llm_output = fields.Boolean(allow_none=True)
     updated_at = fields.DateTime()

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

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

{cs_models-0.0.557.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
@@ -407,11 +410,11 @@ cs_models/resources/LicensingCollabIntervention/__init__.py,sha256=47DEQpj8HBSa-
 cs_models/resources/LicensingCollabIntervention/models.py,sha256=OY3vPH4yILb7LqihPJgLeYy4UFw3EvJKAj7122Wn-zs,983
 cs_models/resources/LicensingCollabIntervention/schemas.py,sha256=smLz2PTr1io9HZNXEFi5xUvfxPF0kQ3fjPxPE_x9rcU,514
 cs_models/resources/LicensingCollabOutbox/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/LicensingCollabOutbox/models.py,sha256=XoMcZRFxO7wo5HNeEaCwm5b2iG1v3NasYUXXrvCPxvg,944
-cs_models/resources/LicensingCollabOutbox/schemas.py,sha256=-_lK1fQkBOyutsuHCEoeQD4_ZYsnyrXLrQnxwMuosmQ,917
+cs_models/resources/LicensingCollabOutbox/models.py,sha256=eWqwjohRqVMmji6slV4yHIgKqHAOobNT8od4VmojCjc,989
+cs_models/resources/LicensingCollabOutbox/schemas.py,sha256=_4PMGxusxfonGUEHut-K2xK3ojEEZmFelhq-fEHwPjA,965
 cs_models/resources/LicensingCollabSentences/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/LicensingCollabSentences/models.py,sha256=kyEPzhSZOTsxxHe-oo3kY-kbCQQ-TQkLjhAuWZe578I,738
-cs_models/resources/LicensingCollabSentences/schemas.py,sha256=WbhqJTlPOqQaLpvOoURBzNzZSt4xXTKPBVVE30xN5PY,383
+cs_models/resources/LicensingCollabSentences/models.py,sha256=mcyLYLnJxTbMIt5i6EFLQpO_0JBofArE2_yRMrIvmXg,856
+cs_models/resources/LicensingCollabSentences/schemas.py,sha256=3efKLBydShsZiZMg-5a8BJSSyjr68mEOXFvoie4FwzM,474
 cs_models/resources/LicensingCollabTarget/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/LicensingCollabTarget/models.py,sha256=AofVwKxsGt6kf0YGQqY3r3I1GVrbSh5lHV1QUUfiB7c,959
 cs_models/resources/LicensingCollabTarget/schemas.py,sha256=jijLVNnVxOMWBMYfij-8-ZLz8m8A51cOn_U37Xqkg2Y,502
@@ -438,14 +441,14 @@ cs_models/resources/MergerAmount/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 cs_models/resources/MergerAmount/models.py,sha256=fwHnW6Z_vR09UBVra0aZG9JVxmr5FPMDth7GUXMaCsg,811
 cs_models/resources/MergerAmount/schemas.py,sha256=Z7WwpVIZZbday1cWZQ3C0kJOJ-gvuW7HWD1_4x9uB0c,432
 cs_models/resources/MergerOutbox/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/MergerOutbox/models.py,sha256=limwmXZZebF0mVM3XM56LDdcs2tHvUheHJy_JTtSb0g,991
-cs_models/resources/MergerOutbox/schemas.py,sha256=tAamWJF3aLpV_3NHAAasqsU9opsZSKCRA7K1LT33yCM,682
+cs_models/resources/MergerOutbox/models.py,sha256=X5M3rgeygH0CBlpxrR8-VZKvWZEOLThs2E_ZyfroGFo,1036
+cs_models/resources/MergerOutbox/schemas.py,sha256=Xr2GyTB5NXGwJ559nl8CuVl2hBhV_Xgcbf5YDqJu1O4,730
 cs_models/resources/MergerOutboxCompany/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/MergerOutboxCompany/models.py,sha256=seR4a8OL9wNXmuOijOEeZdsqurQRWablFkpH3RybwPs,906
 cs_models/resources/MergerOutboxCompany/schemas.py,sha256=PpXAGPuk_tHR9ydmFEPr4euZdX3ZIzE4EUQGZRfTTZ8,1103
 cs_models/resources/MergerSentences/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/MergerSentences/models.py,sha256=s-LJhgm8PeN_OeAE0McHt3fnoFfmq2lCRyyS7OZkX5o,700
-cs_models/resources/MergerSentences/schemas.py,sha256=qX9dteAmspe2UHZdxLNCYTma05LK0sSvZTx2vOMUeWU,364
+cs_models/resources/MergerSentences/models.py,sha256=I3q9pP9Nakzs9nqqgqDt-6HMql-pRHeTZ2JpK5pfvvA,818
+cs_models/resources/MergerSentences/schemas.py,sha256=WmTpRQY42wTwG2Flk-LFJ7uukqYEcYyCj6hv-49Jmwk,455
 cs_models/resources/Modality/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Modality/models.py,sha256=aYM_svz7MwYQizydXMvzn_fPjy8lTFi5ExbgfJK4z5s,447
 cs_models/resources/Modality/schemas.py,sha256=WdDFJ6C26XGN1eu9LdkbGBzlEDUSlvuyo0sPXdfOg_Q,296
@@ -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.557.dist-info/METADATA,sha256=9qouyGVlttrNsigbbIGu8eDV0tTu2RKlmSIdYUSP_bI,791
-cs_models-0.0.557.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.557.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.557.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,,