fix client error type

togethercomputer · Mar 9, 2024 · 6007591 · 6007591
1 parent 8ceaf3d
commit 6007591
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/together/client.py b/src/together/client.py
@@ -3,7 +3,7 @@
 
 from together import resources
 from together.constants import BASE_URL, MAX_CONNECTION_RETRIES, TIMEOUT_SECS
-from together.error import TogetherException
+from together.error import AuthenticationError
 from together.types import TogetherClient
 
 
@@ -40,7 +40,7 @@ def __init__(
             api_key = os.environ.get("TOGETHER_API_KEY")
 
         if api_key is None:
-            raise TogetherException(
+            raise AuthenticationError(
                 "The api_key client option must be set either by passing api_key to the client or by setting the "
                 "TOGETHER_API_KEY environment variable"
             )
@@ -102,7 +102,7 @@ def __init__(
             api_key = os.environ.get("TOGETHER_API_KEY")
 
         if api_key is None:
-            raise TogetherException(
+            raise AuthenticationError(
                 "The api_key client option must be set either by passing api_key to the client or by setting the "
                 "TOGETHER_API_KEY environment variable"
             )