共通エージェント定義をセッション単位で差し替える運用設計

import Anthropic from "@anthropic-ai/sdk";
 
const client = new Anthropic();
 
// 土台エージェントで「許可してよい」上限。オーバーライドがこれを超えたら止める。
const TOOL_ALLOWLIST = new Set([
  "web_search",
  "bash",
  "text_editor",
]);
const MCP_ALLOWLIST = new Set([
  "mcp-claudelab",
  "mcp-gemilab",
  "mcp-antigravitylab",
  "mcp-rorklab",
]);
 
type SiteOverride = {
  site: string;
  systemPrompt: string;
  tools: string[];      // そのセッションでの最終形
  mcpServers: string[]; // そのセッションでの最終形
  skills: string[];
  model?: string;
};
 
function assertSubset(name: string, given: string[], allow: Set<string>) {
  const stray = given.filter((v) => !allow.has(v));
  if (stray.length > 0) {
    // 想定外の権限拡大を、実行前に必ず落とす
    throw new Error(
      `[override guard] ${name} に未許可の項目: ${stray.join(", ")}`
    );
  }
}
 
export function buildSessionAgent(base: string, o: SiteOverride) {
  // オーバーライドは「上書き」なので、渡す前にこちら側で境界を検証する
  assertSubset("tools", o.tools, TOOL_ALLOWLIST);
  assertSubset("mcpServers", o.mcpServers, MCP_ALLOWLIST);
 
  if (o.tools.length === 0) {
    // 空配列は「ツール無し」を意味する。事故を疑いたいので明示ログ。
    console.warn(`[override] ${o.site}: tools が空。意図的か確認すること`);
  }
 
  const overridden = {
    type: "agent_with_overrides" as const,
    agent_id: base,
    overrides: {
      model: o.model,                 // 省略時は土台のモデルを継承
      system: o.systemPrompt,
      tools: o.tools.map((name) => ({ name })),
      mcp_servers: o.mcpServers,
      skills: o.skills,
    },
  };
 
  // 何をどう差し替えたかを一行で残す。あとで挙動を追える。
  console.info(
    `[session] site=${o.site} model=${o.model ?? "(inherit)"} ` +
      `tools=${o.tools.length} mcp=${o.mcpServers.join("+")}`
  );
 
  return overridden;
}

const SITE_OVERRIDES: Record<string, SiteOverride> = {
  claudelab: {
    site: "claudelab",
    systemPrompt: "あなたは Claude 技術ブログの編集者です。敬体で書きます。",
    tools: ["web_search", "text_editor"],
    mcpServers: ["mcp-claudelab"],
    skills: ["article-gate", "templating-gate"],
  },
  rorklab: {
    site: "rorklab",
    systemPrompt: "あなたは Rork アプリ開発ブログの編集者です。敬体で書きます。",
    tools: ["web_search", "text_editor", "bash"], // Rork はビルド確認で bash を使う
    mcpServers: ["mcp-rorklab"],
    skills: ["article-gate"],
    model: "claude-opus-4-8",
  },
};
 
async function runFor(site: string) {
  const override = SITE_OVERRIDES[site];
  if (!override) throw new Error(`未定義のサイト: ${site}`);
 
  const session = await client.beta.messages.sessions.create({
    agent: buildSessionAgent("agent_blog_base", override),
  });
  return session;
}