summaryrefslogtreecommitdiffstats
path: root/g4f/providers/create_images.py
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-03-13 15:17:25 +0100
committerGitHub <noreply@github.com>2024-03-13 15:17:25 +0100
commit9c381f29060d697fe2d2c4768cd51b37d98ef3e0 (patch)
tree08a729476060f61f16ac8e909ed7dfe1cf4b2843 /g4f/providers/create_images.py
parentMerge pull request #1680 from Zero6992/main (diff)
parentUpdate Bing.py (diff)
downloadgpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.tar
gpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.tar.gz
gpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.tar.bz2
gpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.tar.lz
gpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.tar.xz
gpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.tar.zst
gpt4free-9c381f29060d697fe2d2c4768cd51b37d98ef3e0.zip
Diffstat (limited to '')
-rw-r--r--g4f/providers/create_images.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/g4f/providers/create_images.py b/g4f/providers/create_images.py
index 29a2a041..29db9435 100644
--- a/g4f/providers/create_images.py
+++ b/g4f/providers/create_images.py
@@ -6,6 +6,7 @@ import asyncio
from .. import debug
from ..typing import CreateResult, Messages
from .types import BaseProvider, ProviderType
+from ..image import ImageResponse
system_message = """
You can generate images, pictures, photos or img with the DALL-E 3 image generator.
@@ -92,7 +93,9 @@ class CreateImagesProvider(BaseProvider):
messages.insert(0, {"role": "system", "content": self.system_message})
buffer = ""
for chunk in self.provider.create_completion(model, messages, stream, **kwargs):
- if isinstance(chunk, str) and buffer or "<" in chunk:
+ if isinstance(chunk, ImageResponse):
+ yield chunk
+ elif isinstance(chunk, str) and buffer or "<" in chunk:
buffer += chunk
if ">" in buffer:
match = re.search(r'<img data-prompt="(.*?)">', buffer)