Data Extraction
Data extraction allows you to parse unstructured text and extract structured data using AI. This is useful for converting free-form text into typed objects that can be used in your application.
The following example shows how to extract product details from an unstructured product description. The AI model will analyze the text and return a structured object containing the product name, summary, price, and category.
amplify/data/resource.ts
const schema = a.schema({ ProductDetails: a.customType({ name: a.string().required(), summary: a.string().required(), price: a.float().required(), category: a.string().required(), }),
extractProductDetails: a.generation({ aiModel: a.ai.model('Claude 3 Haiku'), systemPrompt: 'Extract the property details from the text provided', }) .arguments({ productDescription: a.string() }) .returns(a.ref('ProductDetails')) .authorization((allow) => allow.authenticated()),});
Data Client Request
import type { Schema } from "../amplify/data/resource";import { generateClient } from "aws-amplify/api";
export const client = generateClient<Schema>();
const productDescription = `The NBA Official Game Basketball is a premiumregulation-size basketball crafted with genuine leather and featuringofficial NBA specifications. This professional-grade ball offers superior gripand durability, with deep channels and a moisture-wicking surface that ensuresconsistent performance during intense game play. Priced at $159.99, this high-endbasketball belongs in our Professional Sports Equipment category and is the same modelused in NBA games.`
const { data, errors } = await client.generations .extractProductDetails({ productDescription })
/**Example response:{ "name": "NBA Official Game Basketball", "summary": "Premium regulation-size NBA basketball made with genuine leather. Features official NBA specifications, superior grip, deep channels, and moisture-wicking surface for consistent game play performance.", "price": 159.99, "category": "Professional Sports Equipment"}*/