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

cs_models/resources/EventLLM/models.py

@@ -16,21 +16,11 @@ class EventLLMModel(Base):
     __tablename__ = "events_llm"
 
     id = Column(Integer, primary_key=True)
-    date = Column(DateTime, nullable=False)
-    end_date = Column(DateTime, nullable=False)
-    event_type = Column(String(128), nullable=False)
-    description = Column(Text, nullable=False)
-    products = Column(Text, nullable=True)
-    company_sec_id = Column(
-        Integer,
-        ForeignKey('companies_sec.id'),
-        nullable=True,
-    )
-    company_ous_id = Column(
-        Integer,
-        ForeignKey('companies_ous.id'),
-        nullable=True,
-    )
+    start_date = Column(DateTime, nullable=False)
+    end_date = Column(DateTime, nullable=True)
+    event_type = Column(String(128), nullable=True)
+    description = Column(Text, nullable=True)
+    tags = Column(Text, nullable=True)
     news_id = Column(
         Integer,
         ForeignKey('newswires.id'),
@@ -41,7 +31,7 @@ class EventLLMModel(Base):
         ForeignKey('companies_sec_filings.id'),
         nullable=True,
     )
-    llm_response = Column(Text, nullable=True)
+    reference_text = Column(Text, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/EventLLM/schemas.py

@@ -11,16 +11,14 @@ class EventLLMResourceSchema(Schema):
     not_blank = validate.Length(min=1, error="Field cannot be blank")
 
     id = fields.Integer(dump_only=True)
-    date = fields.DateTime(required=True)
-    end_date = fields.DateTime(required=True)
-    event_type = fields.String(required=True)
-    description = fields.String(required=True)
-    products = fields.String(allow_none=True)
-    company_sec_id = fields.Integer(allow_none=True)
-    company_ous_id = fields.Integer(allow_none=True)
+    start_date = fields.DateTime(required=True)
+    end_date = fields.DateTime(allow_none=True)
+    event_type = fields.String(allow_none=True)
+    description = fields.String(allow_none=True)
+    tags = fields.String(allow_none=True)
     news_id = fields.Integer(allow_none=True)
     sec_filing_id = fields.Integer(allow_none=True)
-    llm_response = fields.String(allow_none=True)
+    reference_text = fields.String(allow_none=True)
     updated_at = fields.DateTime(dump_only=True)
 
     @pre_load

cs_models/resources/EventsQueue/models.py

@@ -0,0 +1,33 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    Boolean,
+    String,
+    Text,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class EventsQueueModel(Base):
+    __tablename__ = 'events_queue'
+
+    id = Column(Integer, primary_key=True)
+    artifact_id = Column(String(50), nullable=False)
+    processed = Column(
+        Boolean,
+        nullable=True,
+    )
+    attempts = Column(Integer, nullable=True)
+    error = Column(Text, nullable=True)
+    submitted = Column(Boolean, nullable=True)
+    submitted_date = Column(DateTime, 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/EventsQueue/schemas.py

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

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

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

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

@@ -290,8 +290,11 @@ cs_models/resources/Event/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 cs_models/resources/Event/models.py,sha256=wkRLT1tkFjXId8Ga1BiThwflXK5lRaCFqEf_guIwWM0,1227
 cs_models/resources/Event/schemas.py,sha256=63C0x7i-JUu5Q0aAZeNBd1O0_XZ7cahM3KTpkLCbCM8,1690
 cs_models/resources/EventLLM/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/EventLLM/models.py,sha256=ZsGh4a3Q606-KVFsgQD1Tdy6BuiNJ9Wx9MFwIL7N8qY,1281
-cs_models/resources/EventLLM/schemas.py,sha256=9I4kbekU5GEG2YFPfM2D5z85UqT7bAkyVjSoPANo5y4,1415
+cs_models/resources/EventLLM/models.py,sha256=YzZnYW4cJBdDeppLPAI5LihOPD83acOaNlSeGR0LiP8,1052
+cs_models/resources/EventLLM/schemas.py,sha256=sE3b1aiCeAuhRI9uzCk_Uozw2NrNtRFwtdJxvhKCUAc,1319
+cs_models/resources/EventsQueue/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/EventsQueue/models.py,sha256=frFQpfjnbEejlPvUqtOyBb-tN48UNG77uVUOdj-u63A,853
+cs_models/resources/EventsQueue/schemas.py,sha256=X6EGAog4ZTBraLJ1w0FOU88KWdmRiu1rl4XjPAh-nqU,556
 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
@@ -1084,7 +1087,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.657.dist-info/METADATA,sha256=_kW0bbGPv0xwnKC5dq92J1z8BAPPpEzcgKsB7jL6yRI,792
-cs_models-0.0.657.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.657.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.657.dist-info/RECORD,,
+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,,