Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions examples/agent_wait_until_ready.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@
an agent to finish deploying before using it.
"""

from gradient import Gradient
from gradient._exceptions import AgentDeploymentError, AgentDeploymentTimeoutError
from gradient import Gradient, AgentDeploymentError, AgentDeploymentTimeoutError

# Initialize the Gradient client
client = Gradient()
Expand Down
4 changes: 4 additions & 0 deletions src/gradient/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,8 @@
PermissionDeniedError,
UnprocessableEntityError,
APIResponseValidationError,
AgentDeploymentError,
AgentDeploymentTimeoutError,
)
from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
from ._utils._logs import setup_logging as _setup_logging
Expand All @@ -57,6 +59,8 @@
"APITimeoutError",
"APIConnectionError",
"APIResponseValidationError",
"AgentDeploymentError",
"AgentDeploymentTimeoutError",
"BadRequestError",
"AuthenticationError",
"PermissionDeniedError",
Expand Down
Loading