Models and tools

type Model = {
  complete(input: ModelInput): Promise<ModelOutput>;
  stream(input: ModelInput): AsyncIterable<ModelToken>;
};

import type { Model } from "stateweave";

export const model: Model = {
  async complete(input) {
    const response = await provider.generate({
      system: input.system,
      prompt: input.prompt,
      signal: input.signal
    });

    return {
      text: response.text,
      usage: response.usage
    };
  },

  async *stream(input) {
    yield { type: "token", token: (await this.complete(input)).text };
  }
};

import { AnthropicModel, createModelFromEnv, MockModel } from "stateweave";

type Tool = {
  name: string;
  description: string;
  schema: z.ZodSchema;
  execute(args: unknown): Promise<unknown>;
};

import { z } from "zod";
import type { Tool } from "stateweave";

const orderArgs = z.object({
  orderId: z.string().regex(/^ord_[a-z0-9]+$/)
});

export const lookupOrder: Tool = {
  name: "lookup_order",
  description: "Look up one order by its ord_ identifier.",
  schema: orderArgs,
  async execute(args) {
    const { orderId } = orderArgs.parse(args);
    const order = await orders.findById(orderId);
    return order
      ? { ok: true, orderId, status: order.status }
      : { ok: false, orderId, error: "not_found" };
  }
};

const agent = new Agent({ model, tools: [lookupOrder] });

import { createFileSystemTools } from "stateweave";

const tools = createFileSystemTools({
  rootDir: "/srv/sandboxes/run-123",
  timeoutMs: 5_000,
  maxOutputBytes: 32_000,
  allowedBashCommands: ["pwd", "ls", "rg", "node"]
});

return {
  ok: true,
  resource: "order:ord_123",
  version: "7",
  status: "shipped"
};