Skip to content

Commit 22e4798

Browse files
fix(issue): DallEAPIWrapper passes unsupported 'quality' param to OpenAI API (#179)
This PR fixes #177 , quality parameter will now only be sent if specifically mentioned. Signed-off-by: Siddarthreddygsr <[email protected]>
1 parent 0544f05 commit 22e4798

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

libs/community/langchain_community/utilities/dalle_image_generator.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff 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(

0 commit comments

Comments
 (0)