77
88import httpx
99
10- from ..._types import Body , Omit , Query , Headers , NotGiven , omit , not_given
11- from ..._utils import required_args , maybe_transform , async_maybe_transform
12- from ..._compat import cached_property
13- from ..._resource import SyncAPIResource , AsyncAPIResource
14- from ..._response import (
10+ from ..types import image_generate_params
11+ from .._types import Body , Omit , Query , Headers , NotGiven , omit , not_given
12+ from .._utils import required_args , maybe_transform , async_maybe_transform
13+ from .._compat import cached_property
14+ from .._resource import SyncAPIResource , AsyncAPIResource
15+ from .._response import (
1516 to_raw_response_wrapper ,
1617 to_streamed_response_wrapper ,
1718 async_to_raw_response_wrapper ,
1819 async_to_streamed_response_wrapper ,
1920)
20- from ..._streaming import Stream , AsyncStream
21- from ..._base_client import make_request_options
22- from ...types .images import generation_create_params
23- from ...types .shared .image_gen_stream_event import ImageGenStreamEvent
24- from ...types .images .generation_create_response import GenerationCreateResponse
21+ from .._streaming import Stream , AsyncStream
22+ from .._base_client import make_request_options
23+ from ..types .image_generate_response import ImageGenerateResponse
24+ from ..types .shared .image_gen_stream_event import ImageGenStreamEvent
2525
26- __all__ = ["GenerationsResource " , "AsyncGenerationsResource " ]
26+ __all__ = ["ImagesResource " , "AsyncImagesResource " ]
2727
2828
29- class GenerationsResource (SyncAPIResource ):
29+ class ImagesResource (SyncAPIResource ):
3030 @cached_property
31- def with_raw_response (self ) -> GenerationsResourceWithRawResponse :
31+ def with_raw_response (self ) -> ImagesResourceWithRawResponse :
3232 """
3333 This property can be used as a prefix for any HTTP method call to return
3434 the raw response object instead of the parsed content.
3535
3636 For more information, see https://www.github.com/digitalocean/gradient-python#accessing-raw-response-data-eg-headers
3737 """
38- return GenerationsResourceWithRawResponse (self )
38+ return ImagesResourceWithRawResponse (self )
3939
4040 @cached_property
41- def with_streaming_response (self ) -> GenerationsResourceWithStreamingResponse :
41+ def with_streaming_response (self ) -> ImagesResourceWithStreamingResponse :
4242 """
4343 An alternative to `.with_raw_response` that doesn't eagerly read the response body.
4444
4545 For more information, see https://www.github.com/digitalocean/gradient-python#with_streaming_response
4646 """
47- return GenerationsResourceWithStreamingResponse (self )
47+ return ImagesResourceWithStreamingResponse (self )
4848
4949 @overload
50- def create (
50+ def generate (
5151 self ,
5252 * ,
5353 prompt : str ,
@@ -68,7 +68,7 @@ def create(
6868 extra_query : Query | None = None ,
6969 extra_body : Body | None = None ,
7070 timeout : float | httpx .Timeout | None | NotGiven = not_given ,
71- ) -> GenerationCreateResponse :
71+ ) -> ImageGenerateResponse :
7272 """
7373 Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
7474 image generation model with automatic prompt optimization and enhanced visual
@@ -126,7 +126,7 @@ def create(
126126 ...
127127
128128 @overload
129- def create (
129+ def generate (
130130 self ,
131131 * ,
132132 prompt : str ,
@@ -205,7 +205,7 @@ def create(
205205 ...
206206
207207 @overload
208- def create (
208+ def generate (
209209 self ,
210210 * ,
211211 prompt : str ,
@@ -226,7 +226,7 @@ def create(
226226 extra_query : Query | None = None ,
227227 extra_body : Body | None = None ,
228228 timeout : float | httpx .Timeout | None | NotGiven = not_given ,
229- ) -> GenerationCreateResponse | Stream [ImageGenStreamEvent ]:
229+ ) -> ImageGenerateResponse | Stream [ImageGenStreamEvent ]:
230230 """
231231 Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
232232 image generation model with automatic prompt optimization and enhanced visual
@@ -284,7 +284,7 @@ def create(
284284 ...
285285
286286 @required_args (["prompt" ], ["prompt" , "stream" ])
287- def create (
287+ def generate (
288288 self ,
289289 * ,
290290 prompt : str ,
@@ -335,41 +335,41 @@ def create(
335335 "stream" : stream ,
336336 "user" : user ,
337337 },
338- generation_create_params . GenerationCreateParamsStreaming
338+ image_generate_params . ImageGenerateParamsStreaming
339339 if stream
340- else generation_create_params . GenerationCreateParamsNonStreaming ,
340+ else image_generate_params . ImageGenerateParamsNonStreaming ,
341341 ),
342342 options = make_request_options (
343343 extra_headers = headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
344344 ),
345- cast_to = GenerationCreateResponse ,
345+ cast_to = ImageGenerateResponse ,
346346 stream = stream or False ,
347347 stream_cls = Stream [ImageGenStreamEvent ],
348348 )
349349
350350
351- class AsyncGenerationsResource (AsyncAPIResource ):
351+ class AsyncImagesResource (AsyncAPIResource ):
352352 @cached_property
353- def with_raw_response (self ) -> AsyncGenerationsResourceWithRawResponse :
353+ def with_raw_response (self ) -> AsyncImagesResourceWithRawResponse :
354354 """
355355 This property can be used as a prefix for any HTTP method call to return
356356 the raw response object instead of the parsed content.
357357
358358 For more information, see https://www.github.com/digitalocean/gradient-python#accessing-raw-response-data-eg-headers
359359 """
360- return AsyncGenerationsResourceWithRawResponse (self )
360+ return AsyncImagesResourceWithRawResponse (self )
361361
362362 @cached_property
363- def with_streaming_response (self ) -> AsyncGenerationsResourceWithStreamingResponse :
363+ def with_streaming_response (self ) -> AsyncImagesResourceWithStreamingResponse :
364364 """
365365 An alternative to `.with_raw_response` that doesn't eagerly read the response body.
366366
367367 For more information, see https://www.github.com/digitalocean/gradient-python#with_streaming_response
368368 """
369- return AsyncGenerationsResourceWithStreamingResponse (self )
369+ return AsyncImagesResourceWithStreamingResponse (self )
370370
371371 @overload
372- async def create (
372+ async def generate (
373373 self ,
374374 * ,
375375 prompt : str ,
@@ -390,7 +390,7 @@ async def create(
390390 extra_query : Query | None = None ,
391391 extra_body : Body | None = None ,
392392 timeout : float | httpx .Timeout | None | NotGiven = not_given ,
393- ) -> GenerationCreateResponse :
393+ ) -> ImageGenerateResponse :
394394 """
395395 Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
396396 image generation model with automatic prompt optimization and enhanced visual
@@ -448,7 +448,7 @@ async def create(
448448 ...
449449
450450 @overload
451- async def create (
451+ async def generate (
452452 self ,
453453 * ,
454454 prompt : str ,
@@ -527,7 +527,7 @@ async def create(
527527 ...
528528
529529 @overload
530- async def create (
530+ async def generate (
531531 self ,
532532 * ,
533533 prompt : str ,
@@ -548,7 +548,7 @@ async def create(
548548 extra_query : Query | None = None ,
549549 extra_body : Body | None = None ,
550550 timeout : float | httpx .Timeout | None | NotGiven = not_given ,
551- ) -> GenerationCreateResponse | AsyncStream [ImageGenStreamEvent ]:
551+ ) -> ImageGenerateResponse | AsyncStream [ImageGenStreamEvent ]:
552552 """
553553 Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
554554 image generation model with automatic prompt optimization and enhanced visual
@@ -606,7 +606,7 @@ async def create(
606606 ...
607607
608608 @required_args (["prompt" ], ["prompt" , "stream" ])
609- async def create (
609+ async def generate (
610610 self ,
611611 * ,
612612 prompt : str ,
@@ -637,7 +637,6 @@ async def create(
637637 "Authorization" : f"Bearer { self ._client .model_access_key } " ,
638638 ** headers ,
639639 }
640-
641640 return await self ._post (
642641 "/images/generations"
643642 if self ._client ._base_url_overridden
@@ -657,50 +656,50 @@ async def create(
657656 "stream" : stream ,
658657 "user" : user ,
659658 },
660- generation_create_params . GenerationCreateParamsStreaming
659+ image_generate_params . ImageGenerateParamsStreaming
661660 if stream
662- else generation_create_params . GenerationCreateParamsNonStreaming ,
661+ else image_generate_params . ImageGenerateParamsNonStreaming ,
663662 ),
664663 options = make_request_options (
665664 extra_headers = headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
666665 ),
667- cast_to = GenerationCreateResponse ,
666+ cast_to = ImageGenerateResponse ,
668667 stream = stream or False ,
669668 stream_cls = AsyncStream [ImageGenStreamEvent ],
670669 )
671670
672671
673- class GenerationsResourceWithRawResponse :
674- def __init__ (self , generations : GenerationsResource ) -> None :
675- self ._generations = generations
672+ class ImagesResourceWithRawResponse :
673+ def __init__ (self , images : ImagesResource ) -> None :
674+ self ._images = images
676675
677- self .create = to_raw_response_wrapper (
678- generations . create ,
676+ self .generate = to_raw_response_wrapper (
677+ images . generate ,
679678 )
680679
681680
682- class AsyncGenerationsResourceWithRawResponse :
683- def __init__ (self , generations : AsyncGenerationsResource ) -> None :
684- self ._generations = generations
681+ class AsyncImagesResourceWithRawResponse :
682+ def __init__ (self , images : AsyncImagesResource ) -> None :
683+ self ._images = images
685684
686- self .create = async_to_raw_response_wrapper (
687- generations . create ,
685+ self .generate = async_to_raw_response_wrapper (
686+ images . generate ,
688687 )
689688
690689
691- class GenerationsResourceWithStreamingResponse :
692- def __init__ (self , generations : GenerationsResource ) -> None :
693- self ._generations = generations
690+ class ImagesResourceWithStreamingResponse :
691+ def __init__ (self , images : ImagesResource ) -> None :
692+ self ._images = images
694693
695- self .create = to_streamed_response_wrapper (
696- generations . create ,
694+ self .generate = to_streamed_response_wrapper (
695+ images . generate ,
697696 )
698697
699698
700- class AsyncGenerationsResourceWithStreamingResponse :
701- def __init__ (self , generations : AsyncGenerationsResource ) -> None :
702- self ._generations = generations
699+ class AsyncImagesResourceWithStreamingResponse :
700+ def __init__ (self , images : AsyncImagesResource ) -> None :
701+ self ._images = images
703702
704- self .create = async_to_streamed_response_wrapper (
705- generations . create ,
703+ self .generate = async_to_streamed_response_wrapper (
704+ images . generate ,
706705 )
0 commit comments