Fix: Adjusted to handle individual rag config by passing knowledge id

This commit is contained in:
weberm1 2025-05-23 10:40:47 +02:00
parent 8606c7ecf6
commit 5598b28fae

View File

@ -15,6 +15,7 @@ from fastapi import (
UploadFile, UploadFile,
status, status,
Query, Query,
Form
) )
from fastapi.responses import FileResponse, StreamingResponse from fastapi.responses import FileResponse, StreamingResponse
from open_webui.constants import ERROR_MESSAGES from open_webui.constants import ERROR_MESSAGES
@ -87,6 +88,7 @@ def upload_file(
user=Depends(get_verified_user), user=Depends(get_verified_user),
file_metadata: dict = None, file_metadata: dict = None,
process: bool = Query(True), process: bool = Query(True),
knowledge_id: Optional[str] = Form(None)
): ):
log.info(f"file.content_type: {file.content_type}") log.info(f"file.content_type: {file.content_type}")
@ -148,7 +150,7 @@ def upload_file(
process_file( process_file(
request, request,
ProcessFileForm(file_id=id, content=result.get("text", "")), ProcessFileForm(file_id=id, content=result.get("text", ""), knowledge_id=knowledge_id),
user=user, user=user,
) )
elif file.content_type not in [ elif file.content_type not in [
@ -159,12 +161,12 @@ def upload_file(
"video/ogg", "video/ogg",
"video/quicktime", "video/quicktime",
]: ]:
process_file(request, ProcessFileForm(file_id=id), user=user) process_file(request, ProcessFileForm(file_id=id, knowledge_id=knowledge_id), user=user)
else: else:
log.info( log.info(
f"File type {file.content_type} is not provided, but trying to process anyway" f"File type {file.content_type} is not provided, but trying to process anyway"
) )
process_file(request, ProcessFileForm(file_id=id), user=user) process_file(request, ProcessFileForm(file_id=id, knowledge_id=knowledge_id), user=user)
file_item = Files.get_file_by_id(id=id) file_item = Files.get_file_by_id(id=id)
except Exception as e: except Exception as e: