github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-17 18:23:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: (issue #13825) Getting an error with DallEAPIWrapper (#13874)

Browse Source 
- **Description:** As of OpenAI's Python package 1.0, the existing
DallEAPIWrapper does not work correctly, so the example in the LangChain
Documentation link below does not work either.

https://python.langchain.com/docs/integrations/tools/dalle_image_generator
Also, since OpenAI only supports DALL-E version 2 or version 3, I
modified the DallEAPIWrapper to support it.

  - **Issue:** #13825 

  - **Twitter handle:** ggeutzzang
This commit is contained in:
ggeutzzang 2023-11-29 12:31:25 +09:00 committed by GitHub
parent 74045bf5c0
commit 981f78f920
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
libs/langchain/langchain/utilities/dalle_image_generator.py
View File

@ -25,8 +25,10 @@ class DallEAPIWrapper(BaseModel):
"""Size of image to generate""" """Size of image to generate"""
separator: str = "\n" separator: str = "\n"
"""Separator to use when multiple URLs are returned.""" """Separator to use when multiple URLs are returned."""
model: Optional[str] = None model: Optional[str] = "dall-e-2"
"""Model to use for image generation.""" """Model to use for image generation"""
quality: Optional[str] = "standard"
"""Quality of the image that will be generated"""
class Config: class Config:
"""Configuration for this pydantic object.""" """Configuration for this pydantic object."""
@ -40,10 +42,13 @@ class DallEAPIWrapper(BaseModel):
values, "openai_api_key", "OPENAI_API_KEY" values, "openai_api_key", "OPENAI_API_KEY"
) )
try: try:
import openai from openai import OpenAI
openai.api_key = openai_api_key client = OpenAI(
values["client"] = openai.Image api_key=openai_api_key, # this is also the default, it can be omitted
)
values["client"] = client
except ImportError as e: except ImportError as e:
raise ImportError( raise ImportError(
"Could not import openai python package. " "Could not import openai python package. "
@ -53,8 +58,12 @@ class DallEAPIWrapper(BaseModel):
def run(self, query: str) -> str: def run(self, query: str) -> str:
"""Run query through OpenAI and parse result.""" """Run query through OpenAI and parse result."""
response = self.client.create( response = self.client.images.generate(
prompt=query, n=self.n, size=self.size, model=self.model prompt=query,
n=self.n,
size=self.size,
model=self.model,
quality=self.quality,
) )
image_urls = self.separator.join([item["url"] for item in response["data"]]) image_urls = self.separator.join([item.url for item in response.data])
return image_urls if image_urls else "No image was generated" return image_urls if image_urls else "No image was generated"