Extract, classify, and summarize documents at scale

import { Stack0 } from '@stack0/sdk'
 
const stack0 = new Stack0({ apiKey: process.env.STACK0_API_KEY })
 
// Process a document through extraction, classification, and summarization
const result = await stack0.workflows.run({
  steps: [
    {
      id: 'parse',
      type: 'tool',
      tool: 'document.parse',
      input: {
        url: '{{input.documentUrl}}',
        ocr: true, // Enable OCR for scanned documents
      },
    },
    {
      id: 'classify',
      type: 'llm',
      model: 'gpt-4o-mini',
      prompt: 'Classify this document into one of: invoice, contract, report, letter, form. Document: {{steps.parse.output.text}}',
      outputSchema: {
        category: 'string',
        confidence: 'number',
      },
      dependsOn: ['parse'],
    },
    {
      id: 'extract',
      type: 'llm',
      model: 'gpt-4o',
      prompt: 'Extract key entities from this {{steps.classify.output.category}}: {{steps.parse.output.text}}',
      outputSchema: {
        entities: '{ name: string, type: string, value: string }[]',
        dates: 'string[]',
        amounts: '{ value: number, currency: string }[]',
      },
      dependsOn: ['parse', 'classify'],
    },
    {
      id: 'summarize',
      type: 'llm',
      model: 'gpt-4o-mini',
      prompt: 'Summarize this document in 2-3 sentences: {{steps.parse.output.text}}',
      dependsOn: ['parse'],
    },
  ],
  input: {
    documentUrl: 'https://example.com/invoice-2024-001.pdf',
  },
})
 
console.log(result.steps.classify.output) // { category: 'invoice', confidence: 0.97 }
console.log(result.steps.extract.output) // { entities: [...], dates: [...], amounts: [...] }
console.log(result.steps.summarize.output) // "This invoice from Acme Corp..."