@band-ai/sdk 0.1.6 → 0.2.0

package/dist/mcp-claude.js

@@ -1,12 +1,15 @@
 import {
   buildRoomScopedRegistrations,
-  buildZodShape
-} from "./chunk-EDPKUUJL.js";
+  buildSingleContextRegistrations,
+  buildZodShape,
+  resolveSingleRoomTools
+} from "./chunk-NT7TBFX2.js";
 import {
+  MCP_SERVER_NAME,
   mcpToolNames
-} from "./chunk-GVEQ3RAH.js";
+} from "./chunk-RV76SKFT.js";
 import "./chunk-3BKAC4OZ.js";
-import "./chunk-YR3BOQNW.js";
+import "./chunk-JY75VSQK.js";
 
 // src/mcp/sdk.ts
 import {
@@ -14,27 +17,223 @@ import {
   tool
 } from "@anthropic-ai/claude-agent-sdk";
 import { z } from "zod";
-function createThenvoiSdkMcpServer(options) {
-  const registrations = buildRoomScopedRegistrations(
-    options.getToolsForRoom,
-    {
-      enableMemoryTools: options.enableMemoryTools,
-      enableContactTools: true,
-      additionalTools: options.additionalTools
-    }
-  );
+function createBandSdkMcpServer(options) {
+  const registrationOptions = {
+    enableMemoryTools: options.enableMemoryTools,
+    enableContactTools: true,
+    additionalTools: options.additionalTools
+  };
+  const registrations = options.multiRoom === false ? buildSingleContextRegistrations(resolveSingleRoomTools(options.getToolsForRoom), registrationOptions) : buildRoomScopedRegistrations(options.getToolsForRoom, registrationOptions);
   const toolDefinitions = registrations.map(toSdkToolDefinition);
   const toolNames = new Set(registrations.map((r) => r.name));
+  const contextCache = /* @__PURE__ */ new Map();
+  const MAX_CONTEXT_CACHE_ENTRIES = 100;
   const serverConfig = createSdkMcpServer({
-    name: "thenvoi",
+    name: MCP_SERVER_NAME,
     tools: toolDefinitions
   });
   return {
     serverConfig,
     allowedTools: mcpToolNames(toolNames),
-    toolDefinitions
+    toolDefinitions,
+    getSystemPromptContext: async (roomId, contextOptions) => {
+      const context = await getOrBuildSystemPromptContext(
+        roomId,
+        contextOptions,
+        options.getToolsForRoom,
+        contextCache,
+        MAX_CONTEXT_CACHE_ENTRIES
+      );
+      return context.markdown;
+    },
+    getSystemPromptContextData: (roomId, contextOptions) => {
+      return getOrBuildSystemPromptContext(
+        roomId,
+        contextOptions,
+        options.getToolsForRoom,
+        contextCache,
+        MAX_CONTEXT_CACHE_ENTRIES
+      );
+    }
+  };
+}
+async function getOrBuildSystemPromptContext(roomId, contextOptions, getToolsForRoom, contextCache, maxEntries) {
+  const ttlMs = contextOptions?.ttlMs ?? 3e4;
+  const now = Date.now();
+  const cached = contextCache.get(roomId);
+  if (cached && cached.expiresAt > now) {
+    cached.lastAccessedAt = now;
+    return cached.value;
+  }
+  if (cached) {
+    contextCache.delete(roomId);
+  }
+  const tools = getToolsForRoom(roomId);
+  if (!tools) {
+    return buildUnavailableSystemPromptContext(roomId);
+  }
+  const context = await buildSystemPromptContext(roomId, tools);
+  contextCache.set(roomId, {
+    value: context,
+    expiresAt: now + ttlMs,
+    lastAccessedAt: now
+  });
+  evictLeastRecentlyUsedContext(contextCache, maxEntries);
+  return context;
+}
+async function buildSystemPromptContext(roomId, tools) {
+  const participants = await tools.getParticipants();
+  const warnings = [];
+  const agentResolution = await resolveAgentIdentity(tools);
+  if (agentResolution.warning) {
+    warnings.push(agentResolution.warning);
+  }
+  const roomResolution = await resolveRoomTitle(tools, roomId);
+  if (roomResolution.warning) {
+    warnings.push(roomResolution.warning);
+  }
+  const agentIdentity = agentResolution.value;
+  const roomTitle = roomResolution.value;
+  const normalizedParticipants = participants.map((participant) => ({
+    id: String(participant.id),
+    name: String(participant.name ?? "Unknown"),
+    type: String(participant.type ?? "Unknown"),
+    handle: normalizeHandle(participant.handle),
+    isSelf: participant.id === agentIdentity?.id
+  }));
+  const selfHandle = normalizeHandle(agentIdentity?.handle);
+  const selfName = agentIdentity?.name ?? "Agent";
+  const mentionHandles = normalizedParticipants.filter((participant) => !participant.isSelf).map((participant) => participant.handle).filter((handle) => Boolean(handle));
+  const mentionFormat = mentionHandles.length > 0 ? mentionHandles.join(", ") : "No participant handles available";
+  const roomLabel = roomTitle ? `"${roomTitle}"` : "this room";
+  const agentLabel = selfHandle ? `${selfName} (${selfHandle})` : selfName;
+  const participantLines = normalizedParticipants.length > 0 ? normalizedParticipants.map((participant) => {
+    const suffix = participant.isSelf ? " (you)" : "";
+    const handle = participant.handle ? ` (${participant.handle})` : "";
+    return `- **${participant.name}**${handle} -- ${participant.type}${suffix}`;
+  }).join("\n") : "- No participants found";
+  const markdown = [
+    "## Room Context",
+    "",
+    `You are **${agentLabel}** in room ${roomLabel} (id: ${roomId}).`,
+    "",
+    "### Participants",
+    participantLines,
+    "",
+    "### Mention Format",
+    `To address someone, use their exact handle: ${mentionFormat}`,
+    ...warnings.length > 0 ? ["", "### Warnings", ...warnings.map((warning) => `- ${warning}`)] : []
+  ].join("\n");
+  return {
+    roomId,
+    roomTitle,
+    agent: {
+      id: agentIdentity?.id ?? "unknown-agent",
+      name: selfName,
+      handle: selfHandle,
+      description: agentIdentity?.description ?? null
+    },
+    participants: normalizedParticipants,
+    mentionFormat,
+    warnings,
+    markdown
+  };
+}
+async function resolveAgentIdentity(tools) {
+  const maybeTools = tools;
+  try {
+    if (maybeTools.getAgentIdentity) {
+      return { value: await maybeTools.getAgentIdentity(), warning: null };
+    }
+    if (maybeTools.rest?.getAgentMe) {
+      return { value: await maybeTools.rest.getAgentMe(), warning: null };
+    }
+  } catch (error) {
+    return {
+      value: null,
+      warning: `Unable to resolve agent identity: ${error instanceof Error ? error.message : String(error)}`
+    };
+  }
+  return { value: null, warning: null };
+}
+async function resolveRoomTitle(tools, roomId) {
+  const maybeTools = tools;
+  const rest = maybeTools.rest;
+  if (!rest?.listChats) {
+    return { value: null, warning: null };
+  }
+  try {
+    let page = 1;
+    const pageSize = 100;
+    while (true) {
+      const response = await rest.listChats({ page, pageSize });
+      const room = response?.data?.find((entry) => entry.id === roomId);
+      if (typeof room?.title === "string" && room.title.length > 0) {
+        return { value: room.title, warning: null };
+      }
+      const totalPages = response?.metadata?.totalPages ?? page;
+      if (page >= totalPages) {
+        return { value: null, warning: null };
+      }
+      page += 1;
+    }
+  } catch (error) {
+    return {
+      value: null,
+      warning: `Unable to resolve room title: ${error instanceof Error ? error.message : String(error)}`
+    };
+  }
+}
+function buildUnavailableSystemPromptContext(roomId) {
+  const warnings = [`No tool context found for room_id ${roomId}`];
+  const markdown = [
+    "## Room Context",
+    "",
+    `You are **Agent** in this room (id: ${roomId}).`,
+    "",
+    "### Participants",
+    "- No participants found",
+    "",
+    "### Mention Format",
+    "No participant handles available",
+    "",
+    "### Warnings",
+    ...warnings.map((warning) => `- ${warning}`)
+  ].join("\n");
+  return {
+    roomId,
+    roomTitle: null,
+    agent: {
+      id: "unknown-agent",
+      name: "Agent",
+      handle: null,
+      description: null
+    },
+    participants: [],
+    mentionFormat: "No participant handles available",
+    warnings,
+    markdown
   };
 }
+function evictLeastRecentlyUsedContext(contextCache, maxEntries) {
+  if (contextCache.size <= maxEntries) {
+    return;
+  }
+  let oldestKey = null;
+  let oldestAccessedAt = Number.POSITIVE_INFINITY;
+  for (const [key, entry] of contextCache.entries()) {
+    if (entry.lastAccessedAt < oldestAccessedAt) {
+      oldestAccessedAt = entry.lastAccessedAt;
+      oldestKey = key;
+    }
+  }
+  if (oldestKey) {
+    contextCache.delete(oldestKey);
+  }
+}
+function normalizeHandle(handle) {
+  return typeof handle === "string" && handle.trim().length > 0 ? handle.trim() : null;
+}
 function toSdkToolDefinition(registration) {
   const shape = buildZodShape(
     z,
@@ -49,5 +248,5 @@ function toSdkToolDefinition(registration) {
   );
 }
 export {
-  createThenvoiSdkMcpServer
+  createBandSdkMcpServer
 };