Assign api_key on init call

Author: chandrasekharan-zipstack

.gitignore

@@ -159,5 +159,7 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 .idea/
 
+.vscode/
+
 .pdm-python
 .python-version

src/llmwhisperer/client.py

@@ -22,6 +22,8 @@
 
 import requests
 
+from llmwhisperer.utils import LLMWhispererUtils
+
 BASE_URL = "https://llmwhisperer-api.unstract.com/v1"
 
 
@@ -110,6 +112,9 @@ def __init__(
 
         if api_key == "":
             self.api_key = os.getenv("LLMWHISPERER_API_KEY", "")
+        else:
+            self.api_key = api_key
+        self.logger.debug("api_key set to %s", LLMWhispererUtils.redact_key(self.api_key))
 
         self.api_timeout = api_timeout
 

src/llmwhisperer/utils.py

@@ -0,0 +1,21 @@
+class LLMWhispererUtils:
+    @staticmethod
+    def redact_key(api_key: str, reveal_length=4) -> str:
+        """Hides sensitive information partially. Useful for logging keys.
+
+        Args:
+            api_key (str): API key to redact
+
+        Returns:
+            str: Redacted API key
+        """
+        if not isinstance(api_key, str):
+            raise ValueError("API key must be a string")
+
+        if reveal_length < 0:
+            raise ValueError("Reveal length must be a non-negative integer")
+
+        redacted_length = max(len(api_key) - reveal_length, 0)
+        revealed_part = api_key[:reveal_length]
+        redacted_part = "x" * redacted_length
+        return revealed_part + redacted_part