Skip to content

Commit 6c22d61

Browse files
feat(api): manual updates
1 parent bb8e9df commit 6c22d61

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 170
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-9aca3802735e1375125412aa28ac36bf2175144b8218610a73d2e7f775694dff.yml
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-9ecd7b6ec2c5f4b9dddde78a975eba587128529bcd7cea1fc82e041ad9450420.yml
33
openapi_spec_hash: e29d14e3e4679fcf22b3e760e49931b1
4-
config_hash: 99e3cd5dde0beb796f4547410869f726
4+
config_hash: 868a2e440ec27146f18b1327daa1cd63

src/gradient/_client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ def __init__(
113113
- `access_token` from `DIGITALOCEAN_ACCESS_TOKEN`
114114
- `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
115115
- `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
116+
- `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
116117
"""
117118
if access_token is None:
118119
if api_key is not None:
@@ -432,6 +433,7 @@ def __init__(
432433
- `access_token` from `DIGITALOCEAN_ACCESS_TOKEN`
433434
- `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
434435
- `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
436+
- `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
435437
"""
436438
if access_token is None:
437439
if api_key is not None:
@@ -463,6 +465,8 @@ def __init__(
463465
agent_access_key = os.environ.get("GRADIENT_AGENT_KEY")
464466
self.agent_access_key = agent_access_key
465467

468+
if agent_endpoint is None:
469+
agent_endpoint = os.environ.get("GRADIENT_AGENT_ENDPOINT")
466470
self._agent_endpoint = agent_endpoint
467471

468472
self.inference_endpoint = inference_endpoint

0 commit comments

Comments
 (0)