Type Safety

---
name: math-addition
metadata:
  model:
    name: gpt-4o
    settings:
      schema:  # Output type definition
        type: "object"
        properties:
          sum:
            type: "number"
            description: "The sum of the two numbers"
        required: ["sum"]
input_schema:  # Input type definition
  type: "object"
  properties:
    num1:
      type: "number"
      description: "First number to add"
    num2:
      type: "number"
      description: "Second number to add"
  required: ["num1", "num2"]
---

<System>You are a helpful math assistant that performs addition.</System>
<User>What is the sum of {props.num1} and {props.num2}?</User>

// Auto-generated types from Puzzlet
// Do not edit this file directly

interface Math$AdditionIn {
  /** First number to add */
  num1: number;
  /** Second number to add */
  num2: number;
}

interface Math$AdditionOut {
  /** The sum of the two numbers */
  sum: number;
  /** Step by step explanation */
  explanation: string;
}

interface Math$Addition {
  input: Math$AdditionIn;
  output: Math$AdditionOut;
}

export default interface PuzzletTypes {
  "math/addition.prompt.mdx": Math$Addition
}

# Generate types from local files
npx @puzzlet/cli generate-types --root-dir ./prompts > agentmark.types.ts

import PromptTypes from './agentmark.types';
import { FileLoader, createTemplateRunner } from "@puzzlet/agentmark";

const fileLoader = new FileLoader<PromptTypes>("./prompts", createTemplateRunner);

// TypeScript enforces correct types
const prompt = await fileLoader.load("math/addition.prompt.mdx");
const result = await prompt.run({
  num1: 5,   // Must be number
  num2: 3    // Must be number
});
const sum = result.result.sum;  // Type-safe number