makes ActionConfig a TypedModel

playground/streaming/agent/chat.py

@@ -30,8 +30,7 @@
 from vocode.streaming.utils import create_conversation_id
 
 
-class ShoutActionConfig(ActionConfig):
-    type = "shout"
+class ShoutActionConfig(ActionConfig, type="shout"):
     num_exclamation_marks: int
 
 

vocode/streaming/action/nylas_send_email.py

@@ -10,8 +10,8 @@
 )
 
 
-class NylasSendEmailActionConfig(ActionConfig):
-    type = ActionType.NYLAS_SEND_EMAIL
+class NylasSendEmailActionConfig(ActionConfig, type=ActionType.NYLAS_SEND_EMAIL):
+    pass
 
 
 class NylasSendEmailParameters(BaseModel):

vocode/streaming/models/actions.py

@@ -1,17 +1,18 @@
 from enum import Enum
 from typing import Generic, TypeVar
 from pydantic import BaseModel
-
-
-class ActionConfig(BaseModel):
-    type: str
+from vocode.streaming.models.model import TypedModel
 
 
 class ActionType(str, Enum):
     BASE = "action_base"
     NYLAS_SEND_EMAIL = "action_nylas_send_email"
 
 
+class ActionConfig(TypedModel, type=ActionType.BASE):
+    pass
+
+
 ParametersType = TypeVar("ParametersType", bound=BaseModel)
 
 

vocode/streaming/models/model.py

@@ -8,6 +8,11 @@ def __init__(self, **data):
             if isinstance(value, dict):
                 if "type" in value:
                     data[key] = TypedModel.parse_obj(value)
+            if isinstance(value, list):
+                for i, v in enumerate(value):
+                    if isinstance(v, dict):
+                        if "type" in v:
+                            value[i] = TypedModel.parse_obj(v)
         super().__init__(**data)