Add name_of_asset field to receipt processing and update related logic in DocumentProcessor
This commit is contained in:
+6
-4
@@ -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"),
|
||||
|
||||
Reference in New Issue
Block a user