@@ -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