mirror of
https://github.com/open-webui/open-webui
synced 2025-06-26 18:26:48 +00:00
Fix: Adjusted to handle individual rag config by passing knowledge id
This commit is contained in:
parent
8606c7ecf6
commit
5598b28fae
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user