refac: chat input file handling

This commit is contained in:
Timothy J. Baek
2024-10-26 12:56:37 -07:00
parent 032abba88e
commit 15cfdc69a8
3 changed files with 22 additions and 5 deletions

View File

@@ -73,6 +73,8 @@ class FileModelResponse(BaseModel):
created_at: int # timestamp in epoch
updated_at: int # timestamp in epoch
model_config = ConfigDict(extra="allow")
class FileMetadataResponse(BaseModel):
id: str

View File

@@ -38,7 +38,7 @@ router = APIRouter()
############################
@router.post("/")
@router.post("/", response_model=FileModelResponse)
def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
log.info(f"file.content_type: {file.content_type}")
try:
@@ -73,6 +73,12 @@ def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
except Exception as e:
log.exception(e)
log.error(f"Error processing file: {file_item.id}")
file_item = FileModelResponse(
**{
**file_item.model_dump(),
"error": e,
}
)
if file_item:
return file_item