scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-10-10 03:27:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2026-04-12 22:08:27 -05:00
parent 26b8ca5b5e
commit d4b90f93bd
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
backend/open_webui/routers/retrieval.py
View file

@ -40,7 +40,7 @@ from open_webui.models.files import FileModel, FileUpdateForm, Files
from open_webui.utils.access_control.files import has_access_to_file
from open_webui.models.knowledge import Knowledges
from open_webui.storage.provider import Storage
from open_webui.internal.db import get_async_session, get_db
from open_webui.internal.db import get_async_db, get_async_session
from sqlalchemy.ext.asyncio import AsyncSession
@ -1694,10 +1694,10 @@ async def process_file(
try:
# Commit any pending changes before the slow embedding step.
# Note: file is already a Pydantic model (not ORM), so no expunge needed.
db.commit()
await db.commit()
# External embedding API takes time (5-60s+).
# Subsequent updates use fresh sessions via get_db().
# Subsequent updates use fresh async sessions.
result = save_docs_to_vector_db(
request,
docs=docs,
@ -1714,7 +1714,7 @@ async def process_file(
if result:
# Fresh session for the final update.
with get_db() as session:
async with get_async_db() as session:
await Files.update_file_metadata_by_id(
file.id,
{
@ -1744,7 +1744,7 @@ async def process_file(
except Exception as e:
log.exception(e)
# Fresh session for error status update.
with get_db() as session:
async with get_async_db() as session:
await Files.update_file_data_by_id(
file.id,
{'status': 'failed'},