Skip to content

Commit 76392b5

Browse files
authored
Expose agent deployment exceptions via public API and fix example import (#72)
1 parent 9edc2a6 commit 76392b5

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

examples/agent_wait_until_ready.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,7 @@
55
an agent to finish deploying before using it.
66
"""
77

8-
from gradient import Gradient
9-
from gradient._exceptions import AgentDeploymentError, AgentDeploymentTimeoutError
8+
from gradient import Gradient, AgentDeploymentError, AgentDeploymentTimeoutError
109

1110
# Initialize the Gradient client
1211
client = Gradient()

src/gradient/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,8 @@
6161
"APITimeoutError",
6262
"APIConnectionError",
6363
"APIResponseValidationError",
64+
"AgentDeploymentError",
65+
"AgentDeploymentTimeoutError",
6466
"BadRequestError",
6567
"AuthenticationError",
6668
"PermissionDeniedError",

0 commit comments

Comments
 (0)