chore(tests): update snapshots

Author: lukeocodes

deepgram/client.py

@@ -446,28 +446,30 @@ def __init__(
 
         # Handle credential extraction from config first
         if api_key == "" and access_token == "" and config is not None:
-            self._logger.info(
-                "Attempting to set credentials from config object")
+            self._logger.info("Attempting to set credentials from config object")
             api_key = config.api_key
             access_token = config.access_token
 
         # Fallback to environment variables if no explicit credentials provided
         # Prioritize API key for backward compatibility
         if api_key == "" and access_token == "":
             self._logger.info(
-                "Attempting to get credentials from environment variables")
+                "Attempting to get credentials from environment variables"
+            )
             api_key = os.getenv("DEEPGRAM_API_KEY", "")
             if api_key == "":
                 access_token = os.getenv("DEEPGRAM_ACCESS_TOKEN", "")
 
         # Log warnings for missing credentials
         if api_key == "" and access_token == "":
             self._logger.warning(
-                "WARNING: Neither API key nor access token is provided")
+                "WARNING: Neither API key nor access token is provided"
+            )
 
         if config is None:  # Use default configuration
             self._config = DeepgramClientOptions(
-                api_key=api_key, access_token=access_token)
+                api_key=api_key, access_token=access_token
+            )
         else:
             # Update config with credentials only if we have valid credentials
             # This ensures empty strings don't overwrite existing config credentials

deepgram/clients/agent/client.py

@@ -95,4 +95,4 @@
 Input = LatestInput
 Output = LatestOutput
 Audio = LatestAudio
-Endpoint = LatestEndpoint
+Endpoint = LatestEndpoint

deepgram/clients/auth/__init__.py

@@ -6,4 +6,4 @@
 from .client import AsyncAuthRESTClient
 from .client import (
     GrantTokenResponse,
-)
+)

deepgram/clients/auth/v1/__init__.py

@@ -5,4 +5,4 @@
 from .async_client import AsyncAuthRESTClient
 from .response import (
     GrantTokenResponse,
-)
+)

deepgram/clients/auth/v1/async_client.py

@@ -42,7 +42,9 @@ async def grant_token(self):
 
         url = f"{self._config.url}/{self._endpoint}"
         self._logger.info("url: %s", url)
-        result = await self.post(url, headers={"Authorization": f"Token {self._config.api_key}"})
+        result = await self.post(
+            url, headers={"Authorization": f"Token {self._config.api_key}"}
+        )
         self._logger.info("json: %s", result)
         res = GrantTokenResponse.from_json(result)
         self._logger.verbose("result: %s", res)

deepgram/clients/auth/v1/client.py

@@ -42,7 +42,9 @@ def grant_token(self):
 
         url = f"{self._config.url}/{self._endpoint}"
         self._logger.info("url: %s", url)
-        result = self.post(url, headers={"Authorization": f"Token {self._config.api_key}"})
+        result = self.post(
+            url, headers={"Authorization": f"Token {self._config.api_key}"}
+        )
         self._logger.info("json: %s", result)
         res = GrantTokenResponse.from_json(result)
         self._logger.verbose("result: %s", res)

deepgram/clients/auth/v1/response.py

@@ -9,16 +9,18 @@
     BaseResponse,
 )
 
+
 @dataclass
 class GrantTokenResponse(BaseResponse):
     """
     The response object for the authentication grant token endpoint.
     """
+
     access_token: str = field(
-        metadata=dataclass_config(field_name='access_token'),
+        metadata=dataclass_config(field_name="access_token"),
         default="",
     )
     expires_in: int = field(
-        metadata=dataclass_config(field_name='expires_in'),
+        metadata=dataclass_config(field_name="expires_in"),
         default=30,
-    )
+    )

deepgram/options.py

@@ -30,6 +30,7 @@ class DeepgramClientOptions:  # pylint: disable=too-many-instance-attributes
     """
 
     _logger: verboselogs.VerboseLogger
+    headers: Dict[str, str]
     _inspect_listen: bool = False
     _inspect_speak: bool = False
 
@@ -82,7 +83,7 @@ def set_apikey(self, api_key: str):
         """
         self.api_key = api_key
         self.access_token = ""  # Clear access token when setting API key
-        self._update_headers(headers=getattr(self, '_custom_headers', {}))
+        self._update_headers(headers=getattr(self, "_custom_headers", {}))
 
     def set_access_token(self, access_token: str):
         """
@@ -93,7 +94,7 @@ def set_access_token(self, access_token: str):
         """
         self.access_token = access_token
         self.api_key = ""  # Clear API key when setting access token
-        self._update_headers(headers=getattr(self, '_custom_headers', {}))
+        self._update_headers(headers=getattr(self, "_custom_headers", {}))
 
     def _get_url(self, url) -> str:
         if not re.match(r"^https?://", url, re.IGNORECASE):
@@ -102,12 +103,15 @@ def _get_url(self, url) -> str:
 
     def _update_headers(self, headers: Optional[Dict] = None):
         # Initialize headers if not already set, otherwise preserve existing custom headers
-        if not hasattr(self, 'headers') or self.headers is None:
+        if not hasattr(self, "headers") or self.headers is None:
             self.headers = {}
         else:
             # Preserve existing custom headers but allow auth headers to be updated
-            existing_custom_headers = {k: v for k, v in self.headers.items()
-                                       if k not in ['Accept', 'Authorization', 'User-Agent']}
+            existing_custom_headers = {
+                k: v
+                for k, v in self.headers.items()
+                if k not in ["Accept", "Authorization", "User-Agent"]
+            }
             self.headers = existing_custom_headers
 
         self.headers["Accept"] = "application/json"
