promptlayer 1.0.35__py3-none-any.whl → 1.0.78__py3-none-any.whl

promptlayer/__init__.py

@@ -1,4 +1,39 @@
+from .exceptions import (
+    PromptLayerAPIConnectionError,
+    PromptLayerAPIError,
+    PromptLayerAPIStatusError,
+    PromptLayerAPITimeoutError,
+    PromptLayerAuthenticationError,
+    PromptLayerBadRequestError,
+    PromptLayerConflictError,
+    PromptLayerError,
+    PromptLayerInternalServerError,
+    PromptLayerNotFoundError,
+    PromptLayerPermissionDeniedError,
+    PromptLayerRateLimitError,
+    PromptLayerUnprocessableEntityError,
+    PromptLayerValidationError,
+)
 from .promptlayer import AsyncPromptLayer, PromptLayer
 
-__version__ = "1.0.35"
-__all__ = ["PromptLayer", "AsyncPromptLayer", "__version__"]
+__version__ = "1.0.78"
+__all__ = [
+    "PromptLayer",
+    "AsyncPromptLayer",
+    "__version__",
+    # Exceptions
+    "PromptLayerError",
+    "PromptLayerAPIError",
+    "PromptLayerBadRequestError",
+    "PromptLayerAuthenticationError",
+    "PromptLayerPermissionDeniedError",
+    "PromptLayerNotFoundError",
+    "PromptLayerConflictError",
+    "PromptLayerUnprocessableEntityError",
+    "PromptLayerRateLimitError",
+    "PromptLayerInternalServerError",
+    "PromptLayerAPIStatusError",
+    "PromptLayerAPIConnectionError",
+    "PromptLayerAPITimeoutError",
+    "PromptLayerValidationError",
+]

promptlayer/exceptions.py

@@ -0,0 +1,119 @@
+class PromptLayerError(Exception):
+    """Base exception for all PromptLayer SDK errors."""
+
+    def __init__(self, message: str, response=None, body=None):
+        super().__init__(message)
+        self.message = message
+        self.response = response
+        self.body = body
+
+    def __str__(self):
+        return self.message
+
+
+class PromptLayerAPIError(PromptLayerError):
+    """Base exception for API-related errors."""
+
+    pass
+
+
+class PromptLayerBadRequestError(PromptLayerAPIError):
+    """Exception raised for 400 Bad Request errors.
+
+    Indicates that the request was malformed or contained invalid parameters.
+    """
+
+    pass
+
+
+class PromptLayerAuthenticationError(PromptLayerAPIError):
+    """Exception raised for 401 Unauthorized errors.
+
+    Indicates that the API key is missing, invalid, or expired.
+    """
+
+    pass
+
+
+class PromptLayerPermissionDeniedError(PromptLayerAPIError):
+    """Exception raised for 403 Forbidden errors.
+
+    Indicates that the API key doesn't have permission to perform the requested operation.
+    """
+
+    pass
+
+
+class PromptLayerNotFoundError(PromptLayerAPIError):
+    """Exception raised for 404 Not Found errors.
+
+    Indicates that the requested resource (e.g., prompt template) was not found.
+    """
+
+    pass
+
+
+class PromptLayerConflictError(PromptLayerAPIError):
+    """Exception raised for 409 Conflict errors.
+
+    Indicates that the request conflicts with the current state of the resource.
+    """
+
+    pass
+
+
+class PromptLayerUnprocessableEntityError(PromptLayerAPIError):
+    """Exception raised for 422 Unprocessable Entity errors.
+
+    Indicates that the request was well-formed but contains semantic errors.
+    """
+
+    pass
+
+
+class PromptLayerRateLimitError(PromptLayerAPIError):
+    """Exception raised for 429 Too Many Requests errors.
+
+    Indicates that the API rate limit has been exceeded.
+    """
+
+    pass
+
+
+class PromptLayerInternalServerError(PromptLayerAPIError):
+    """Exception raised for 500+ Internal Server errors.
+
+    Indicates that the PromptLayer API encountered an internal error.
+    """
+
+    pass
+
+
+class PromptLayerAPIStatusError(PromptLayerAPIError):
+    """Exception raised for other API errors not covered by specific exception classes."""
+
+    pass
+
+
+class PromptLayerAPIConnectionError(PromptLayerError):
+    """Exception raised when unable to connect to the API.
+
+    This can be due to network issues, timeouts, or connection errors.
+    """
+
+    pass
+
+
+class PromptLayerAPITimeoutError(PromptLayerError):
+    """Exception raised when an API request times out."""
+
+    pass
+
+
+class PromptLayerValidationError(PromptLayerError):
+    """Exception raised when input validation fails.
+
+    This can be due to invalid types, out of range values, or malformed data.
+    """
+
+    pass

promptlayer/groups/__init__.py

@@ -2,19 +2,23 @@ from promptlayer.groups.groups import acreate, create
 
 
 class GroupManager:
-    def __init__(self, api_key: str):
+    def __init__(self, api_key: str, base_url: str, throw_on_error: bool):
         self.api_key = api_key
+        self.base_url = base_url
+        self.throw_on_error = throw_on_error
 
     def create(self):
-        return create(self.api_key)
+        return create(self.api_key, self.base_url, self.throw_on_error)
 
 
 class AsyncGroupManager:
-    def __init__(self, api_key: str):
+    def __init__(self, api_key: str, base_url: str, throw_on_error: bool):
         self.api_key = api_key
+        self.base_url = base_url
+        self.throw_on_error = throw_on_error
 
-    async def create(self) -> str:
-        return await acreate(self.api_key)
+    async def create(self):
+        return await acreate(self.api_key, self.base_url, self.throw_on_error)
 
 
 __all__ = ["GroupManager", "AsyncGroupManager"]

promptlayer/groups/groups.py

@@ -1,11 +1,9 @@
 from promptlayer.utils import apromptlayer_create_group, promptlayer_create_group
 
 
-def create(api_key: str = None):
-    """Create a new group."""
-    return promptlayer_create_group(api_key)
+def create(api_key: str, base_url: str, throw_on_error: bool):
+    return promptlayer_create_group(api_key, base_url, throw_on_error)
 
 
-async def acreate(api_key: str = None) -> str:
-    """Asynchronously create a new group."""
-    return await apromptlayer_create_group(api_key)
+async def acreate(api_key: str, base_url: str, throw_on_error: bool):
+    return await apromptlayer_create_group(api_key, base_url, throw_on_error)