implementation and unit tests

Author: jpvajda

deepgram/clients/agent/v1/websocket/options.py

@@ -273,26 +273,52 @@ class Agent(BaseResponse):
             or (isinstance(f, Think) and not f)
         ),
     )
-    speak: Speak = field(
-        default_factory=Speak,
-        metadata=dataclass_config(
-            exclude=lambda f: f is None
-            or (isinstance(f, dict) and not f)
-            or (isinstance(f, Speak) and not f)
-        ),
-    )
+    speak: Union[Speak, List[Speak]] = field(default_factory=Speak)
     greeting: Optional[str] = field(
         default=None, metadata=dataclass_config(exclude=lambda f: f is None)
     )
 
+    def __post_init__(self):
+        """Handle conversion of dict/list data to proper Speak objects"""
+        # Handle listen conversion (existing pattern)
+        if (
+            not isinstance(self.listen, Listen)
+            and self.listen is not None
+            and not (isinstance(self.listen, dict) and not self.listen)
+        ):
+            self.listen = Listen.from_dict(self.listen)
+
+        # Handle think conversion (existing pattern)
+        if (
+            not isinstance(self.think, Think)
+            and self.think is not None
+            and not (isinstance(self.think, dict) and not self.think)
+        ):
+            self.think = Think.from_dict(self.think)
+
+        # Handle speak conversion (new OneOf pattern)
+        if isinstance(self.speak, list):
+            # Convert list of dicts to list of Speak objects
+            self.speak = [
+                Speak.from_dict(item) if isinstance(item, dict) else item
+                for item in self.speak
+            ]
+        elif isinstance(self.speak, dict):
+            # Convert single dict to Speak object
+            self.speak = Speak.from_dict(self.speak)
+        # If it's already a Speak object or None, leave it as is
+
     def __getitem__(self, key):
         _dict = self.to_dict()
         if "listen" in _dict and isinstance(_dict["listen"], dict):
             _dict["listen"] = Listen.from_dict(_dict["listen"])
         if "think" in _dict and isinstance(_dict["think"], dict):
             _dict["think"] = Think.from_dict(_dict["think"])
-        if "speak" in _dict and isinstance(_dict["speak"], dict):
-            _dict["speak"] = Speak.from_dict(_dict["speak"])
+        if "speak" in _dict:
+            if isinstance(_dict["speak"], list):
+                _dict["speak"] = [Speak.from_dict(item) for item in _dict["speak"]]
+            elif isinstance(_dict["speak"], dict):
+                _dict["speak"] = Speak.from_dict(_dict["speak"])
         return _dict[key]
 
 
@@ -395,7 +421,19 @@ class UpdateSpeakOptions(BaseResponse):
     """
 
     type: str = str(AgentWebSocketEvents.UpdateSpeak)
-    speak: Speak = field(default_factory=Speak)
+    speak: Union[Speak, List[Speak]] = field(default_factory=Speak)
+
+    def __post_init__(self):
+        """Handle conversion of dict/list data to proper Speak objects"""
+        if isinstance(self.speak, list):
+            # Convert list of dicts to list of Speak objects
+            self.speak = [
+                Speak.from_dict(item) if isinstance(item, dict) else item
+                for item in self.speak
+            ]
+        elif isinstance(self.speak, dict):
+            # Convert single dict to Speak object
+            self.speak = Speak.from_dict(self.speak)
 
 
 # InjectAgentMessage