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):
6262 """Size of image to generate"""
6363 separator : str = "\n "
6464 """Separator to use when multiple URLs are returned."""
65- quality : Optional [str ] = "standard"
65+ quality : Optional [str ] = None
6666 """Quality of the image that will be generated"""
6767 max_retries : int = 2
6868 """Maximum number of retries to make when generating."""
@@ -140,15 +140,16 @@ def validate_environment(self) -> Self:
140140
141141 def run (self , query : str ) -> str :
142142 """Run query through OpenAI and parse result."""
143-
144143 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 )
152153 image_urls = self .separator .join ([item .url for item in response .data ])
153154 else :
154155 response = self .client .create (
You can’t perform that action at this time.
0 commit comments