OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: tasks

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-06-08 20:58:31 +04:00
parent 51fe33395b
commit 0c57980e72
4 changed files with 42 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/utils/middleware.py
View File

@@ -2266,7 +2266,9 @@ async def process_chat_response(
if "data:image/png;base64" in line:
image_url = ""
# Extract base64 image data from the line
image_data, content_type = load_b64_image_data(line)
image_data, content_type = (
load_b64_image_data(line)
)
if image_data is not None:
image_url = upload_image(
request,
@@ -2412,7 +2414,7 @@ async def process_chat_response(
# background_tasks.add_task(post_response_handler, response, events)
task_id, _ = create_task(
post_response_handler(response, events), id=metadata["chat_id"]
request, post_response_handler(response, events), id=metadata["chat_id"]
)
return {"status": True, "task_id": task_id}

9
backend/open_webui/utils/redis.py
View File

@@ -2,6 +2,7 @@ import socketio
import redis
from redis import asyncio as aioredis
from urllib.parse import urlparse
from typing import Optional
def parse_redis_service_url(redis_url):
@@ -18,7 +19,9 @@ def parse_redis_service_url(redis_url):
}
def get_redis_connection(redis_url, redis_sentinels, decode_responses=True):
def get_redis_connection(
redis_url, redis_sentinels, decode_responses=True
) -> Optional[redis.Redis]:
if redis_sentinels:
redis_config = parse_redis_service_url(redis_url)
sentinel = redis.sentinel.Sentinel(
@@ -32,9 +35,11 @@ def get_redis_connection(redis_url, redis_sentinels, decode_responses=True):
# Get a master connection from Sentinel
return sentinel.master_for(redis_config["service"])
else:
elif redis_url:
# Standard Redis connection
return redis.Redis.from_url(redis_url, decode_responses=decode_responses)
else:
return None
def get_sentinels_from_env(sentinel_hosts_env, sentinel_port_env):