Merge pull request #1444 from hlohaus/async

Bing
pull/1448/head 0.1.9.11
H Lohaus 6 months ago committed by GitHub
commit 517895845e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -249,7 +249,7 @@ async def stream_generate(
cookies: dict = None, cookies: dict = None,
web_search: bool = False, web_search: bool = False,
gpt4_turbo: bool = False, gpt4_turbo: bool = False,
timeout = int = 900 timeout: int = 900
): ):
headers = Defaults.headers headers = Defaults.headers
if cookies: if cookies:

@ -71,14 +71,17 @@ class CreateImagesProvider(BaseProvider):
) -> str: ) -> str:
messages.insert(0, {"role": "system", "content": self.system_message}) messages.insert(0, {"role": "system", "content": self.system_message})
response = await self.provider.create_async(model, messages, **kwargs) response = await self.provider.create_async(model, messages, **kwargs)
matches = re.findall(r'(<img data-prompt="(.*?)">)', result) matches = re.findall(r'(<img data-prompt="(.*?)">)', response)
results = [] results = []
for _, prompt in matches: placeholders = []
for placeholder, prompt in matches:
if placeholder not in placeholders:
results.append(self.create_images_async(prompt)) results.append(self.create_images_async(prompt))
placeholders.append(placeholder)
results = await asyncio.gather(*results) results = await asyncio.gather(*results)
for idx, result in enumerate(results): for idx, result in enumerate(results):
placeholder = matches[idx][0] placeholder = placeholder[idx]
if self.include_placeholder: if self.include_placeholder:
result = placeholder + result result = placeholder + result
response = response.replace(placeholder, result) response = response.replace(placeholder, result)
return result return response
Loading…
Cancel
Save