@@ -119,9 +123,9 @@ def _update_headers(self, headers: Optional[Dict] = None):
         elif self.access_token:
             self.headers["Authorization"] = f"Bearer {self.access_token}"
 
-        self.headers[
-            "User-Agent"
-        ] = f"@deepgram/sdk/{__version__} python/{sys.version_info[1]}.{sys.version_info[2]}"
+        self.headers["User-Agent"] = (
+            f"@deepgram/sdk/{__version__} python/{sys.version_info[1]}.{sys.version_info[2]}"
+        )
         # Overwrite / add any headers that were passed in
         if headers:
             self.headers.update(headers)
@@ -138,8 +142,7 @@ def is_auto_flush_reply_enabled(self) -> bool:
         """
         is_auto_flush_reply_enabled: Returns True if the client is configured to auto-flush for listen.
         """
-        auto_flush_reply_delta = float(
-            self.options.get("auto_flush_reply_delta", 0))
+        auto_flush_reply_delta = float(self.options.get("auto_flush_reply_delta", 0))
         return (
             isinstance(auto_flush_reply_delta, numbers.Number)
             and auto_flush_reply_delta > 0
@@ -149,8 +152,7 @@ def is_auto_flush_speak_enabled(self) -> bool:
         """
         is_auto_flush_speak_enabled: Returns True if the client is configured to auto-flush for speak.
         """
-        auto_flush_speak_delta = float(
-            self.options.get("auto_flush_speak_delta", 0))
+        auto_flush_speak_delta = float(self.options.get("auto_flush_speak_delta", 0))
         return (
             isinstance(auto_flush_speak_delta, numbers.Number)
             and auto_flush_speak_delta > 0
@@ -206,10 +208,10 @@ def __init__(
 
         # Require at least one form of authentication
         if api_key == "" and access_token == "":
-            self._logger.critical(
-                "Neither Deepgram API KEY nor ACCESS TOKEN is set")
+            self._logger.critical("Neither Deepgram API KEY nor ACCESS TOKEN is set")
             raise DeepgramApiKeyError(
-                "Neither Deepgram API KEY nor ACCESS TOKEN is set")
+                "Neither Deepgram API KEY nor ACCESS TOKEN is set"
+            )
 
         if url == "":
             url = os.getenv("DEEPGRAM_HOST", "api.deepgram.com")
@@ -258,8 +260,7 @@ def __init__(
             for x in range(0, 20):
                 header = os.getenv(f"DEEPGRAM_HEADER_{x}", None)
                 if header is not None:
-                    headers[header] = os.getenv(
-                        f"DEEPGRAM_HEADER_VALUE_{x}", None)
+                    headers[header] = os.getenv(f"DEEPGRAM_HEADER_VALUE_{x}", None)
                     self._logger.debug(
                         "Deepgram header %s is set with value %s",
                         header,
@@ -276,8 +277,7 @@ def __init__(
             for x in range(0, 20):
                 param = os.getenv(f"DEEPGRAM_PARAM_{x}", None)
                 if param is not None:
-                    options[param] = os.getenv(
-                        f"DEEPGRAM_PARAM_VALUE_{x}", None)
+                    options[param] = os.getenv(f"DEEPGRAM_PARAM_VALUE_{x}", None)
                     self._logger.debug(
                         "Deepgram option %s is set with value %s", param, options[param]
                     )
@@ -288,5 +288,10 @@ def __init__(
                 options = None
 
         super().__init__(
-            api_key=api_key, access_token=access_token, url=url, verbose=verbose, headers=headers, options=options
+            api_key=api_key,
+            access_token=access_token,
+            url=url,
+            verbose=verbose,
+            headers=headers,
+            options=options,
         )

examples/agent/arbitrary_keys/main.py

@@ -22,6 +22,7 @@
 )
 from deepgram.clients.agent.v1.websocket.options import SettingsOptions
 
+
 def main():
     try:
         # Initialize the Voice Agent
@@ -64,17 +65,17 @@ def main():
 
         def on_welcome(self, welcome, **kwargs):
             print(f"Welcome message received: {welcome}")
-            with open("chatlog.txt", 'a') as chatlog:
+            with open("chatlog.txt", "a") as chatlog:
                 chatlog.write(f"Welcome message: {welcome}\n")
 
         def on_settings_applied(self, settings_applied, **kwargs):
             print(f"Settings applied: {settings_applied}")
-            with open("chatlog.txt", 'a') as chatlog:
+            with open("chatlog.txt", "a") as chatlog:
                 chatlog.write(f"Settings applied: {settings_applied}\n")
 
         def on_error(self, error, **kwargs):
             print(f"Error received: {error}")
-            with open("chatlog.txt", 'a') as chatlog:
+            with open("chatlog.txt", "a") as chatlog:
                 chatlog.write(f"Error: {error}\n")
 
         # Register handlers
@@ -93,11 +94,14 @@ def on_error(self, error, **kwargs):
 
         # Cleanup
         connection.finish()
-        print("Finished! You should see an error for the arbitrary key - scroll up and you can see it is included in the settings payload.")
+        print(
+            "Finished! You should see an error for the arbitrary key - scroll up and you can see it is included in the settings payload."
+        )
         print("If you do not see that error, this example has failed.")
 
     except Exception as e:
         print(f"Error: {str(e)}")
 
+
 if __name__ == "__main__":
     main()

examples/agent/async_simple/main.py

@@ -116,7 +116,6 @@ async def on_unhandled(self, unhandled, **kwargs):
         options.agent.listen.provider.type = "deepgram"
         options.agent.language = "en"
 
-
         print("\n\nPress Enter to stop...\n\n")
         if await dg_connection.start(options) is False:
             print("Failed to start connection")