feat: doc upload error handling

This commit is contained in:
Timothy J. Baek 2024-01-08 01:54:03 -08:00
parent eddb6fc7b7
commit 1ff3eb8038
3 changed files with 13 additions and 5 deletions

View file

@ -49,13 +49,13 @@ async def create_new_doc(form_data: DocumentForm, user=Depends(get_current_user)
return doc return doc
else: else:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT(), detail=ERROR_MESSAGES.FILE_EXISTS,
) )
else: else:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.COMMAND_TAKEN, detail=ERROR_MESSAGES.NAME_TAG_TAKEN,
) )

View file

@ -18,6 +18,8 @@ class ERROR_MESSAGES(str, Enum):
"Uh-oh! This username is already registered. Please choose another username." "Uh-oh! This username is already registered. Please choose another username."
) )
COMMAND_TAKEN = "Uh-oh! This command is already registered. Please choose another command string." COMMAND_TAKEN = "Uh-oh! This command is already registered. Please choose another command string."
FILE_EXISTS = "Uh-oh! This file is already registered. Please choose another file."
NAME_TAG_TAKEN = "Uh-oh! This name tag is already registered. Please choose another name tag string." NAME_TAG_TAKEN = "Uh-oh! This name tag is already registered. Please choose another name tag string."
INVALID_TOKEN = ( INVALID_TOKEN = (
"Your session has expired or the token is invalid. Please sign in again." "Your session has expired or the token is invalid. Please sign in again."

View file

@ -29,7 +29,10 @@
}; };
const uploadDoc = async (file) => { const uploadDoc = async (file) => {
const res = await uploadDocToVectorDB(localStorage.token, '', file); const res = await uploadDocToVectorDB(localStorage.token, '', file).catch((error) => {
toast.error(error);
return null;
});
if (res) { if (res) {
await createNewDoc( await createNewDoc(
@ -38,7 +41,10 @@
res.filename, res.filename,
transformFileName(res.filename), transformFileName(res.filename),
res.filename res.filename
); ).catch((error) => {
toast.error(error);
return null;
});
await documents.set(await getDocs(localStorage.token)); await documents.set(await getDocs(localStorage.token));
} }
}; };