Skip to content

Commit f1c2eb2

Browse files
stainless-app[bot]stainless-app[bot]
committed
feat(api): manual updates
1 parent ee792a1 commit f1c2eb2

File tree

10 files changed

+48
-412
lines changed

10 files changed

+48
-412
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: 189
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-ca993fd0bc66f703323a773c92da75207266f7f9d8c54ddac2fbd271a3cdaf86.yml
3-
openapi_spec_hash: 35d7edb04aab2ab28bc7e5851a54b4e3
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-0778b2e9d56c826f92ee69ef081d8d73fd94c139b85e11becaa88bf1cbe95fb9.yml
3+
openapi_spec_hash: 49daca0dd735cad7200ca1c741a5dd43
44
config_hash: fad48c8ac796b240fe3b90181586d1a4

‎src/gradient/_client.py‎

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -142,10 +142,7 @@ def __init__(
142142
self._agent_endpoint = agent_endpoint
143143

144144
if inference_endpoint is None:
145-
inference_endpoint = (
146-
os.environ.get("GRADIENT_INFERENCE_ENDPOINT")
147-
or "https://inference.do-ai.run"
148-
)
145+
inference_endpoint = os.environ.get("GRADIENT_INFERENCE_ENDPOINT") or "https://inference.do-ai.run"
149146
self.inference_endpoint = inference_endpoint
150147

151148
if kbass_endpoint is None:
@@ -302,9 +299,7 @@ def default_headers(self) -> dict[str, str | Omit]:
302299

303300
@override
304301
def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
305-
if (
306-
self.access_token or self.agent_access_key or self.model_access_key
307-
) and headers.get("Authorization"):
302+
if (self.access_token or self.agent_access_key or self.model_access_key) and headers.get("Authorization"):
308303
return
309304
if isinstance(custom_headers.get("Authorization"), Omit):
310305
return
@@ -513,10 +508,7 @@ def __init__(
513508
self._agent_endpoint = agent_endpoint
514509

515510
if inference_endpoint is None:
516-
inference_endpoint = (
517-
os.environ.get("GRADIENT_INFERENCE_ENDPOINT")
518-
or "https://inference.do-ai.run"
519-
)
511+
inference_endpoint = os.environ.get("GRADIENT_INFERENCE_ENDPOINT") or "https://inference.do-ai.run"
520512
self.inference_endpoint = inference_endpoint
521513

522514
if kbass_endpoint is None:
@@ -673,9 +665,7 @@ def default_headers(self) -> dict[str, str | Omit]:
673665

674666
@override
675667
def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
676-
if (
677-
self.access_token or self.agent_access_key or self.model_access_key
678-
) and headers.get("Authorization"):
668+
if (self.access_token or self.agent_access_key or self.model_access_key) and headers.get("Authorization"):
679669
return
680670
if isinstance(custom_headers.get("Authorization"), Omit):
681671
return

‎src/gradient/resources/agents/chat/completions.py‎

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -472,9 +472,7 @@ def create(
472472
headers = {"Authorization": f"Bearer {self._client.agent_access_key}", **headers}
473473

474474
return self._post(
475-
"/chat/completions?agent=true"
476-
if self._client._base_url_overridden
477-
else f"{self._client.agent_endpoint}/api/v1/chat/completions?agent=true",
475+
"/chat/completions?agent=true",
478476
body=maybe_transform(
479477
{
480478
"messages": messages,
@@ -958,9 +956,7 @@ async def create(
958956
headers = {"Authorization": f"Bearer {self._client.agent_access_key}", **headers}
959957

960958
return await self._post(
961-
"/chat/completions?agent=true"
962-
if self._client._base_url_overridden
963-
else f"{self._client.agent_endpoint}/api/v1/chat/completions?agent=true",
959+
"/chat/completions?agent=true",
964960
body=await async_maybe_transform(
965961
{
966962
"messages": messages,

‎src/gradient/resources/chat/completions.py‎

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -473,9 +473,7 @@ def create(
473473
}
474474

475475
return self._post(
476-
"/chat/completions"
477-
if self._client._base_url_overridden
478-
else f"{self._client.inference_endpoint}/v1/chat/completions",
476+
"/chat/completions",
479477
body=maybe_transform(
480478
{
481479
"messages": messages,
@@ -963,9 +961,7 @@ async def create(
963961
}
964962

965963
return await self._post(
966-
"/chat/completions"
967-
if self._client._base_url_overridden
968-
else f"{self._client.inference_endpoint}/v1/chat/completions",
964+
"/chat/completions",
969965
body=await async_maybe_transform(
970966
{
971967
"messages": messages,

‎src/gradient/resources/images.py‎

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -317,9 +317,7 @@ def generate(
317317
}
318318

319319
return self._post(
320-
"/images/generations"
321-
if self._client._base_url_overridden
322-
else f"{self._client.inference_endpoint}/v1/images/generations",
320+
"/images/generations",
323321
body=maybe_transform(
324322
{
325323
"prompt": prompt,
@@ -638,9 +636,7 @@ async def generate(
638636
**headers,
639637
}
640638
return await self._post(
641-
"/images/generations"
642-
if self._client._base_url_overridden
643-
else f"{self._client.inference_endpoint}/v1/images/generations",
639+
"/images/generations",
644640
body=await async_maybe_transform(
645641
{
646642
"prompt": prompt,

‎src/gradient/resources/knowledge_bases/knowledge_bases.py‎

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -434,9 +434,7 @@ def wait_for_database(
434434
KnowledgeBaseTimeoutError: If the timeout is exceeded before the database becomes ONLINE
435435
"""
436436
if not uuid:
437-
raise ValueError(
438-
f"Expected a non-empty value for `uuid` but received {uuid!r}"
439-
)
437+
raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
440438

441439
start_time = time.time()
442440
failed_states = {"DECOMMISSIONED", "UNHEALTHY"}
@@ -462,9 +460,7 @@ def wait_for_database(
462460
return response
463461

464462
if status in failed_states:
465-
raise KnowledgeBaseDatabaseError(
466-
f"Knowledge base database entered failed state: {status}"
467-
)
463+
raise KnowledgeBaseDatabaseError(f"Knowledge base database entered failed state: {status}")
468464

469465
# Sleep before next poll, but don't exceed timeout
470466
remaining_time = timeout - elapsed
@@ -874,9 +870,7 @@ async def wait_for_database(
874870
KnowledgeBaseTimeoutError: If the timeout is exceeded before the database becomes ONLINE
875871
"""
876872
if not uuid:
877-
raise ValueError(
878-
f"Expected a non-empty value for `uuid` but received {uuid!r}"
879-
)
873+
raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
880874

881875
start_time = time.time()
882876
failed_states = {"DECOMMISSIONED", "UNHEALTHY"}
@@ -902,9 +896,7 @@ async def wait_for_database(
902896
return response
903897

904898
if status in failed_states:
905-
raise KnowledgeBaseDatabaseError(
906-
f"Knowledge base database entered failed state: {status}"
907-
)
899+
raise KnowledgeBaseDatabaseError(f"Knowledge base database entered failed state: {status}")
908900

909901
# Sleep before next poll, but don't exceed timeout
910902
remaining_time = timeout - elapsed

0 commit comments

Comments
 (0)