fix(sdk/python): add model_dump compatibility

Author: marcusolsson

assemblyai/streaming/v3/client.py

@@ -41,6 +41,12 @@ def _user_agent() -> str:
     )
 
 
+def _dump_model(model: BaseModel):
+    if hasattr(model, "model_dump"):
+        return model.model_dump(exclude_none=True)
+    return model.dict(exclude_none=True)
+
+
 class StreamingClient:
     def __init__(self, options: StreamingClientOptions):
         self._options = options
@@ -56,7 +62,7 @@ def __init__(self, options: StreamingClientOptions):
         self._stop_event = threading.Event()
 
     def connect(self, params: StreamingParameters) -> None:
-        params_dict = params.model_dump(exclude_none=True)
+        params_dict = _dump_model(params)
         params_encoded = urlencode(params_dict)
 
         uri = f"wss://{self._options.api_host}/v3/ws?{params_encoded}"
@@ -105,7 +111,7 @@ def stream(
             self._write_queue.put(chunk)
 
     def set_params(self, params: StreamingSessionParameters):
-        message = UpdateConfiguration(**params.model_dump())
+        message = UpdateConfiguration(**_dump_model(params))
         self._write_queue.put(message)
 
     def on(self, event: StreamingEvents, handler: Callable) -> None:
@@ -126,8 +132,7 @@ def _write_message(self) -> None:
                 if isinstance(data, bytes):
                     self._websocket.send(data)
                 elif isinstance(data, BaseModel):
-                    message = data.model_dump_json(exclude_none=True)
-                    self._websocket.send(message)
+                    self._websocket.send(json.dumps(data))
                 else:
                     raise ValueError(f"Attempted to send invalid message: {type(data)}")
             except websockets.exceptions.ConnectionClosed as exc: