Get api key at client initialization (#57)

Author: antoine-lizee

src/mistralai/async_client.py

@@ -34,7 +34,7 @@
 class MistralAsyncClient(ClientBase):
     def __init__(
         self,
-        api_key: Optional[str] = os.environ.get("MISTRAL_API_KEY", None),
+        api_key: Optional[str] = None,
         endpoint: str = ENDPOINT,
         max_retries: int = 5,
         timeout: int = 120,

src/mistralai/client.py

@@ -31,7 +31,7 @@ class MistralClient(ClientBase):
 
     def __init__(
         self,
-        api_key: Optional[str] = os.environ.get("MISTRAL_API_KEY", None),
+        api_key: Optional[str] = None,
         endpoint: str = ENDPOINT,
         max_retries: int = 5,
         timeout: int = 120,

src/mistralai/client_base.py

@@ -22,8 +22,14 @@ def __init__(
         self._max_retries = max_retries
         self._timeout = timeout
 
-        self._endpoint = endpoint
+        if api_key is None:
+            api_key = os.environ.get("MISTRAL_API_KEY")
+        if api_key is None:
+            raise MistralException(
+                message="API key not provided. Please set MISTRAL_API_KEY environment variable."
+            )
         self._api_key = api_key
+        self._endpoint = endpoint
         self._logger = logging.getLogger(__name__)
 
         # For azure endpoints, we default to the mistral model