Merge pull request #297 from dvonthenen/fix-enum-serializable

Object of type Sentiment is not JSON serializable

Author: davidvonthenen

deepgram/clients/analyze/enums.py

@@ -9,8 +9,8 @@
 """
 
 
-class Sentiment(Enum):
-    UNKNOWN = ""
-    NEGATIVE = "negative"
-    NEUTRAL = "neutral"
-    POSITIVE = "positive"
+class Sentiment(str, Enum):
+    UNKNOWN: str = ""
+    NEGATIVE: str = "negative"
+    NEUTRAL: str = "neutral"
+    POSITIVE: str = "positive"

deepgram/clients/live/enums.py

@@ -9,12 +9,12 @@
 """
 
 
-class LiveTranscriptionEvents(Enum):
-    Open = "Open"
-    Close = "Close"
-    Transcript = "Results"
-    Metadata = "Metadata"
-    UtteranceEnd = "UtteranceEnd"
-    SpeechStarted = "SpeechStarted"
-    Error = "Error"
-    Warning = "Warning"
+class LiveTranscriptionEvents(str, Enum):
+    Open: str = "Open"
+    Close: str = "Close"
+    Transcript: str = "Results"
+    Metadata: str = "Metadata"
+    UtteranceEnd: str = "UtteranceEnd"
+    SpeechStarted: str = "SpeechStarted"
+    Error: str = "Error"
+    Warning: str = "Warning"

deepgram/clients/prerecorded/enums.py

@@ -9,8 +9,8 @@
 """
 
 
-class Sentiment(Enum):
-    UNKNOWN = ""
-    NEGATIVE = "negative"
-    NEUTRAL = "neutral"
-    POSITIVE = "positive"
+class Sentiment(str, Enum):
+    UNKNOWN: str = ""
+    NEGATIVE: str = "negative"
+    NEUTRAL: str = "neutral"
+    POSITIVE: str = "positive"