refac: deprecate docs_dir

This commit is contained in:
Timothy J. Baek
2024-10-04 18:22:55 -07:00
parent a6c797d4c2
commit 79c005a041
6 changed files with 122 additions and 70 deletions

View File

@@ -92,6 +92,7 @@ def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
@router.post("/upload/dir")
def upload_dir(user=Depends(get_admin_user)):
file_ids = []
for path in Path(DOCS_DIR).rglob("./**/*"):
if path.is_file() and not path.name.startswith("."):
try:
@@ -126,13 +127,14 @@ def upload_dir(user=Depends(get_admin_user)):
try:
process_file(ProcessFileForm(file_id=id))
log.debug(f"File processed: {path}, {file.id}")
file_ids.append(file.id)
except Exception as e:
log.exception(e)
log.error(f"Error processing file: {file.id}")
except Exception as e:
log.exception(e)
pass
return True
return file_ids
############################

View File

@@ -160,6 +160,10 @@ def add_file_to_knowledge_by_id(
process_file(ProcessFileForm(file_id=form_data.file_id, collection_name=id))
except Exception as e:
log.debug(e)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=str(e),
)
if knowledge:
data = knowledge.data or {}