fix(python/sdk): update Realtime error codes and add check to avoid KeyError (#5854)

GitOrigin-RevId: b934ad7caebe865c546c45d944f109f524e630f2

Author: ploeber

assemblyai/transcriber.py

@@ -1313,7 +1313,11 @@ def _handle_error(self, error: websockets.exceptions.ConnectionClosed) -> None:
         - https://www.iana.org/assignments/websocket/websocket.xhtml#close-code-number
         - https://www.assemblyai.com/docs/Guides/real-time_streaming_transcription#closing-and-status-codes
         """
-        if error.code >= 4000 and error.code <= 4999:
+        if (
+            error.code >= 4000
+            and error.code <= 4999
+            and error.code in types.RealtimeErrorMapping
+        ):
             error_message = types.RealtimeErrorMapping[error.code]
         else:
             error_message = error.reason

assemblyai/types.py

@@ -2278,9 +2278,11 @@ class RealtimeError(AssemblyAIError):
     4031: "Session idle for too long",
     4032: "Audio duration is too short",
     4033: "Audio duration is too long",
+    4034: "Audio too small to transcode",
     4100: "Endpoint received invalid JSON",
     4101: "Endpoint received a message with an invalid schema",
     4102: "This account has exceeded the number of allowed streams",
     4103: "The session has been reconnected. This websocket is no longer valid.",
+    4104: "Could not parse word boost parameter",
     1013: "Temporary server condition forced blocking client's request",
 }