cs-models 0.0.679__py3-none-any.whl → 0.0.681__py3-none-any.whl

cs_models/resources/AssistantUserQuery/models.py

@@ -1,7 +1,7 @@
 import enum
 from datetime import datetime
 
-from sqlalchemy import Column, DateTime, Enum, ForeignKey, Integer, String
+from sqlalchemy import Column, DateTime, Enum, ForeignKey, Integer, String, Text
 from sqlalchemy.orm import relationship
 
 from ...database import Base
@@ -28,6 +28,10 @@ class AssistantUserQueryModel(Base):
         String,
         nullable=False,
     )
+    filter = Column(
+        Text,
+        nullable=True,
+    )
     type = Column(
         String,
         nullable=False,

cs_models/resources/AssistantUserQuery/schemas.py

@@ -1,15 +1,54 @@
+import json
 from marshmallow import Schema, fields
-
 from .models import AssistantUserQueryStatusEnum
 from ..AssistantCommand.schemas import (
     AssistantCommandResourceSchema,
 )
 
 
+class AssistantUserQueryFilterField(fields.Field):
+    """Field that stores result for the Assistant command."""
+
+    def _serialize(self, value, attr, obj, **kwargs):
+        """
+        In the DB, the `result` field is a text field. We persist
+        data by performing the following:
+
+        AssistantCommandModel(
+            ...
+            result=json.dumps({...}),
+        )
+
+        So here we need to perform the inverse operation (i.e `json.loads(..)`)
+        """
+        if value is None:
+            return None
+        return json.loads(value)
+
+    def _deserialize(self, value, attr, data, **kwargs):
+        """
+        In the DB, the `result` field is a text field. We persist
+        data by performing the following:
+
+
+
+        AssistantCommandModel(
+            ***AssistantCommandResourceSchema().load({
+                ...
+                "result": [{"some_key": 1}, {"some_key": 2}],
+            }),
+        )
+        """
+        if value is None:
+            return None
+        return json.dumps(value)
+
+
 class AssistantUserQueryResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     session_id = fields.Integer(required=True)
     value = fields.String(required=True)
+    filter = AssistantUserQueryFilterField(allow_none=True)
     type = fields.String(required=True)
     status = fields.Enum(AssistantUserQueryStatusEnum, required=True, by_value=True)
     commands = fields.Nested(

{cs_models-0.0.679.dist-info → cs_models-0.0.681.dist-info}/METADATA

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

{cs_models-0.0.679.dist-info → cs_models-0.0.681.dist-info}/RECORD

@@ -59,8 +59,8 @@ cs_models/resources/AssistantSession/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeu
 cs_models/resources/AssistantSession/models.py,sha256=_aWsZ71HFytf0ufekL_dGGECYKDPPzlRSPXAv9dR0rY,1417
 cs_models/resources/AssistantSession/schemas.py,sha256=v4orSJwiNDzmcjdUHoW9CZpkS7pxsUwEdQDUJpOqwg4,781
 cs_models/resources/AssistantUserQuery/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/AssistantUserQuery/models.py,sha256=fCM9N65laJanb_hVzFgTMzVLV09KuqSQKa27RuHhnMU,1753
-cs_models/resources/AssistantUserQuery/schemas.py,sha256=dGAHxkNxY7v8iD0EtKi_XJrIBF67ie-r7j6wNnAewP0,716
+cs_models/resources/AssistantUserQuery/models.py,sha256=Je_-aQ702dWxQSTH398dspF8_glRCHtSA6_LdWLW410,1823
+cs_models/resources/AssistantUserQuery/schemas.py,sha256=ju2-XM9xdXCNwcFy0YDzJ0dH0Nj2BxTRZPqnItFZMVo,1819
 cs_models/resources/B3CRSS/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/B3CRSS/models.py,sha256=rl538IgpQSzZu0FzpfxbDDCRWGkAGTpqcw9or5tMOg8,629
 cs_models/resources/B3CRSS/schemas.py,sha256=mArXYiVl15oEFaz9RLQ6WN68ueI0ec83NeLrNnP-sMM,346
@@ -1138,7 +1138,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.679.dist-info/METADATA,sha256=gqnZJ0KdGOBZRZ_oJ_jsttZlwoumIlSdtrIrZdrrh34,792
-cs_models-0.0.679.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.679.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.679.dist-info/RECORD,,
+cs_models-0.0.681.dist-info/METADATA,sha256=MMmOvmkbDD3LH9MTcyv3swU1GXvG8Uy7am_WjlumBEA,792
+cs_models-0.0.681.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.681.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.681.dist-info/RECORD,,