Skip to content

Commit 4b81c5c

Browse files
committed
chore(client): support model_access_key in image generations
1 parent 76d29b6 commit 4b81c5c

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/gradient/resources/images/generations.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,16 @@ def create(
119119
120120
timeout: Override the client-level default timeout for this request, in seconds
121121
"""
122+
if not self._client.model_access_key:
123+
raise TypeError(
124+
"Could not resolve authentication method. Expected model_access_key to be set for chat completions."
125+
)
126+
headers = extra_headers or {}
127+
headers = {
128+
"Authorization": f"Bearer {self._client.model_access_key}",
129+
**headers,
130+
}
131+
122132
return self._post(
123133
"/images/generations"
124134
if self._client._base_url_overridden

0 commit comments

Comments
 (0)