File tree Expand file tree Collapse file tree 1 file changed +10
-9
lines changed Expand file tree Collapse file tree 1 file changed +10
-9
lines changed Original file line number Diff line number Diff line change @@ -62,7 +62,7 @@ class DallEAPIWrapper(BaseModel):
62
62
"""Size of image to generate"""
63
63
separator : str = "\n "
64
64
"""Separator to use when multiple URLs are returned."""
65
- quality : Optional [str ] = "standard"
65
+ quality : Optional [str ] = None
66
66
"""Quality of the image that will be generated"""
67
67
max_retries : int = 2
68
68
"""Maximum number of retries to make when generating."""
@@ -140,15 +140,16 @@ def validate_environment(self) -> Self:
140
140
141
141
def run (self , query : str ) -> str :
142
142
"""Run query through OpenAI and parse result."""
143
-
144
143
if is_openai_v1 ():
145
- response = self .client .generate (
146
- prompt = query ,
147
- n = self .n ,
148
- size = self .size ,
149
- model = self .model_name ,
150
- quality = self .quality ,
151
- )
144
+ kwargs = {
145
+ "prompt" : query ,
146
+ "n" : self .n ,
147
+ "size" : self .size ,
148
+ "model" : self .model_name ,
149
+ }
150
+ if self .quality is not None :
151
+ kwargs ["quality" ] = self .quality
152
+ response = self .client .generate (** kwargs )
152
153
image_urls = self .separator .join ([item .url for item in response .data ])
153
154
else :
154
155
response = self .client .create (
You can’t perform that action at this time.
0 commit comments