diff --git a/app/billing/backend/supplier_invoices.py b/app/billing/backend/supplier_invoices.py index 6cd5334..d8d4914 100644 --- a/app/billing/backend/supplier_invoices.py +++ b/app/billing/backend/supplier_invoices.py @@ -2329,19 +2329,25 @@ async def reprocess_uploaded_file(file_id: int): # Store AI extracted data in extractions table extraction_id = execute_insert( - """INSERT INTO supplier_invoice_extractions - (file_id, vendor_id, invoice_number, invoice_date, due_date, - total_amount, currency, document_type, confidence, llm_data) - VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s) RETURNING extraction_id""", - (file_id, vendor_id, + """INSERT INTO extractions + (file_id, vendor_matched_id, vendor_name, vendor_cvr, + document_id, document_date, due_date, + total_amount, currency, document_type, document_type_detected, + confidence, llm_response_json, status) + VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s) RETURNING extraction_id""", + (file_id, vendor_id, + llm_result.get('vendor_name'), + llm_result.get('vendor_cvr'), llm_result.get('invoice_number'), llm_result.get('invoice_date'), llm_result.get('due_date'), llm_result.get('total_amount'), llm_result.get('currency', 'DKK'), - llm_result.get('document_type'), + llm_result.get('document_type', 'invoice'), + llm_result.get('document_type', 'invoice'), confidence, - json.dumps(llm_result)) + json.dumps(llm_result), + 'extracted') ) # Insert line items if extracted