Skip to content

Commit 4712ba6

Browse files
feat(api): remove inf endpoint env
1 parent d98189e commit 4712ba6

2 files changed

Lines changed: 5 additions & 7 deletions

File tree

.stats.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 193
22
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-24bdeb83fbe576e4dd92498321be977e820fd755babee77eb8ff419a6229b22f.yml
33
openapi_spec_hash: f1faf4e4e0fc37e518fe55ff7c01086d
4-
config_hash: 3f968a57adb20643373c134efc9af01a
4+
config_hash: 7566062db4ea811277129e8f8e848f6c

src/gradient/_client.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def __init__(
9191
model_access_key: str | None = None,
9292
agent_access_key: str | None = None,
9393
agent_endpoint: str | None = None,
94-
inference_endpoint: str | None = None,
94+
inference_endpoint: str | None = "inference.do-ai.run",
9595
kbass_endpoint: str | None = None,
9696
base_url: str | httpx.URL | None = None,
9797
timeout: float | Timeout | None | NotGiven = not_given,
@@ -119,7 +119,6 @@ def __init__(
119119
- `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
120120
- `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
121121
- `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
122-
- `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
123122
- `kbass_endpoint` from `GRADIENT_KBASS_ENDPOINT`
124123
"""
125124
if access_token is None:
@@ -139,7 +138,7 @@ def __init__(
139138
self.agent_endpoint = agent_endpoint
140139

141140
if inference_endpoint is None:
142-
inference_endpoint = os.environ.get("GRADIENT_INFERENCE_ENDPOINT") or "inference.do-ai.run"
141+
inference_endpoint = "inference.do-ai.run"
143142
self.inference_endpoint = inference_endpoint
144143

145144
if kbass_endpoint is None:
@@ -469,7 +468,7 @@ def __init__(
469468
model_access_key: str | None = None,
470469
agent_access_key: str | None = None,
471470
agent_endpoint: str | None = None,
472-
inference_endpoint: str | None = None,
471+
inference_endpoint: str | None = "inference.do-ai.run",
473472
kbass_endpoint: str | None = None,
474473
base_url: str | httpx.URL | None = None,
475474
timeout: float | Timeout | None | NotGiven = not_given,
@@ -497,7 +496,6 @@ def __init__(
497496
- `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
498497
- `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
499498
- `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
500-
- `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
501499
- `kbass_endpoint` from `GRADIENT_KBASS_ENDPOINT`
502500
"""
503501
if access_token is None:
@@ -517,7 +515,7 @@ def __init__(
517515
self.agent_endpoint = agent_endpoint
518516

519517
if inference_endpoint is None:
520-
inference_endpoint = os.environ.get("GRADIENT_INFERENCE_ENDPOINT") or "inference.do-ai.run"
518+
inference_endpoint = "inference.do-ai.run"
521519
self.inference_endpoint = inference_endpoint
522520

523521
if kbass_endpoint is None:

0 commit comments

Comments
 (0)