From 0220adf64af4770c352e176ec330c683b2e67764 Mon Sep 17 00:00:00 2001
From: Jason Maldonis <jason.maldonis@deepgram.com>
Date: Mon, 4 Dec 2023 14:20:31 -0600
Subject: [PATCH] made the api key optional for on-prem

---
 deepgram/client.py  |  4 +++-
 deepgram/options.py | 28 +++++++++++++++-------------
 2 files changed, 18 insertions(+), 14 deletions(-)

diff --git a/deepgram/client.py b/deepgram/client.py
index 3ee7050f..857c37cd 100644
--- a/deepgram/client.py
+++ b/deepgram/client.py
@@ -5,6 +5,7 @@
 from typing import Optional
 from importlib import import_module
 import logging, verboselogs
+import os
 
 from .clients.listen import ListenClient, PreRecordedClient
 from .clients.manage.client import ManageClient
@@ -39,7 +40,8 @@ def __init__(self, api_key: str, config: Optional[DeepgramClientOptions] = None)
         self.logger.addHandler(logging.StreamHandler())
 
         if not api_key:
-            raise DeepgramApiKeyError("Deepgram API key is required")
+            # Default to `None` for on-prem instances where an API key is not required
+            api_key = os.getenv("DEEPGRAM_API_KEY", None)
 
         self.api_key = api_key
         if config is None:  # Use default configuration
diff --git a/deepgram/options.py b/deepgram/options.py
index 823aba07..5218f544 100644
--- a/deepgram/options.py
+++ b/deepgram/options.py
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: MIT
 
 import logging, verboselogs
-from typing import Dict
+from typing import Dict, Optional
 import re
 
 
@@ -31,15 +31,7 @@ def __init__(
     ):
         self.verbose = verbose
         self.api_key = api_key
-        if headers is None:
-            self.headers = {
-                "Accept": "application/json",
-                "Authorization": f"Token {self.api_key}",
-            }
-        else:
-            self.headers.update(
-                {"Accept": "application/json", "Authorization": f"Token {self.api_key}"}
-            )
+        self._update_headers(headers=headers)
         if len(url) == 0:
             url = "api.deepgram.com"
         self.url = self._get_url(url)
@@ -49,11 +41,21 @@ def __init__(
 
     def set_apikey(self, api_key: str):
         self.api_key = api_key
-        self.headers.update(
-            {"Accept": "application/json", "Authorization": f"Token {self.api_key}"}
-        )
+        self._update_headers()
 
     def _get_url(self, url):
         if not re.match(r"^https?://", url, re.IGNORECASE):
             url = "https://" + url
         return url.strip("/")
+
+    def _update_headers(self, headers: Optional[Dict[str, str]] = None):
+        if not hasattr(self, "headers") or self.headers is None:
+            self.headers = {}
+        self.headers["Accept"] = "application/json"
+        if self.api_key:
+            self.headers["Authorization"] = f"Token {self.api_key}"
+        elif "Authorization" in self.headers:
+            del self.headers["Authorization"]
+        # Overwrite / add any headers that were passed in
+        if headers:
+            self.headers.update(headers)