scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-09-08 22:21:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Merge 092351d76d into 56d296ef1b

Browse source
This commit is contained in:
G30 2026-08-26 09:02:46 +03:00 • committed by GitHub
commit 1731fbdc26
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 41 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
backend/open_webui/routers/knowledge.py
View file

@ -1541,24 +1541,39 @@ async def update_file_from_knowledge_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
# Validate the file actually belongs to this knowledge base
if not await Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.NOT_FOUND,
)
is_linked = await Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db)
# Remove content from the vector database
await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': form_data.file_id})
# Add content to the vector database
try:
await process_file(
request,
ProcessFileForm(file_id=form_data.file_id, collection_name=id),
user=user,
db=db,
)
if is_linked:
await ASYNC_VECTOR_DB_CLIENT.delete(
collection_name=knowledge.id, filter={'file_id': form_data.file_id}
)
await process_file(
request,
ProcessFileForm(file_id=form_data.file_id, collection_name=id),
user=user,
db=db,
)
else:
await process_file(
request,
ProcessFileForm(file_id=form_data.file_id),
user=user,
db=db,
)
await process_file(
request,
ProcessFileForm(file_id=form_data.file_id, collection_name=id),
user=user,
db=db,
)
await Knowledges.add_file_to_knowledge_by_id(
knowledge_id=id,
file_id=form_data.file_id,
user_id=user.id,
directory_id=form_data.directory_id,
db=db,
)
except Exception as e:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@ -1631,14 +1646,8 @@ async def remove_file_from_knowledge_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
# Validate the file actually belongs to this knowledge base
if not await Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.NOT_FOUND,
)
await Knowledges.remove_file_from_knowledge_by_id(knowledge_id=id, file_id=form_data.file_id, db=db)
if await Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
await Knowledges.remove_file_from_knowledge_by_id(knowledge_id=id, file_id=form_data.file_id, db=db)
# Remove content from the vector database
try:

10
src/lib/components/workspace/Knowledge/KnowledgeBase.svelte
View file

@ -137,6 +137,7 @@
let deleteDirectoryContents = true;
let pendingPollTimer: ReturnType<typeof setInterval> | null = null;
let pendingPollCount = 0;
let externalTestQuery = '';
let externalTestResult: {
documents?: string[];
@ -226,15 +227,20 @@
// Start polling for completion (if not already polling)
if (!pendingPollTimer) {
pendingPollCount = 0;
pendingPollTimer = setInterval(async () => {
pendingPollCount++;
try {
const still = await getPendingKnowledgeFiles(localStorage.token, knowledgeId);
if (!still || still.length === 0) {
if (!still || still.length === 0 || pendingPollCount >= 60) {
clearInterval(pendingPollTimer);
pendingPollTimer = null;
init();
}
} catch {}
} catch {
clearInterval(pendingPollTimer);
pendingPollTimer = null;
}
}, 5000);
}
}