playground.ts

import { createHostNodeRuntime } from "./src/host-node/index.ts";
import {
  createClient,
  createProgramContract,
  defineTool,
} from "./src/index.ts";
import { testSchema } from "./src/testing/schema.ts";

async function main() {
  const FlightRequest = testSchema({
    type: "object",
    properties: {
      source: { type: "string", description: "The source airport code." },
      dest: {
        type: "string",
        description: "The destination airport code.",
      },
      departureDate: {
        type: "string",
        description: "The date of the departure.",
        format: "date",
      },
      returnDate: {
        type: "string",
        description: "The date of the return.",
        format: "date",
      },
    },
    required: ["source", "dest", "departureDate", "returnDate"],
    additionalProperties: false,
  } as const);
  const listFlights = defineTool(
    "listFlights",
    {
      description: "List air flights matching a set of criteria.",
      inputSchema: FlightRequest,
      outputSchema: testSchema({
        type: "object",
        properties: {
          flights: {
            type: "array",
            items: {
              type: "object",
              properties: { code: { type: "string" } },
              required: ["code"],
              additionalProperties: false,
            },
          },
        },
        required: ["flights"],
        additionalProperties: false,
      } as const),
    },
    async (_ctx, query) => {
      console.log(query);
      return {
        flights: [],
      };
    },
  );
  const contract = createProgramContract({
    inputSchema: FlightRequest,
    tools: [listFlights],
  });

  const runtime = await createHostNodeRuntime({
    nodePath: process.execPath,
    cwd: process.cwd(),
  }, AbortSignal.timeout(5_000));

  try {
    const client = createClient({
      runtime,
      contract,
    });

    console.log(contract.typeDefinitions);

    const source = `export default async function ({ input, codemode, result }: AgentProgramScope) {
      const response = await codemode.listFlights(input);
      result.appendText(JSON.stringify(response));
    }`;

    console.dir(await client.validate(source, AbortSignal.timeout(5_000)));

    const outcome = await client.run(source, {
      input: {
        source: "YYZ",
        dest: "LHR",
        departureDate: "2026-07-01",
        returnDate: "2026-07-08",
      },
      signal: AbortSignal.timeout(5_000),
    });
    if (outcome.kind === "program-failed" && outcome.error.details !== null) {
      console.dir(outcome.error.details.report);
    }
    if (outcome.kind === "success") {
      console.dir(outcome.content);
    }
  } finally {
    await runtime[Symbol.asyncDispose]();
  }
}

main().catch((e) => {
  console.error(e);
  process.exit(1);
});