npm run format

This commit is contained in:
Taylor Wilsdon
2024-12-18 12:18:31 -05:00
parent 89e86f5e2e
commit 366158ff04
6 changed files with 100 additions and 84 deletions

View File

@@ -951,7 +951,7 @@ async def get_app_config(request: Request):
},
"google_drive": {
"client_id": GOOGLE_DRIVE_CLIENT_ID.value,
"api_key": GOOGLE_DRIVE_API_KEY.value
"api_key": GOOGLE_DRIVE_API_KEY.value,
},
**(
{

View File

@@ -11,7 +11,12 @@ from open_webui.models.knowledge import (
)
from open_webui.models.files import Files, FileModel
from open_webui.retrieval.vector.connector import VECTOR_DB_CLIENT
from open_webui.routers.retrieval import process_file, ProcessFileForm, process_files_batch, BatchProcessFilesForm
from open_webui.routers.retrieval import (
process_file,
ProcessFileForm,
process_files_batch,
BatchProcessFilesForm,
)
from open_webui.constants import ERROR_MESSAGES
@@ -519,6 +524,7 @@ async def reset_knowledge_by_id(id: str, user=Depends(get_verified_user)):
# AddFilesToKnowledge
############################
@router.post("/{id}/files/batch/add", response_model=Optional[KnowledgeFilesResponse])
def add_files_to_knowledge_batch(
id: str,
@@ -555,27 +561,25 @@ def add_files_to_knowledge_batch(
# Process files
try:
result = process_files_batch(BatchProcessFilesForm(
files=files,
collection_name=id
))
except Exception as e:
log.error(f"add_files_to_knowledge_batch: Exception occurred: {e}", exc_info=True)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=str(e)
result = process_files_batch(
BatchProcessFilesForm(files=files, collection_name=id)
)
except Exception as e:
log.error(
f"add_files_to_knowledge_batch: Exception occurred: {e}", exc_info=True
)
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e))
# Add successful files to knowledge base
data = knowledge.data or {}
existing_file_ids = data.get("file_ids", [])
# Only add files that were successfully processed
successful_file_ids = [r.file_id for r in result.results if r.status == "completed"]
for file_id in successful_file_ids:
if file_id not in existing_file_ids:
existing_file_ids.append(file_id)
data["file_ids"] = existing_file_ids
knowledge = Knowledges.update_knowledge_data_by_id(id=id, data=data)
@@ -587,11 +591,10 @@ def add_files_to_knowledge_batch(
files=Files.get_files_by_ids(existing_file_ids),
warnings={
"message": "Some files failed to process",
"errors": error_details
}
"errors": error_details,
},
)
return KnowledgeFilesResponse(
**knowledge.model_dump(),
files=Files.get_files_by_ids(existing_file_ids)
**knowledge.model_dump(), files=Files.get_files_by_ids(existing_file_ids)
)

View File

@@ -37007,16 +37007,14 @@
Pe.createElement('span', { className: 'brace-close' }, '}')
),
pe.size
? pe
.entrySeq()
.map(([s, o]) =>
Pe.createElement(xe, {
key: `${s}-${o}`,
propKey: s,
propVal: o,
propClass: 'property'
})
)
? pe.entrySeq().map(([s, o]) =>
Pe.createElement(xe, {
key: `${s}-${o}`,
propKey: s,
propVal: o,
propClass: 'property'
})
)
: null
);
}
@@ -37167,16 +37165,14 @@
)
: null,
C && z.size
? z
.entrySeq()
.map(([s, o]) =>
Pe.createElement(le, {
key: `${s}-${o}`,
propKey: s,
propVal: o,
propClass: rs
})
)
? z.entrySeq().map(([s, o]) =>
Pe.createElement(le, {
key: `${s}-${o}`,
propKey: s,
propVal: o,
propClass: rs
})
)
: null,
U ? Pe.createElement(ie, { source: U }) : null,
Z &&
@@ -57290,20 +57286,18 @@
Pe.createElement(
'div',
{ className: 'modal-ux-content' },
x
.valueSeq()
.map((x, j) =>
Pe.createElement(C, {
key: j,
AST: w,
definitions: x,
getComponent: i,
errSelectors: u,
authSelectors: s,
authActions: o,
specSelectors: _
})
)
x.valueSeq().map((x, j) =>
Pe.createElement(C, {
key: j,
AST: w,
definitions: x,
getComponent: i,
errSelectors: u,
authSelectors: s,
authActions: o,
specSelectors: _
})
)
)
)
)