Allow an override on the API key and version bump (#38)

* Allow an override on the API key

* version bump

* Add future import

* Use old school type hint

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "cloai"
-version = "0.0.1a12"
+version = "0.0.1a13"
 description = "A CLI for OpenAI's API"
 authors = ["Reinder Vos de Wael <[email protected]>"]
 license = "LGPL-2.1"

src/cloai/core/config.py

@@ -4,7 +4,7 @@
 import logging
 import pathlib
 from importlib import metadata
-from typing import Literal
+from typing import Literal, Optional
 
 import pydantic
 import pydantic_settings
@@ -86,8 +86,8 @@ class Settings(pydantic_settings.BaseSettings):
     LOGGER_NAME: str = "cloai"
     LOGGER_VERBOSITY: Literal["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"] = "INFO"
 
-    OPENAI_API_KEY: pydantic.SecretStr = pydantic.Field(
-        ...,
+    OPENAI_API_KEY: Optional[pydantic.SecretStr] = pydantic.Field(  # noqa: UP007 Disable because pydantic and future annotations don't play well together.
+        None,
         json_schema_extra={
             "env": "OPENAI_API_KEY",
             "description": "The API key for OpenAI.",

src/cloai/openai_api.py

@@ -34,9 +34,17 @@ class OpenAIBaseClass(abc.ABC):
         client: The OpenAI client used to interact with the model.
     """
 
-    def __init__(self) -> None:
+    def __init__(self, api_key: str | None = None) -> None:
         """Initializes a new instance of the OpenAIBaseClass class."""
-        self.client = openai.AsyncOpenAI(api_key=OPENAI_API_KEY.get_secret_value())
+        if api_key:
+            key = api_key
+        elif OPENAI_API_KEY:
+            key = OPENAI_API_KEY.get_secret_value()
+        else:
+            msg = "No API key provided."
+            raise exceptions.OpenAIError(msg)
+
+        self.client = openai.AsyncOpenAI(api_key=key)
 
     @abc.abstractmethod
     async def run(self, *_args: Any, **_kwargs: Any) -> Any:  # noqa: ANN401