Schemas

{
  "vendor": {"type": "string", "description": "Company name on the invoice"},
  "total": {"type": "number", "description": "Total amount due including tax"},
  "is_paid": {"type": "boolean", "description": "Whether the invoice is marked as paid"}
}

{
  "status": {
    "type": "string",
    "enum": ["paid", "unpaid", "overdue"],
    "description": "Payment status"
  }
}

{
  "invoice_number": {
    "type": "string",
    "description": "Invoice reference number",
    "required": true
  }
}

{
  "line_items": {
    "type": "array",
    "description": "Each line item on the invoice",
    "items": {
      "type": "object",
      "properties": {
        "description": {"type": "string", "description": "Item description"},
        "quantity": {"type": "integer", "description": "Quantity"},
        "unit_price": {"type": "number", "description": "Price per unit"}
      }
    }
  }
}

{
  "vendor": {
    "type": "object",
    "description": "Vendor information",
    "properties": {
      "name": {"type": "string", "description": "Company name"},
      "address": {"type": "string", "description": "Full address"},
      "tax_id": {"type": "string", "description": "Tax identification number"}
    }
  }
}

from pydantic import BaseModel, Field
from typing import Optional
from enum import Enum

class Status(str, Enum):
    paid = "paid"
    unpaid = "unpaid"
    overdue = "overdue"

class LineItem(BaseModel):
    description: str = Field(description="Item description")
    quantity: int = Field(description="Quantity")
    unit_price: float = Field(description="Price per unit")

class Invoice(BaseModel):
    invoice_number: str = Field(description="The invoice number")
    vendor: str = Field(description="Company name")
    total: float = Field(description="Total amount due")
    status: Status = Field(description="Payment status")
    notes: Optional[str] = Field(default=None, description="Additional notes")
    line_items: list[LineItem] = Field(description="Invoice line items")

result = client.extract(file="invoice.pdf", schema=Invoice)

import { TheDriveAI, fromZod } from "@thedriveai/sdk";
import { z } from "zod";

const LineItem = z.object({
  description: z.string().describe("Item description"),
  quantity: z.number().describe("Quantity"),
  unit_price: z.number().describe("Price per unit"),
});

const Invoice = z.object({
  invoice_number: z.string().describe("The invoice number"),
  vendor: z.string().describe("Company name"),
  total: z.number().describe("Total amount due"),
  status: z.enum(["paid", "unpaid", "overdue"]).describe("Payment status"),
  notes: z.string().optional().describe("Additional notes"),
  line_items: z.array(LineItem).describe("Invoice line items"),
});

const result = await client.extract({
  file: readFileSync("invoice.pdf"),
  schema: fromZod(Invoice),
});