Quickstart

pip install thedriveai

from thedriveai import TheDriveAI

client = TheDriveAI(api_key="tda_live_...")

result = client.extract(
    file="invoice.pdf",
    schema={
        "invoice_number": {"type": "string", "description": "The invoice number"},
        "vendor": {"type": "string", "description": "Company name"},
        "total": {"type": "number", "description": "Total amount due"},
        "is_paid": {"type": "boolean", "description": "Whether the invoice is marked as paid"},
    },
)

print(result.data)
# {"invoice_number": "INV-2024-0042", "vendor": "Acme Corp", "total": 1234.56, "is_paid": false}

print(result.confidence)
# {"invoice_number": 0.96, "vendor": 0.94, "total": 0.91, "is_paid": 0.85}

from pydantic import BaseModel, Field
from thedriveai import TheDriveAI

class Invoice(BaseModel):
    invoice_number: str = Field(description="The invoice number")
    vendor: str = Field(description="Company name")
    total: float = Field(description="Total amount due")
    is_paid: bool = Field(description="Whether the invoice is marked as paid")

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

print(result.data["vendor"])  # "Acme Corp"
print(result.data["total"])   # 1234.56

result = client.extract(
    url="https://example.com/pricing",
    schema={
        "plans": {
            "type": "array",
            "description": "Available pricing plans",
            "items": {
                "type": "object",
                "properties": {
                    "name": {"type": "string", "description": "Plan name"},
                    "price": {"type": "number", "description": "Monthly price"},
                }
            }
        }
    },
)

from thedriveai import TheDriveAI, TheDriveAIError

try:
    result = client.extract(file="doc.pdf", schema={...})
except TheDriveAIError as e:
    print(e.status_code)  # 400, 413, 503, etc.
    print(e.detail)       # Error detail from the API