Report auth error. (#5)

hotgluexyz · Jun 3, 2024 · 0f73c18 · 0f73c18
1 parent 1823224
commit 0f73c18
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tap_klaviyo/client.py b/tap_klaviyo/client.py
@@ -30,15 +30,18 @@ class KlaviyoStream(RESTStream):
 
     @property
     def authenticator(self):
+        api_key = self.config.get("api_private_key") or self.config.get("api_key")
         # auth with access token
         if self.config.get("refresh_token"):
             return KlaviyoAuthenticator.create_for_stream(self)
         # auth with api key
-        elif self.config.get("api_private_key"):
-            api_key = f'Klaviyo-API-Key {self.config.get("api_private_key")}'
+        elif api_key:
+            api_key = f"Klaviyo-API-Key {api_key}"
             return APIKeyAuthenticator.create_for_stream(
                 self, key="Authorization", value=api_key, location="header"
             )
+        else:
+            raise FatalAPIError("No valid authentication method found")    
 
     @property
     def http_headers(self) -> dict: