AgentOffice is a Pi extension.

import type { ExtensionAPI, ToolResult } from "@mariozechner/pi-coding-agent";
import { z } from "zod";

/**
 * MyExtension — sample extension for AgentOffice
 * Registers tools, CLI commands, and event handlers.
 */
export default function myExtension(api: ExtensionAPI) {

  // ── Tool available to the agents ──────────────────────────
  api.registerTool("fetch_data", {
    description: "Fetch external data and return structured JSON",
    parameters: z.object({
      url:     z.string().url().describe("Endpoint URL"),
      timeout: z.number().optional().default(5000)
    }),
    execute: async ({ url, timeout }): Promise<ToolResult> => {
      const controller = new AbortController();
      const id = setTimeout(() => controller.abort(), timeout);
      try {
        const res = await fetch(url, { signal: controller.signal });
        const data = await res.json();
        return { content: JSON.stringify(data, null, 2) };
      } catch (err) {
        return { isError: true, content: String(err) };
      } finally {
        clearTimeout(id);
      }
    }
  });

  // ── CLI command: agentoffice my-extension sync ────────────
  api.registerCommand("sync", async (args) => {
    const dry = args.includes("--dry-run");
    console.log(`[my-extension] sync ${dry ? "(dry-run)" : ""}`);
    if (!dry) await performSync();
  });

  // ── Lifecycle hook ─────────────────────────────────────────
  api.on("agent:start", ({ agentId }) => {
    console.log(`[my-extension] agent ${agentId} started`);
  });
}

async function performSync() {
  // your sync logic here
}