5.1.2 client.constant -> client.constants fix

.gitignore

@@ -0,0 +1 @@
+tests

prodiapy/_client.py

@@ -1,5 +1,6 @@
 
 from prodiapy.resources.engine import SyncAPIClient
+from prodiapy.resources import constants
 from prodiapy import resources
 from typing import Optional
 from prodiapy._exceptions import *
@@ -8,6 +9,7 @@
 
 
 class Prodia(SyncAPIClient):
+    """Prodia Client"""
     api_key: str
     sd: resources.StableDiffusion
     sdxl: resources.StableDiffusionXL
@@ -24,10 +26,7 @@ def __init__(
             - `api_key` from `PRODIA_API_KEY`
         """
         if api_key is None:
-            raise AuthenticationError(
-                "The api_key client option must be set either by passing api_key to the client or by setting the \
-PRODIA_API_KEY environment variable"
-            )
+            raise AuthenticationError(constants.AUTHENTICATION_ERROR)
         self.api_key = api_key
 
         super().__init__(
@@ -44,7 +43,7 @@ def __init__(
         self.upscale = general.upscale
         self.create = general.create
         self.job = general.job
-        self.constant = general.constants
+        self.constants = general.constants
         self.wait = general.wait
 
     @property

prodiapy/aio.py

@@ -1,13 +1,14 @@
+from typing import Optional
+import os
 
 from prodiapy.resources.engine import AsyncAPIClient
+from prodiapy.resources import constants
 from prodiapy import resources
-from typing import Optional
-from prodiapy._exceptions import *
-
-import os
+from prodiapy._exceptions import AuthenticationError
 
 
 class Prodia(AsyncAPIClient):
+    """Asynchronous Prodia Client"""
     api_key: str
     sd: resources.AsyncStableDiffusion
     sdxl: resources.AsyncStableDiffusionXL
@@ -24,10 +25,7 @@ def __init__(
             - `api_key` from `PRODIA_API_KEY`
         """
         if api_key is None:
-            raise AuthenticationError(
-                "The api_key client option must be set either by passing api_key to the client or by setting the \
-PRODIA_API_KEY environment variable"
-            )
+            raise AuthenticationError(constants.AUTHENTICATION_ERROR)
         self.api_key = api_key
 
         super().__init__(
@@ -44,10 +42,10 @@ def __init__(
         self.upscale = general.upscale
         self.create = general.create
         self.job = general.job
-        self.constant = general.constants
+        self.constants = general.constants
         self.wait = general.wait
 
     @property
     def auth_headers(self) -> dict:
         api_key = self.api_key
-        return {'X-Prodia-Key': api_key, 'Content-Type': 'application/json'}
+        return {'X-Prodia-Key': api_key, 'Content-Type': 'application/json'}

prodiapy/resources/constants.py

@@ -0,0 +1,3 @@
+
+AUTHENTICATION_ERROR = "The api_key client option must be set either by passing api_key to the client \
+or by setting the PRODIA_API_KEY environment variable"

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prodiapy"
-version = "5.1.1"
+version = "5.1.2"
 description = "Package for using Prodia API"
 authors = ["zenafey <[email protected]>"]
 readme = "README.md"