Skip to content

Commit c202e81

Browse files
committed
chore(client): support model_access_key in image generations for real
1 parent 4b81c5c commit c202e81

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

src/gradient/resources/images/generations.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def create(
151151
generation_create_params.GenerationCreateParams,
152152
),
153153
options=make_request_options(
154-
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
154+
extra_headers=headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
155155
),
156156
cast_to=GenerationCreateResponse,
157157
)
@@ -253,6 +253,16 @@ async def create(
253253
254254
timeout: Override the client-level default timeout for this request, in seconds
255255
"""
256+
if not self._client.model_access_key:
257+
raise TypeError(
258+
"Could not resolve authentication method. Expected model_access_key to be set for chat completions."
259+
)
260+
headers = extra_headers or {}
261+
headers = {
262+
"Authorization": f"Bearer {self._client.model_access_key}",
263+
**headers,
264+
}
265+
256266
return await self._post(
257267
"/images/generations"
258268
if self._client._base_url_overridden
@@ -275,7 +285,7 @@ async def create(
275285
generation_create_params.GenerationCreateParams,
276286
),
277287
options=make_request_options(
278-
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
288+
extra_headers=headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
279289
),
280290
cast_to=GenerationCreateResponse,
281291
)

0 commit comments

Comments
 (0)