Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-04-14 17:27:31 -05:00
parent f1be85d997
commit 5dae600ce7
81 changed files with 247 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/main.py
+20 -12
View File
@@ -1707,7 +1707,9 @@ async def chat_completion(
},
'messages': [
{'role': 'user', 'content': user_message.get('content', '')},
] if user_message_id else [],
]
if user_message_id
else [],
'tags': [],
'timestamp': int(time.time() * 1000),
},
@@ -1734,9 +1736,7 @@ async def chat_completion(
pass
else:
# Existing chat — verify ownership
if (
not await Chats.is_chat_owner(chat_id, user.id) and user.role != 'admin'
):
if not await Chats.is_chat_owner(chat_id, user.id) and user.role != 'admin':
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=ERROR_MESSAGES.DEFAULT(),
@@ -1787,16 +1787,16 @@ async def chat_completion(
# Link user message → all assistant messages (childrenIds)
if user_message_id and all_assistant_ids:
existing_user_message = await Chats.get_message_by_id_and_message_id(
chat_id, user_message_id
)
existing_user_message = await Chats.get_message_by_id_and_message_id(chat_id, user_message_id)
if existing_user_message:
child_ids = existing_user_message.get('childrenIds', [])
for assistant_id in all_assistant_ids:
if assistant_id not in child_ids:
child_ids.append(assistant_id)
await Chats.upsert_message_to_chat_by_id_and_message_id(
chat_id, user_message_id, {'childrenIds': child_ids},
chat_id,
user_message_id,
{'childrenIds': child_ids},
)
# Save each assistant placeholder
@@ -1956,11 +1956,19 @@ async def chat_completion(
task_id, _ = await create_task(
request.app.state.redis,
process_chat(
request, model_form_data, user, per_model_metadata, resolved_model,
tasks if idx == 0 else {
k: v for k, v in (tasks or {}).items()
request,
model_form_data,
user,
per_model_metadata,
resolved_model,
tasks
if idx == 0
else {
k: v
for k, v in (tasks or {}).items()
if k not in (TASKS.TITLE_GENERATION, TASKS.TAGS_GENERATION)
} or None,
}
or None,
),
id=chat_id,
)