Build agents

import { defineAgent } from "@keystrokehq/keystroke/agent";
import { lookupCustomer } from "../actions/lookup-customer";

export default defineAgent({
  slug: "support",
  systemPrompt: "Help customers. Use lookupCustomer to find account details.",
  model: "openai/gpt-5.6-luna",
  tools: [lookupCustomer],
});

import { defineAgent } from "@keystrokehq/keystroke/agent";
import { gmailFetchEmails, gmailSendEmail } from "@keystrokehq/gmail/actions";

export default defineAgent({
  slug: "secretary",
  systemPrompt: "Help manage Gmail. Confirm ambiguous sends before acting.",
  model: "anthropic/claude-sonnet-5",
  tools: [gmailFetchEmails, gmailSendEmail],
});

import { defineAgent } from "@keystrokehq/keystroke/agent";
import researcher from "./researcher";

export default defineAgent({
  slug: "orchestrator",
  systemPrompt:
    "You are an orchestrator. For research tasks, call researcher before answering.",
  model: "openai/gpt-5.6-terra",
  tools: [researcher],
});

import { defineAgent } from "@keystrokehq/keystroke/agent";
import refundOrder from "../workflows/refund-order";

export default defineAgent({
  slug: "support",
  systemPrompt:
    "Help customers. To issue a refund, call the refund-order tool with the order ID.",
  model: "xai/grok-4.5",
  tools: [refundOrder],
});

import { defineAgent, defineMcp } from "@keystrokehq/keystroke/agent";

const deepwiki = defineMcp({
  key: "deepwiki",
  transport: { type: "http", url: "https://mcp.deepwiki.com/mcp" },
});

export default defineAgent({
  slug: "researcher",
  systemPrompt: "Use the DeepWiki tools to answer questions about GitHub repositories.",
  model: "anthropic/claude-opus-4.8",
  tools: [deepwiki],
});

import { defineSandbox } from "@keystrokehq/keystroke/sandbox";

export default defineAgent({
  slug: "support",
  systemPrompt:
    "Read /workspace/agent/product-guide.md and the support skill before answering.",
  model: "openai/gpt-5.6-luna",
  skills: ["support"],
  sandbox: defineSandbox({ files: true }),
});

export default defineAgent({
  slug: "support",
  systemPrompt: "Read the support handbook before answering.",
  model: "xai/grok-4.5",
  sandbox: defineSandbox({ files: "support-handbook" }),
});

export default defineAgent({
  slug: "one-shot-classifier",
  systemPrompt: "Classify the message and return only the label.",
  model: "deepseek/deepseek-v4-flash",
  memory: false,
});

export default defineAgent({
  slug: "support",
  systemPrompt: "Help the user using memory when it is relevant.",
  model: "openai/gpt-5.6-sol",
  memory: {
    memoryCharLimit: 3000,
  },
});

export default defineAgent({
  slug: "researcher",
  systemPrompt: "Research the user's question and cite sources when available.",
  model: "openai/gpt-5.6-luna",
});

export default defineAgent({
  slug: "planner",
  systemPrompt: "Plan carefully, then give a concise answer.",
  model: "openai/gpt-5.6-terra",
  thinkingLevel: "high",
});

import { z } from "zod";
import researcher from "../agents/signup-researcher";

const Summary = z.object({ company: z.string(), summary: z.string() });

const result = await researcher.prompt({
  message: "Research Acme Corp",
  outputSchema: Summary,
});
if (result.error) throw new Error(result.error);

const { company, summary } = result.output!; // typed { company: string; summary: string }

const Decision = z.discriminatedUnion("action", [
  z.object({ action: z.literal("drop"), newPrice: z.number(), reasoning: z.string() }),
  z.object({ action: z.literal("no_more_offers"), reasoning: z.string() }),
  z.object({ action: z.literal("payment_plan"), months: z.number(), reasoning: z.string() }),
]);

export default defineAgent({
  slug: "web-researcher",
  systemPrompt:
    "Use web_search before answering factual questions about current companies or events. Use web_fetch to inspect promising sources.",
  model: "google/gemini-3.5-flash",
});

import { defineAgent } from "@keystrokehq/keystroke/agent";
import { defineSandbox } from "@keystrokehq/keystroke/sandbox";

const seeded = defineSandbox({
  files: [{ path: "context/seed.txt", file: "seeded" }],
});

export default defineAgent({
  slug: "sandbox-seeded",
  systemPrompt: "Read files from /workspace before answering.",
  model: "zai/glm-5.2",
  sandbox: seeded,
});

export default defineAgent({
  slug: "repo-worker",
  systemPrompt: "Clone the repo into /workspace, then make the requested change.",
  model: "openai/gpt-5.6-luna",
  sandbox: defineSandbox({ mode: "vm" }),
});

import { github } from "@keystrokehq/github";

sandbox: defineSandbox({
  mode: "vm",
  credentials: [github.credential] as const,
  env: ({ github }) => ({
    // Optional aliases — GH_TOKEN / GITHUB_TOKEN are injected automatically when `git` is set
    CUSTOM_GITHUB_TOKEN: github.accessToken,
  }),
  git: {
    clone: {
      repository: "owner/repo",
      credential: "github",
      // path defaults to /workspace/session/repo
    },
  },
  // Runs once when the session sandbox is first created — not on later prompts
  setup: ["npm i -g @posthog/cli"],
}),