返回提交历史
Modified
g4f/Provider/create_images.py
+8
-5
XFEstudio/gpt4free
Fix create_async in CreateImagesProvider
4343082d
代码差异
1 个文件
+8
-5
@@ -71,14 +71,17 @@ class CreateImagesProvider(BaseProvider):
71
71
) -> str:
72
72
messages.insert(0, {"role": "system", "content": self.system_message})
73
73
response = await self.provider.create_async(model, messages, **kwargs)
74
matches = re.findall(r'(<img data-prompt="(.*?)">)', result)
74
matches = re.findall(r'(<img data-prompt="(.*?)">)', response)
75
75
results = []
76
for _, prompt in matches:
77
results.append(self.create_images_async(prompt))
76
placeholders = []
77
for placeholder, prompt in matches:
78
if placeholder not in placeholders:
79
results.append(self.create_images_async(prompt))
80
placeholders.append(placeholder)
78
81
results = await asyncio.gather(*results)
79
82
for idx, result in enumerate(results):
80
placeholder = matches[idx][0]
83
placeholder = placeholder[idx]
81
84
if self.include_placeholder:
82
85
result = placeholder + result
83
86
response = response.replace(placeholder, result)
84
return result
87
return response