cs-models 0.0.602__py3-none-any.whl → 0.0.604__py3-none-any.whl

cs_models/resources/AssistantCopilot/models.py

@@ -0,0 +1,30 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    Text,
+    String,
+)
+
+from ...database import Base
+
+
+class AssistantCopilotModel(Base):
+    __tablename__ = "assistant_copilot"
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    query = Column(String(256), nullable=False)
+    timestamp = Column(DateTime, nullable=False)
+    llm_result = Column(Text, nullable=True)
+    status_code = Column(String(20), nullable=True)
+    status = Column(String(191), 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/AssistantCopilot/schemas.py

@@ -0,0 +1,12 @@
+from marshmallow import Schema, fields
+
+
+class AssistantCopilotResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True)
+    query = fields.String(required=True)
+    timestamp = fields.DateTime(required=True)
+    llm_result = fields.String(allow_none=True)
+    status = fields.String(allow_none=True)
+    status_code = fields.String(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/AssistantCopilotTasks/models.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+)
+
+from ...database import Base
+
+
+class AssistantCopilotTaskModel(Base):
+    __tablename__ = "assistant_copilot_tasks"
+
+    id = Column(Integer, primary_key=True)
+    assistant_copilot_id = Column(
+        Integer,
+        ForeignKey('assistant_copilot.id'),
+        nullable=False,
+    )
+    assistant_search_copilot_id = Column(
+        Integer,
+        ForeignKey('assistant_search_copilot.id'),
+        nullable=True,
+    )
+    assistant_session_id = Column(
+        Integer,
+        ForeignKey('assistant_sessions.id'),
+        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/AssistantCopilotTasks/schemas.py

@@ -0,0 +1,9 @@
+from marshmallow import Schema, fields
+
+
+class AssistantCopilotTaskResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    assistant_copilot_id = fields.Integer(required=True)
+    assistant_search_copilot_id = fields.Integer(allow_none=True)
+    assistant_session_id = fields.Integer(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/AssistantSearchCopilot/models.py

@@ -22,6 +22,7 @@ class AssistantSearchCopilotModel(Base):
         nullable=True,
     )
     status = Column(String(191), nullable=True)
+    status_code = Column(String(20), nullable=True)
     search_payloads = Column(Text, nullable=True)
     updated_at = Column(
         DateTime,

cs_models/resources/AssistantSearchCopilot/schemas.py

@@ -5,5 +5,6 @@ class AssistantSearchCopilotResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     assistant_user_query_id = fields.Integer(allow_none=True)
     status = fields.String(allow_none=True)
+    status_code = fields.String(allow_none=True)
     search_payloads = fields.String(allow_none=True)
     updated_at = fields.DateTime(dump_only=True)

{cs_models-0.0.602.dist-info → cs_models-0.0.604.dist-info}/METADATA

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

{cs_models-0.0.602.dist-info → cs_models-0.0.604.dist-info}/RECORD

@@ -28,12 +28,18 @@ cs_models/resources/AssistantCommandEdge/models.py,sha256=3blYNPb6zH6ewfj-VK7z77
 cs_models/resources/AssistantCommandTable/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AssistantCommandTable/models.py,sha256=spy6CuB3JTYvVVlns526l2hIRlFSi6PcdKHbwb3U9DU,1109
 cs_models/resources/AssistantCommandTable/schemas.py,sha256=VvIQKKXEpuI6VR0nHvKkK11LiM30G5hJL2CowUXc1Lw,665
+cs_models/resources/AssistantCopilot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/AssistantCopilot/models.py,sha256=qPtY7lap-nb2yOcjtQS3lLzhIZU18JeVruOdbAQjIGI,834
+cs_models/resources/AssistantCopilot/schemas.py,sha256=5Knq3Zwg9dpPYuR5JBTJOP9JTPnvdrIH0mG-qZ0XEVE,448
+cs_models/resources/AssistantCopilotTasks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/AssistantCopilotTasks/models.py,sha256=JiHoyqJlMqNLVSd__3aoNPoG1d8yQ1abRS1U8UQaDcg,937
+cs_models/resources/AssistantCopilotTasks/schemas.py,sha256=5T64fscZLthM1bxl8EPfr9Pkzw8wgzuX4YoCxV0FIwU,362
 cs_models/resources/AssistantScratchPad/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AssistantScratchPad/models.py,sha256=A2XS32dkIEdx2ToL_vA9i4b9cTLWKVRXhP35RbROW9Q,1025
 cs_models/resources/AssistantScratchPad/schemas.py,sha256=yAuRxZfnmbYMvtGxE56ub8OEPmk84NAJAJJZbuFQE94,468
 cs_models/resources/AssistantSearchCopilot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/AssistantSearchCopilot/models.py,sha256=rmv5rpzvI2adtkpJKVpy9xuRz9-1SE0FI9lAWBgnCq8,802
-cs_models/resources/AssistantSearchCopilot/schemas.py,sha256=QtroKzlks9TOSULptaja-CDH5zv_zO9MF3Irns4d5ho,341
+cs_models/resources/AssistantSearchCopilot/models.py,sha256=goeBVLehkfgMotWyIHZUbXEkaAfj5STxmZi_C3ezjsg,854
+cs_models/resources/AssistantSearchCopilot/schemas.py,sha256=eA8fQxrW7q7mZUKAIV-a0uRS6eLI-L2MxHmx7DoQGJY,390
 cs_models/resources/AssistantSearchCopilotMarkdown/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AssistantSearchCopilotMarkdown/models.py,sha256=1kP7JU2Bxlcl_dG2aSg88SertHL4mcNrlCnoECHq3vU,760
 cs_models/resources/AssistantSearchCopilotMarkdown/schemas.py,sha256=74k6fc1Ev9CBqLv_8YdxVIq_tppLP_1CKUYtH-nkcv0,291
@@ -985,7 +991,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.602.dist-info/METADATA,sha256=69Que16CSfs6hAKtrHL3skeqpv28p2DUNk2dhhBPUIM,791
-cs_models-0.0.602.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.602.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.602.dist-info/RECORD,,
+cs_models-0.0.604.dist-info/METADATA,sha256=wPLObkSfe4XyVJFAHd1UDve1kF0QYqfKcRHTfLIF6FY,791
+cs_models-0.0.604.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.604.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.604.dist-info/RECORD,,