pi-coding-agent SDK 与扩展系统

import { createAgentSession } from "@earendil-works/pi-coding-agent";

// 默认行为：使用当前工作目录，自动发现资源、读取已保存的设置和凭据
const { session } = await createAgentSession();

try {
  await session.prompt("当前目录下有哪些文件？"); // prompt() 会在这次运行结束后返回
  console.log(session.getLastAssistantText());
} finally {
  session.dispose(); // 中止进行中的工作，并清理扩展上下文和事件监听器
}

import { createAgentSession, SessionManager } from "@earendil-works/pi-coding-agent";

const { session } = await createAgentSession({
  cwd: process.cwd(),
  tools: ["read", "grep", "find", "ls"], // 只读模式：不启用 edit、write、bash
  sessionManager: SessionManager.inMemory(), // 不写会话文件
});

const unsubscribe = session.subscribe((event) => {
  if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
    process.stdout.write(event.assistantMessageEvent.delta); // 流式输出文本
  }
});

try {
  await session.prompt("阅读 README，然后用三句话介绍这个项目。");
  // 运行进行中可以这样插话：
  //   session.steer("...")：在当前这轮工具执行完之后插入
  //   session.followUp("...")：在这次运行结束后再追加一个任务
} finally {
  unsubscribe();
  session.dispose();
}

import {
  createAgentSession,
  createCodemodeExtension,
  createMcpExtension,
  createToolSearchExtension,
  DefaultResourceLoader,
  getAgentDir,
  SessionManager,
  SettingsManager,
} from "@earendil-works/pi-coding-agent";

const cwd = process.cwd();
const resourceLoader = new DefaultResourceLoader({
  cwd,
  agentDir: getAgentDir(),
  extensionFactories: [
    createCodemodeExtension({ mode: "on" }),
    createToolSearchExtension(),
    createMcpExtension(), // 和 CLI 一样，从 agent 目录和受信任的项目中读取 mcp.json
  ],
});
await resourceLoader.reload();

const settingsManager = SettingsManager.create(cwd);
settingsManager.applyOverrides({ defaultTools: ["+codemode", "+tool_search"] }); // +name 表示在默认工具的基础上追加

const { session } = await createAgentSession({ resourceLoader, settingsManager, sessionManager: SessionManager.inMemory() });
try {
  await session.bindExtensions({}); // 触发 session_start 事件，MCP server 会在后台连接
  console.log("当前可用的工具：", session.getActiveToolNames().join(", "));
} finally {
  session.dispose();
}

import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";

export default function (pi: ExtensionAPI) {
  const dangerousPatterns = [/\brm\s+(-rf?|--recursive)/i, /\bsudo\b/i, /\b(chmod|chown)\b.*777/i];

  pi.on("tool_call", async (event, ctx) => {
    if (event.toolName !== "bash") return undefined;

    const command = event.input.command as string;
    const isDangerous = dangerousPatterns.some((p) => p.test(command));

    if (isDangerous) {
      if (!ctx.hasUI) {
        // 非交互模式下默认拦截
        return { block: true, reason: "Dangerous command blocked (no UI for confirmation)" };
      }

      const choice = await ctx.ui.select(`⚠️ Dangerous command:\n\n  ${command}\n\nAllow?`, ["Yes", "No"]);

      if (choice !== "Yes") {
        return { block: true, reason: "Blocked by user" };
      }
    }

    return undefined;
  });
}

import { Type } from "@earendil-works/pi-ai";
import {
  createAgentSession,
  DefaultResourceLoader,
  getAgentDir,
  SessionManager,
} from "@earendil-works/pi-coding-agent";

const resourceLoader = new DefaultResourceLoader({
  cwd: process.cwd(),
  agentDir: getAgentDir(),
  extensionFactories: [
    (pi) => {
      // 1) 注册一个工具，模型可以直接调用
      pi.registerTool({
        name: "get_weather",
        label: "查询天气",
        description: "查询城市今天的天气",
        parameters: Type.Object({ city: Type.String({ description: "城市名" }) }),
        execute: async (_toolCallId, params) => ({
          content: [{ type: "text", text: params.city === "北京" ? "晴，25°C" : "暂无数据" }],
          details: { city: params.city }, // details 用于界面渲染或状态重建，不会发给模型
        }),
      });
      // 2) 审计日志：观察每一次工具调用。返回 { block: true, reason } 可以拦截
      pi.on("tool_call", async (event) => {
        console.log(`[audit] ${event.toolName}`);
        return undefined;
      });
      // 3) 注册一个斜杠命令
      pi.registerCommand("weather-help", {
        description: "显示天气工具的用法",
        handler: async (_args, ctx) => ctx.ui.notify("直接问：北京今天天气怎么样？", "info"),
      });
    },
  ],
});
await resourceLoader.reload();

const { session } = await createAgentSession({ resourceLoader, sessionManager: SessionManager.inMemory() });
try {
  await session.prompt("北京今天适合跑步吗？");
  console.log(session.getLastAssistantText());
} finally {
  session.dispose();
}