Set Field in Metadata as Optional

Author: davidvonthenen

deepgram/clients/live/v1/response.py

@@ -118,9 +118,7 @@ def __str__(self) -> str:
 @dataclass
 class Metadata:
     request_id: str = ""
-    model_info: ModelInfo = field(
-        default=None, metadata=config(exclude=lambda f: f is None)
-    )
+    model_info: ModelInfo = None
     model_uuid: str = ""
     extra: Optional[Dict[str, str]] = field(
         default=None, metadata=config(exclude=lambda f: f is None)
@@ -213,8 +211,12 @@ class MetadataResponse:
     created: str = ""
     duration: float = 0
     channels: int = 0
-    models: List[str] = None
-    model_info: Dict[str, ModelInfo] = None
+    models: Optional[List[str]] = field(
+        default=None, metadata=config(exclude=lambda f: f is None)
+    )
+    model_info: Optional[Dict[str, ModelInfo]] = field(
+        default=None, metadata=config(exclude=lambda f: f is None)
+    )
     extra: Optional[Dict] = field(
         default=None, metadata=config(exclude=lambda f: f is None)
     )

deepgram/clients/manage/v1/response.py

@@ -522,7 +522,7 @@ class UsageFieldsResponse:
     models: List[UsageModel] = None
     processing_methods: List[str] = None
     features: List[str] = None
-    languages: List[str] = field(
+    languages: Optional[List[str]] = field(
         default=None, metadata=config(exclude=lambda f: f is None)
     )
 

tests/edge_cases/reconnect_same_object/main.py

@@ -0,0 +1,34 @@
+import time
+import logging, verboselogs
+
+from deepgram import DeepgramClient, DeepgramClientOptions, LiveOptions
+
+
+def main():
+    # config: DeepgramClientOptions = DeepgramClientOptions(verbose=logging.DEBUG)
+    config: DeepgramClientOptions = DeepgramClientOptions()
+    deepgram: DeepgramClient = DeepgramClient("", config)
+    options: LiveOptions = LiveOptions()
+
+    dg_connection = deepgram.listen.live.v("1")
+
+    for x in range(0, 10):
+        if x > 0:
+            # wait for the connection to close
+            time.sleep(5)
+
+        if x == 0:
+            print(f"Starting connection #{x}...")
+        else:
+            print(f"Restarting connection #{x}...")
+
+        dg_connection.start(options)
+        time.sleep(15)
+        print("Calling stop...")
+        dg_connection.finish()
+
+    print("Finished")
+
+
+if __name__ == "__main__":
+    main()