Generate a fitting schema from documents automatically

• supplier_address
• quotation_number
• due_date
• line_items
• description
• quantity
• unit_price
• amount

[
  { "name": "supplier_address",
    "type": "address" },
  { "name": "quotation_number",
    "type": "string" },
  { "name": "due_date", "type": "date" },
  { "name": "line_items",
    "type": "array",
    "arrayItemType": "object",
    "arrayItemFields": [
      { "name": "description",
        "type": "string" },
      { "name": "quantity",
        "type": "number" },
      { "name": "unit_price",
        "type": "currency" },
      { "name": "amount",
        "type": "currency" }
    ] }
]

import { z } from "zod"

export const extractionSchema = z.object({
  supplier_address: z.object({
    street: z.string().nullable(),
    city: z.string().nullable(),
    // house_number, postal_code, …
  }).nullable().describe("Postal address"),
  quotation_number: z.string().nullable(),
  due_date: z.string().nullable(),
  line_items: z.array(z.object({
    description: z.string().nullable(),
    quantity: z.number().nullable(),
    unit_price: z.object({
      amount: z.number().nullable(),
      currency: z.enum([…]).nullable()
    }).nullable(),
    // amount: same shape as unit_price
  })).nullable()
})

# Every field defaults to None
class SupplierAddress(BaseModel):
    street: str | None
    city: str | None
    # house_number, postal_code, …

class LineItemsItem(BaseModel):
    description: str | None
    quantity: float | None
    unit_price: UnitPrice | None
    amount: Amount | None

class UnitPrice(BaseModel):
    amount: float | None
    currency: str | None

class Quotation(BaseModel):
    supplier_address: SupplierAddress | None
    quotation_number: str | None
    due_date: str | None
    line_items: list[LineItemsItem] | None

CREATE TABLE IF NOT EXISTS "Quotation" (
  id uuid PRIMARY KEY DEFAULT gen_random_uuid(),
  document_id uuid,
  "supplier_address" jsonb,
  "quotation_number" text,
  "due_date" date,
  "line_items" jsonb,
  created_at timestamptz NOT NULL DEFAULT now()
);