Code

{
  "id": "code_1",
  "type": "code",
  "code": "def transform(input_data):\n    total = sum(row.amount or 0 for row in input_data.line_items or [])\n    return Output(total=total)",
  "output_schema": {
    "fields": [{ "name": "total", "data_type": "float", "description": "Sum of the line item amounts." }]
  }
}

import os

from anyformat.sdk import Client
from anyformat.workflow import Schema

client = Client(api_key=os.environ["ANYFORMAT_API_KEY"])

workflow = (
    client.workflow("Invoice total")
    .parse()
    .extract([Schema.float("subtotal", "Subtotal"), Schema.float("tax", "Tax")])
    .code(
        "def transform(input_data):\n    return Output(total=(input_data.subtotal or 0) + (input_data.tax or 0))",
        [Schema.float("total", "Subtotal plus tax")],
    )
    .create()
)

import { Anyformat, Schema } from "@anyformat/sdk";

const af = new Anyformat({ apiKey: process.env.ANYFORMAT_API_KEY! });

const workflow = await af
  .workflow("Invoice total")
  .parse()
  .extract([Schema.float("subtotal", "Subtotal"), Schema.float("tax", "Tax")])
  .code(
    "def transform(input_data):\n    return Output(total=(input_data.subtotal or 0) + (input_data.tax or 0))",
    [Schema.float("total", "Subtotal plus tax")],
  )
  .create();