ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Merge pull request #15490 from sihyeonn/chore/sh-print

Browse source 
refactor: improve logging in task management system
This commit is contained in:
Tim Jaeryang Baek 2025-07-03 17:14:35 +04:00 committed by GitHub
commit 74167f4b48
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
backend/open_webui/main.py
View file

@ -1507,7 +1507,7 @@ async def list_tasks_by_chat_id_endpoint(
task_ids = await list_task_ids_by_chat_id(request, chat_id) task_ids = await list_task_ids_by_chat_id(request, chat_id)
print(f"Task IDs for chat {chat_id}: {task_ids}") log.debug(f"Task IDs for chat {chat_id}: {task_ids}")
return {"task_ids": task_ids} return {"task_ids": task_ids}

9
backend/open_webui/tasks.py
View file

@ -3,10 +3,17 @@ import asyncio
from typing import Dict from typing import Dict
from uuid import uuid4 from uuid import uuid4
import json import json
import logging
from redis.asyncio import Redis from redis.asyncio import Redis
from fastapi import Request from fastapi import Request
from typing import Dict, List, Optional from typing import Dict, List, Optional
from open_webui.env import SRC_LOG_LEVELS
log = logging.getLogger(__name__)
log.setLevel(SRC_LOG_LEVELS["MAIN"])
# A dictionary to keep track of active tasks # A dictionary to keep track of active tasks
tasks: Dict[str, asyncio.Task] = {} tasks: Dict[str, asyncio.Task] = {}
chat_tasks = {} chat_tasks = {}
@ -38,7 +45,7 @@ async def redis_task_command_listener(app):
if local_task: if local_task:
local_task.cancel() local_task.cancel()
except Exception as e: except Exception as e:
print(f"Error handling distributed task command: {e}") log.exception(f"Error handling distributed task command: {e}")
### ------------------------------ ### ------------------------------