7
7
8
8
import httpx
9
9
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 (
15
16
to_raw_response_wrapper ,
16
17
to_streamed_response_wrapper ,
17
18
async_to_raw_response_wrapper ,
18
19
async_to_streamed_response_wrapper ,
19
20
)
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
25
25
26
- __all__ = ["GenerationsResource " , "AsyncGenerationsResource " ]
26
+ __all__ = ["ImagesResource " , "AsyncImagesResource " ]
27
27
28
28
29
- class GenerationsResource (SyncAPIResource ):
29
+ class ImagesResource (SyncAPIResource ):
30
30
@cached_property
31
- def with_raw_response (self ) -> GenerationsResourceWithRawResponse :
31
+ def with_raw_response (self ) -> ImagesResourceWithRawResponse :
32
32
"""
33
33
This property can be used as a prefix for any HTTP method call to return
34
34
the raw response object instead of the parsed content.
35
35
36
36
For more information, see https://www.github.com/digitalocean/gradient-python#accessing-raw-response-data-eg-headers
37
37
"""
38
- return GenerationsResourceWithRawResponse (self )
38
+ return ImagesResourceWithRawResponse (self )
39
39
40
40
@cached_property
41
- def with_streaming_response (self ) -> GenerationsResourceWithStreamingResponse :
41
+ def with_streaming_response (self ) -> ImagesResourceWithStreamingResponse :
42
42
"""
43
43
An alternative to `.with_raw_response` that doesn't eagerly read the response body.
44
44
45
45
For more information, see https://www.github.com/digitalocean/gradient-python#with_streaming_response
46
46
"""
47
- return GenerationsResourceWithStreamingResponse (self )
47
+ return ImagesResourceWithStreamingResponse (self )
48
48
49
49
@overload
50
- def create (
50
+ def generate (
51
51
self ,
52
52
* ,
53
53
prompt : str ,
@@ -68,7 +68,7 @@ def create(
68
68
extra_query : Query | None = None ,
69
69
extra_body : Body | None = None ,
70
70
timeout : float | httpx .Timeout | None | NotGiven = not_given ,
71
- ) -> GenerationCreateResponse :
71
+ ) -> ImageGenerateResponse :
72
72
"""
73
73
Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
74
74
image generation model with automatic prompt optimization and enhanced visual
@@ -126,7 +126,7 @@ def create(
126
126
...
127
127
128
128
@overload
129
- def create (
129
+ def generate (
130
130
self ,
131
131
* ,
132
132
prompt : str ,
@@ -205,7 +205,7 @@ def create(
205
205
...
206
206
207
207
@overload
208
- def create (
208
+ def generate (
209
209
self ,
210
210
* ,
211
211
prompt : str ,
@@ -226,7 +226,7 @@ def create(
226
226
extra_query : Query | None = None ,
227
227
extra_body : Body | None = None ,
228
228
timeout : float | httpx .Timeout | None | NotGiven = not_given ,
229
- ) -> GenerationCreateResponse | Stream [ImageGenStreamEvent ]:
229
+ ) -> ImageGenerateResponse | Stream [ImageGenStreamEvent ]:
230
230
"""
231
231
Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
232
232
image generation model with automatic prompt optimization and enhanced visual
@@ -284,7 +284,7 @@ def create(
284
284
...
285
285
286
286
@required_args (["prompt" ], ["prompt" , "stream" ])
287
- def create (
287
+ def generate (
288
288
self ,
289
289
* ,
290
290
prompt : str ,
@@ -305,7 +305,7 @@ def create(
305
305
extra_query : Query | None = None ,
306
306
extra_body : Body | None = None ,
307
307
timeout : float | httpx .Timeout | None | NotGiven = not_given ,
308
- ) -> GenerationCreateResponse | Stream [ImageGenStreamEvent ]:
308
+ ) -> ImageGenerateResponse | Stream [ImageGenStreamEvent ]:
309
309
if not self ._client .model_access_key :
310
310
raise TypeError (
311
311
"Could not resolve authentication method. Expected model_access_key to be set for chat completions."
@@ -335,41 +335,41 @@ def create(
335
335
"stream" : stream ,
336
336
"user" : user ,
337
337
},
338
- generation_create_params . GenerationCreateParamsStreaming
338
+ image_generate_params . ImageGenerateParamsStreaming
339
339
if stream
340
- else generation_create_params . GenerationCreateParamsNonStreaming ,
340
+ else image_generate_params . ImageGenerateParamsNonStreaming ,
341
341
),
342
342
options = make_request_options (
343
343
extra_headers = headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
344
344
),
345
- cast_to = GenerationCreateResponse ,
345
+ cast_to = ImageGenerateResponse ,
346
346
stream = stream or False ,
347
347
stream_cls = Stream [ImageGenStreamEvent ],
348
348
)
349
349
350
350
351
- class AsyncGenerationsResource (AsyncAPIResource ):
351
+ class AsyncImagesResource (AsyncAPIResource ):
352
352
@cached_property
353
- def with_raw_response (self ) -> AsyncGenerationsResourceWithRawResponse :
353
+ def with_raw_response (self ) -> AsyncImagesResourceWithRawResponse :
354
354
"""
355
355
This property can be used as a prefix for any HTTP method call to return
356
356
the raw response object instead of the parsed content.
357
357
358
358
For more information, see https://www.github.com/digitalocean/gradient-python#accessing-raw-response-data-eg-headers
359
359
"""
360
- return AsyncGenerationsResourceWithRawResponse (self )
360
+ return AsyncImagesResourceWithRawResponse (self )
361
361
362
362
@cached_property
363
- def with_streaming_response (self ) -> AsyncGenerationsResourceWithStreamingResponse :
363
+ def with_streaming_response (self ) -> AsyncImagesResourceWithStreamingResponse :
364
364
"""
365
365
An alternative to `.with_raw_response` that doesn't eagerly read the response body.
366
366
367
367
For more information, see https://www.github.com/digitalocean/gradient-python#with_streaming_response
368
368
"""
369
- return AsyncGenerationsResourceWithStreamingResponse (self )
369
+ return AsyncImagesResourceWithStreamingResponse (self )
370
370
371
371
@overload
372
- async def create (
372
+ async def generate (
373
373
self ,
374
374
* ,
375
375
prompt : str ,
@@ -390,7 +390,7 @@ async def create(
390
390
extra_query : Query | None = None ,
391
391
extra_body : Body | None = None ,
392
392
timeout : float | httpx .Timeout | None | NotGiven = not_given ,
393
- ) -> GenerationCreateResponse :
393
+ ) -> ImageGenerateResponse :
394
394
"""
395
395
Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
396
396
image generation model with automatic prompt optimization and enhanced visual
@@ -448,7 +448,7 @@ async def create(
448
448
...
449
449
450
450
@overload
451
- async def create (
451
+ async def generate (
452
452
self ,
453
453
* ,
454
454
prompt : str ,
@@ -527,7 +527,7 @@ async def create(
527
527
...
528
528
529
529
@overload
530
- async def create (
530
+ async def generate (
531
531
self ,
532
532
* ,
533
533
prompt : str ,
@@ -548,7 +548,7 @@ async def create(
548
548
extra_query : Query | None = None ,
549
549
extra_body : Body | None = None ,
550
550
timeout : float | httpx .Timeout | None | NotGiven = not_given ,
551
- ) -> GenerationCreateResponse | AsyncStream [ImageGenStreamEvent ]:
551
+ ) -> ImageGenerateResponse | AsyncStream [ImageGenStreamEvent ]:
552
552
"""
553
553
Creates a high-quality image from a text prompt using GPT-IMAGE-1, the latest
554
554
image generation model with automatic prompt optimization and enhanced visual
@@ -606,7 +606,7 @@ async def create(
606
606
...
607
607
608
608
@required_args (["prompt" ], ["prompt" , "stream" ])
609
- async def create (
609
+ async def generate (
610
610
self ,
611
611
* ,
612
612
prompt : str ,
@@ -627,7 +627,7 @@ async def create(
627
627
extra_query : Query | None = None ,
628
628
extra_body : Body | None = None ,
629
629
timeout : float | httpx .Timeout | None | NotGiven = not_given ,
630
- ) -> GenerationCreateResponse | AsyncStream [ImageGenStreamEvent ]:
630
+ ) -> ImageGenerateResponse | AsyncStream [ImageGenStreamEvent ]:
631
631
if not self ._client .model_access_key :
632
632
raise TypeError (
633
633
"Could not resolve authentication method. Expected model_access_key to be set for chat completions."
@@ -637,7 +637,6 @@ async def create(
637
637
"Authorization" : f"Bearer { self ._client .model_access_key } " ,
638
638
** headers ,
639
639
}
640
-
641
640
return await self ._post (
642
641
"/images/generations"
643
642
if self ._client ._base_url_overridden
@@ -657,50 +656,50 @@ async def create(
657
656
"stream" : stream ,
658
657
"user" : user ,
659
658
},
660
- generation_create_params . GenerationCreateParamsStreaming
659
+ image_generate_params . ImageGenerateParamsStreaming
661
660
if stream
662
- else generation_create_params . GenerationCreateParamsNonStreaming ,
661
+ else image_generate_params . ImageGenerateParamsNonStreaming ,
663
662
),
664
663
options = make_request_options (
665
664
extra_headers = headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
666
665
),
667
- cast_to = GenerationCreateResponse ,
666
+ cast_to = ImageGenerateResponse ,
668
667
stream = stream or False ,
669
668
stream_cls = AsyncStream [ImageGenStreamEvent ],
670
669
)
671
670
672
671
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
676
675
677
- self .create = to_raw_response_wrapper (
678
- generations . create ,
676
+ self .generate = to_raw_response_wrapper (
677
+ images . generate ,
679
678
)
680
679
681
680
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
685
684
686
- self .create = async_to_raw_response_wrapper (
687
- generations . create ,
685
+ self .generate = async_to_raw_response_wrapper (
686
+ images . generate ,
688
687
)
689
688
690
689
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
694
693
695
- self .create = to_streamed_response_wrapper (
696
- generations . create ,
694
+ self .generate = to_streamed_response_wrapper (
695
+ images . generate ,
697
696
)
698
697
699
698
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
703
702
704
- self .create = async_to_streamed_response_wrapper (
705
- generations . create ,
703
+ self .generate = async_to_streamed_response_wrapper (
704
+ images . generate ,
706
705
)
0 commit comments