Add name_of_asset field to receipt processing and update related logic in DocumentProcessor

This commit is contained in:
bolade
2025-10-07 20:35:43 +01:00
parent 5116fb5efb
commit f582110674
4 changed files with 23 additions and 6 deletions
+6 -4
View File
@@ -369,7 +369,9 @@ async def upload_multiple_documents(
response_model=DocumentProcessResponse,
tags=["Document Processing"],
)
async def process_document(file_id: str, request: DocumentProcessRequest, db: db_dependency):
async def process_document(
file_id: str, request: DocumentProcessRequest, db: db_dependency
):
"""
Process a previously uploaded document to extract receipt information.
@@ -383,9 +385,7 @@ async def process_document(file_id: str, request: DocumentProcessRequest, db: db
# Get file info from database
db_uploaded_file = get_uploaded_file_from_db(db, file_id)
if not db_uploaded_file:
raise HTTPException(
status_code=404, detail=f"File {file_id} not found"
)
raise HTTPException(status_code=404, detail=f"File {file_id} not found")
# Process the file using the stored file path
receipt_data = await document_processor.process_file(
@@ -423,6 +423,7 @@ async def process_document(file_id: str, request: DocumentProcessRequest, db: db
is_depreciable=str(receipt_data.get("is_depreciable"))
if receipt_data.get("is_depreciable") is not None
else None,
name_of_asset=receipt_data.get("name_of_asset"),
cca_rate=receipt_data.get("cca_rate"),
useful_life=receipt_data.get("useful_life"),
residual_value=receipt_data.get("residual_value"),
@@ -448,6 +449,7 @@ async def process_document(file_id: str, request: DocumentProcessRequest, db: db
receipt_location=receipt_data.get("location"),
calculated_tax=receipt_data.get("calculated_tax"),
is_depreciable=receipt_data.get("is_depreciable"),
name_of_asset=receipt_data.get("name_of_asset"),
cca_rate=receipt_data.get("cca_rate"),
useful_life=receipt_data.get("useful_life"),
residual_value=receipt_data.get("residual_value"),