UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 16:51:17 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: backend format test

Browse Source
This commit is contained in:
Michael Poluektov 2024-08-08 12:41:41 +01:00
parent 309cd645f1
commit fa4d1d42a5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/apps/images/utils/comfyui.py
View File

@ -1,6 +1,5 @@
import asyncio import asyncio
import websocket # NOTE: websocket-client (https://github.com/websocket-client/websocket-client) import websocket # NOTE: websocket-client (https://github.com/websocket-client/websocket-client)
import uuid
import json import json
import urllib.request import urllib.request
import urllib.parse import urllib.parse
@ -398,7 +397,9 @@ async def comfyui_generate_image(
return None return None
try: try:
images = await asyncio.to_thread(get_images, ws, comfyui_prompt, client_id, base_url) images = await asyncio.to_thread(
get_images, ws, comfyui_prompt, client_id, base_url
)
except Exception as e: except Exception as e:
log.exception(f"Error while receiving images: {e}") log.exception(f"Error while receiving images: {e}")
images = None images = None