@cognipeer/agent-sdk 0.7.1 → 0.7.3

package/dist/index.mjs

@@ -3026,6 +3026,17 @@ function createToolsNode(initialTools, opts) {
         }
         const durationMs = Date.now() - start;
         const executionId = nanoid();
+        if (output && typeof output === "object" && Array.isArray(output.__runtimeToolsDelta)) {
+          const delta = output.__runtimeToolsDelta;
+          delete output.__runtimeToolsDelta;
+          const currentRuntime = state.agent || runtime;
+          const currentTools = currentRuntime?.tools ?? [];
+          const present = new Set(currentTools.map((tool2) => tool2?.name));
+          const additions = delta.filter((tool2) => tool2 && tool2.name && !present.has(tool2.name));
+          if (additions.length > 0 && currentRuntime) {
+            state.agent = { ...currentRuntime, tools: [...currentTools, ...additions] };
+          }
+        }
         if (output && typeof output === "object" && output.__handoff && output.__handoff.runtime) {
           toolHistory.push({ executionId, toolName, args, output: "handoff:ok", rawOutput: output, timestamp: (/* @__PURE__ */ new Date()).toISOString(), tool_call_id: tc.id, status: "handoff" });
           push({ role: "tool", content: "ok", tool_call_id: tc.id || `${tc.name}_${toolCount}`, name: tc.name });
@@ -6183,6 +6194,190 @@ async function writeSummaryFactsToMemory(store, config, summary, sourceTurn) {
   return store.get(config.memory.scope, { limit: config.context.lastTurnsToKeep + summary.stable_facts.length });
 }
 
+// src/smart/skills/types.ts
+var DEFAULT_SKILL_POLICY = {
+  maxOpenSkills: 4,
+  maxBoundToolsPerSkill: 12,
+  maxBoundToolsTotal: 40
+};
+var SMALL_TIER_SKILL_POLICY = {
+  maxOpenSkills: 1,
+  maxBoundToolsPerSkill: 6,
+  maxBoundToolsTotal: 18,
+  modelTier: "small"
+};
+function createSkillRegistryRef() {
+  return { openedSkillKeys: [], boundSkillTools: [] };
+}
+
+// src/smart/skills/registry.ts
+function composeToolSets(input) {
+  const without = [...input.base, ...input.boundSkillTools];
+  const withRec = input.recoveryTool ? [...without, input.recoveryTool] : [...without];
+  return { toolsWithoutRecovery: without, toolsWithRecovery: withRec };
+}
+function appendBoundTools(ref, newTools, policy) {
+  const present = new Set(ref.boundSkillTools.map((tool) => tool.name));
+  const added = [];
+  for (const tool of newTools) {
+    if (ref.boundSkillTools.length >= policy.maxBoundToolsTotal) break;
+    if (present.has(tool.name)) continue;
+    present.add(tool.name);
+    ref.boundSkillTools.push(tool);
+    added.push(tool);
+  }
+  return added;
+}
+function canOpenSkill(ref, policy) {
+  if (ref.openedSkillKeys.length >= policy.maxOpenSkills) {
+    return { ok: false, reason: `max ${policy.maxOpenSkills} skill(s) already open` };
+  }
+  if (ref.boundSkillTools.length >= policy.maxBoundToolsTotal) {
+    return { ok: false, reason: "skill tool budget exhausted" };
+  }
+  return { ok: true };
+}
+async function resolveAvailableSkills(skills, opts = {}) {
+  const out = [];
+  for (const skill of skills) {
+    if (skill.minModelTier === "large" && opts.modelTier === "small") continue;
+    const available = skill.isAvailable ? await skill.isAvailable() : true;
+    if (available) out.push(skill);
+  }
+  return out;
+}
+function buildSkillHeaderBlock(skills) {
+  if (skills.length === 0) return "";
+  const lines = skills.map((skill) => `- ${skill.key}: ${skill.header}`);
+  return [
+    "<available_skills>",
+    "Open a skill with open_skill(skillKey) to load its tools on demand. Only open what the task needs; close nothing \u2014 opened skills stay available for the rest of the run.",
+    ...lines,
+    "</available_skills>"
+  ].join("\n");
+}
+function dedupeToolsByName(tools) {
+  const seen = /* @__PURE__ */ new Set();
+  const out = [];
+  for (const tool of tools) {
+    if (seen.has(tool.name)) continue;
+    seen.add(tool.name);
+    out.push(tool);
+  }
+  return out;
+}
+
+// src/smart/skills/skillTools.ts
+var MAX_RETURNED_INDEX = 40;
+var OPEN_SKILL_DESCRIPTION = "Load a capability on demand. Pass a skillKey from <available_skills>. A small skill binds all its tools immediately; a large one returns a tool index \u2014 then call bind_skill_tools with only the tools you need. Open only what the task needs.";
+var BIND_SKILL_TOOLS_DESCRIPTION = "After open_skill on a large skill, bind a chosen subset of its tools by name. Pick the few tools the task actually needs.";
+function findSkill(deps, key) {
+  return deps.skills.find((skill) => skill.key === key);
+}
+function createOpenSkillTool(deps) {
+  const tool = createTool({
+    name: "open_skill",
+    description: OPEN_SKILL_DESCRIPTION,
+    schema: z.object({
+      skillKey: z.string().min(1).describe("A key from <available_skills>."),
+      query: z.string().optional().describe("What you need from the skill \u2014 used to rank a large skill's tools.")
+    }).strict(),
+    func: async ({ skillKey, query }) => {
+      const { registryRef: ref, policy } = tool._deps;
+      const skill = findSkill(tool._deps, skillKey);
+      if (!skill) {
+        return { error: `Unknown skill: ${skillKey}. Choose a key listed in <available_skills>.` };
+      }
+      if (skill.isAvailable && !await skill.isAvailable()) {
+        return { error: `Skill ${skillKey} is not available in this workspace.` };
+      }
+      const alreadyOpen = ref.openedSkillKeys.includes(skillKey);
+      if (!alreadyOpen) {
+        const guard = canOpenSkill(ref, policy);
+        if (!guard.ok) {
+          return {
+            error: `Cannot open ${skillKey}: ${guard.reason}. Work with the skills already open.`
+          };
+        }
+        ref.openedSkillKeys.push(skillKey);
+      }
+      const prompt = typeof skill.prompt === "function" ? await skill.prompt() : skill.prompt;
+      const index = await skill.listToolIndex();
+      if (index.length <= policy.maxBoundToolsPerSkill) {
+        const bound = await skill.bindTools();
+        const added = appendBoundTools(ref, bound, policy);
+        if (added.length > 0) ref.__onToolsChanged?.();
+        return {
+          skillKey,
+          prompt,
+          needsSandbox: Boolean(skill.needsSandbox),
+          boundTools: added.map((t) => t.name),
+          // Inject the bound tools into the live runtime for this loop.
+          __runtimeToolsDelta: added
+        };
+      }
+      let defaultBound = [];
+      let defaultAdded = [];
+      if ((!query || query.trim().length < 3) && skill.defaultBindNames?.length) {
+        const bound = await skill.bindTools(skill.defaultBindNames.slice(0, policy.maxBoundToolsPerSkill));
+        defaultAdded = appendBoundTools(ref, bound, policy);
+        if (defaultAdded.length > 0) ref.__onToolsChanged?.();
+        defaultBound = defaultAdded.map((t) => t.name);
+      }
+      const headers = query && query.trim().length >= 3 && skill.rankToolHeaders ? await skill.rankToolHeaders(query, index) : index;
+      return {
+        skillKey,
+        prompt,
+        needsSandbox: Boolean(skill.needsSandbox),
+        boundTools: defaultBound,
+        toolIndex: headers.slice(0, MAX_RETURNED_INDEX).map((h) => ({ name: h.name, description: h.description })),
+        hint: "Call bind_skill_tools(skillKey, toolNames) with only the tools you need.",
+        ...defaultAdded.length > 0 ? { __runtimeToolsDelta: defaultAdded } : {}
+      };
+    }
+  });
+  tool._deps = deps;
+  tool.__source = "smart-skills";
+  return tool;
+}
+function createBindSkillToolsTool(deps) {
+  const tool = createTool({
+    name: "bind_skill_tools",
+    description: BIND_SKILL_TOOLS_DESCRIPTION,
+    schema: z.object({
+      skillKey: z.string().min(1),
+      toolNames: z.array(z.string().min(1)).min(1).describe("Tool names from the skill's index.")
+    }).strict(),
+    func: async ({ skillKey, toolNames }) => {
+      const { registryRef: ref, policy } = tool._deps;
+      const skill = findSkill(tool._deps, skillKey);
+      if (!skill) {
+        return { error: `Unknown skill: ${skillKey}.` };
+      }
+      if (!ref.openedSkillKeys.includes(skillKey)) {
+        return { error: `Call open_skill("${skillKey}") before binding its tools.` };
+      }
+      const capped = toolNames.slice(0, policy.maxBoundToolsPerSkill);
+      const bound = await skill.bindTools(capped);
+      const added = appendBoundTools(ref, bound, policy);
+      if (added.length > 0) ref.__onToolsChanged?.();
+      return {
+        skillKey,
+        boundTools: added.map((t) => t.name),
+        requested: capped.length,
+        truncated: toolNames.length > capped.length,
+        ...added.length > 0 ? { __runtimeToolsDelta: added } : {}
+      };
+    }
+  });
+  tool._deps = deps;
+  tool.__source = "smart-skills";
+  return tool;
+}
+function createSkillTools(deps) {
+  return [createOpenSkillTool(deps), createBindSkillToolsTool(deps)];
+}
+
 // src/smart/index.ts
 function createSmartAgent(opts) {
   const resolved = normalizeSmartAgentOptions(opts);
@@ -6217,6 +6412,9 @@ function createSmartAgent(opts) {
   if (askUserConfig) {
     factoryToolsWithoutRecovery.push(createAskUserQuestionTool(factoryStateRef, askUserConfig));
   }
+  const skills = opts.skills ?? [];
+  const skillPolicy = opts.skillPolicy ?? DEFAULT_SKILL_POLICY;
+  const skillsEnabled = skills.length > 0;
   const soManagerFactory = opts.outputSchema ? new StructuredOutputManager(opts.outputSchema, resolveStrategy(opts.model)) : void 0;
   const modelCapabilities = opts.outputSchema ? getModelCapabilities(opts.model) : void 0;
   const responseFinalizeTool = soManagerFactory && modelCapabilities?.structuredOutput !== "native" ? soManagerFactory.getResponseTool() : void 0;
@@ -6224,12 +6422,31 @@ function createSmartAgent(opts) {
     const ref = { toolHistory: void 0, toolHistoryArchived: void 0, todoList: void 0, planVersion: 0, adherenceScore: 0 };
     const ctxTools = createContextTools(ref, { planningEnabled, includeGetToolResponse: false });
     const recoveryTool = createGetToolResponseTool(ref);
-    const base2 = [...userTools, ...ctxTools];
-    if (responseFinalizeTool) base2.push(responseFinalizeTool);
-    if (askUserConfig) base2.push(createAskUserQuestionTool(ref, askUserConfig));
-    const without = base2;
-    const withRec = [...without, recoveryTool];
-    return { stateRef: ref, toolsWithoutRecovery: without, toolsWithRecovery: withRec };
+    const coreBase = [...userTools, ...ctxTools];
+    if (responseFinalizeTool) coreBase.push(responseFinalizeTool);
+    if (askUserConfig) coreBase.push(createAskUserQuestionTool(ref, askUserConfig));
+    if (!skillsEnabled) {
+      const without = coreBase;
+      const withRec = [...without, recoveryTool];
+      return { stateRef: ref, toolsWithoutRecovery: without, toolsWithRecovery: withRec };
+    }
+    const toolSet = { stateRef: ref, toolsWithoutRecovery: [], toolsWithRecovery: [] };
+    const skillRegistryRef = createSkillRegistryRef();
+    const skillMetaTools = createSkillTools({ registryRef: skillRegistryRef, skills, policy: skillPolicy });
+    const baseWithMeta = [...coreBase, ...skillMetaTools];
+    const rebuild = () => {
+      const composed = composeToolSets({
+        base: baseWithMeta,
+        boundSkillTools: skillRegistryRef.boundSkillTools,
+        recoveryTool
+      });
+      toolSet.toolsWithoutRecovery = composed.toolsWithoutRecovery;
+      toolSet.toolsWithRecovery = composed.toolsWithRecovery;
+    };
+    skillRegistryRef.__onToolsChanged = rebuild;
+    toolSet.skillRegistryRef = skillRegistryRef;
+    rebuild();
+    return toolSet;
   };
   const base = createAgent({ ...runtimeOpts, tools: factoryToolsWithoutRecovery });
   base.__runtime.runtimeProfile = resolved.runtimeProfile;
@@ -6280,9 +6497,9 @@ function createSmartAgent(opts) {
       planVersion: ref.planVersion || state.planVersion || 1
     };
   }
-  function systemMessage2() {
+  function systemMessage2(extraBlock) {
     const sys = buildSystemPrompt(
-      [opts.systemPrompt, runtimeHint, structuredOutputHint].filter(Boolean).join("\n"),
+      [opts.systemPrompt, runtimeHint, structuredOutputHint, extraBlock].filter(Boolean).join("\n"),
       planningEnabled,
       opts.name || "Agent",
       opts.todoListPrompt
@@ -6332,8 +6549,10 @@ function createSmartAgent(opts) {
       stateRef.planVersion = input.planVersion || input.plan?.version || 0;
       stateRef.adherenceScore = input.plan?.adherenceScore || 0;
       const syncPlanFromRef = (state2) => syncPlanStateWith(stateRef, state2);
+      const availableSkills = skillsEnabled ? await resolveAvailableSkills(skills, { modelTier: skillPolicy.modelTier }) : [];
+      const skillBlock = availableSkills.length > 0 ? buildSkillHeaderBlock(availableSkills) : void 0;
       const alreadyHasSystem = Array.isArray(input.messages) && input.messages[0]?.role === "system";
-      const seedMessages = alreadyHasSystem ? [...input.messages || []] : [systemMessage2(), ...input.messages || []];
+      const seedMessages = alreadyHasSystem ? [...input.messages || []] : [systemMessage2(skillBlock), ...input.messages || []];
       let state = syncRuntimeTools(await syncMemory({ ...input, messages: seedMessages }), seedMessages, toolSet);
       let lastResult = null;
       let rawMessages = [...seedMessages];
@@ -7599,7 +7818,7 @@ var AnthropicProvider = class extends BaseProvider {
     const { system, messages } = this.splitSystemMessages(request.messages);
     const body = {
       model: request.model || this.defaultModel,
-      messages: messages.map((m) => this.toAnthropicMessage(m)),
+      messages: this.toAnthropicMessages(messages),
       max_tokens: request.maxTokens ?? 4096,
       stream
     };
@@ -7655,20 +7874,44 @@ var AnthropicProvider = class extends BaseProvider {
       messages: rest
     };
   }
-  toAnthropicMessage(m) {
-    if (m.role === "tool" && m.toolCallId) {
-      const content = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
-      return {
-        role: "user",
-        content: [
-          {
-            type: "tool_result",
-            tool_use_id: m.toolCallId,
-            content
-          }
-        ]
-      };
+  /**
+   * Converts unified messages to Anthropic format, coalescing consecutive
+   * tool-result messages into a single `user` turn.
+   *
+   * The unified message list carries one `role: "tool"` entry per tool call
+   * (mirroring OpenAI's shape). Anthropic's Messages API expects every
+   * `tool_result` block answering a single assistant turn's `tool_use`
+   * block(s) to be batched into ONE `user` message rather than sent as
+   * separate consecutive `user` messages.
+   */
+  toAnthropicMessages(messages) {
+    const result = [];
+    let mergingToolResults = false;
+    for (const m of messages) {
+      if (m.role === "tool" && m.toolCallId) {
+        const toolResult = this.toAnthropicToolResult(m);
+        if (mergingToolResults) {
+          result[result.length - 1].content.push(toolResult);
+        } else {
+          result.push({ role: "user", content: [toolResult] });
+          mergingToolResults = true;
+        }
+        continue;
+      }
+      mergingToolResults = false;
+      result.push(this.toAnthropicMessage(m));
     }
+    return result;
+  }
+  toAnthropicToolResult(m) {
+    const content = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
+    return {
+      type: "tool_result",
+      tool_use_id: m.toolCallId,
+      content
+    };
+  }
+  toAnthropicMessage(m) {
     if (m.role === "assistant" && m.toolCalls?.length) {
       const content = [];
       appendReasoningBlocks(content, m.reasoning?.blocks);
@@ -8053,7 +8296,7 @@ var BedrockProvider = class extends BaseProvider {
   buildConverseBody(request) {
     const { system, messages } = this.splitSystemMessages(request.messages);
     const body = {
-      messages: messages.map((m) => this.toBedrockMessage(m))
+      messages: this.toBedrockMessages(messages)
     };
     if (system.length > 0) {
       const systemBlocks = system.map((s) => ({ text: s }));
@@ -8103,22 +8346,47 @@ var BedrockProvider = class extends BaseProvider {
     }
     return { system, messages: rest };
   }
-  toBedrockMessage(m) {
-    if (m.role === "tool" && m.toolCallId) {
-      const text = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
-      return {
-        role: "user",
-        content: [
-          {
-            toolResult: {
-              toolUseId: m.toolCallId,
-              content: [{ text }],
-              status: "success"
-            }
-          }
-        ]
-      };
+  /**
+   * Converts unified messages to Bedrock Converse format, coalescing
+   * consecutive tool-result messages into a single `user` turn.
+   *
+   * The unified message list carries one `role: "tool"` entry per tool call
+   * (mirroring OpenAI's shape). Bedrock's Converse API requires every
+   * `toolResult` block answering a single assistant turn's `toolUse`
+   * block(s) to be batched into ONE `user` message — sending them as
+   * separate consecutive `user` messages triggers: "Expected toolResult
+   * blocks at messages.N.content for the following Ids: ...".
+   */
+  toBedrockMessages(messages) {
+    const result = [];
+    let mergingToolResults = false;
+    for (const m of messages) {
+      if (m.role === "tool" && m.toolCallId) {
+        const toolResult = this.toBedrockToolResult(m);
+        if (mergingToolResults) {
+          result[result.length - 1].content.push(toolResult);
+        } else {
+          result.push({ role: "user", content: [toolResult] });
+          mergingToolResults = true;
+        }
+        continue;
+      }
+      mergingToolResults = false;
+      result.push(this.toBedrockMessage(m));
     }
+    return result;
+  }
+  toBedrockToolResult(m) {
+    const text = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
+    return {
+      toolResult: {
+        toolUseId: m.toolCallId,
+        content: [{ text }],
+        status: "success"
+      }
+    };
+  }
+  toBedrockMessage(m) {
     if (m.role === "assistant" && m.toolCalls?.length) {
       const content2 = [];
       appendBedrockReasoning(content2, m.reasoning?.blocks);
@@ -8424,12 +8692,24 @@ var VertexProvider = class extends BaseProvider {
   buildContents(messages) {
     const systemParts = [];
     const contents = [];
+    let mergingToolResults = false;
     for (const m of messages) {
       if (m.role === "system") {
         const text = typeof m.content === "string" ? m.content : m.content.filter((p) => p.type === "text").map((p) => p.text).join("\n");
         if (text) systemParts.push(text);
         continue;
       }
+      if (m.role === "tool" && m.toolCallId) {
+        const part = this.toGeminiFunctionResponsePart(m);
+        if (mergingToolResults) {
+          contents[contents.length - 1].parts.push(part);
+        } else {
+          contents.push({ role: "user", parts: [part] });
+          mergingToolResults = true;
+        }
+        continue;
+      }
+      mergingToolResults = false;
       contents.push(this.toGeminiContent(m));
     }
     return {
@@ -8437,27 +8717,22 @@ var VertexProvider = class extends BaseProvider {
       contents
     };
   }
-  toGeminiContent(m) {
-    if (m.role === "tool" && m.toolCallId) {
-      const text = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
-      let response;
-      try {
-        response = JSON.parse(text);
-      } catch {
-        response = { result: text };
-      }
-      return {
-        role: "user",
-        parts: [
-          {
-            functionResponse: {
-              name: m.name ?? m.toolCallId,
-              response
-            }
-          }
-        ]
-      };
+  toGeminiFunctionResponsePart(m) {
+    const text = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
+    let response;
+    try {
+      response = JSON.parse(text);
+    } catch {
+      response = { result: text };
     }
+    return {
+      functionResponse: {
+        name: m.name ?? m.toolCallId,
+        response
+      }
+    };
+  }
+  toGeminiContent(m) {
     if (m.role === "assistant" && m.toolCalls?.length) {
       const parts2 = [];
       if (typeof m.content === "string" && m.content) {
@@ -9114,6 +9389,6 @@ function createProvider(config) {
   }
 }
 
-export { AnthropicProvider, AzureProvider, BaseProvider, BedrockProvider, DEFAULT_PROFILE_CONFIGS, GuardrailPhase, InMemoryMemoryStore, NativeJsonSchemaStrategy, OpenAICompatibleProvider, OpenAIProvider, ProviderError, StructuredOutputManager, ToolBasedStrategy, VertexProvider, agentVerdictRule, buildSystemPrompt, captureSnapshot, codePresenceRule, cognipeerSink, countApproxTokens, countMessagesTokens, createAgent, createAgentNode, createAskUserQuestionTool, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGetToolResponseTool, createGuardrail, createJsonGuardrail, createManageTodoTool, createProvider, createRegexGuardrail, createResolverNode, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, defaultTokenCounter, emptyUsage, evaluateGuardrails, fileSink, formatZodError, fromLangchainModel, fromLangchainTools, fromNativeProvider, generateSpanId, generateTraceId, getModelCapabilities, getNestedValue, getResolvedSmartConfig, getTokenCounter, hasToolResponseRecoveryReference, httpSink, isSmartModel, jsonSchemaRule, normalizeSmartAgentOptions, otlpSink, readMemoryFacts, regexRule, resolveMemoryStore, resolveReasoning, resolveStrategy, resolveToolApprovalState, resolveUserQuestionState, restoreSnapshot, runSmartAgentEvalHarness, setTokenCounter, startStreamingSession, traceSessionToOtlp, validateReasoningConfig, withTools, writeSummaryFactsToMemory };
+export { AnthropicProvider, AzureProvider, BaseProvider, BedrockProvider, DEFAULT_PROFILE_CONFIGS, DEFAULT_SKILL_POLICY, GuardrailPhase, InMemoryMemoryStore, NativeJsonSchemaStrategy, OpenAICompatibleProvider, OpenAIProvider, ProviderError, SMALL_TIER_SKILL_POLICY, StructuredOutputManager, ToolBasedStrategy, VertexProvider, agentVerdictRule, appendBoundTools, buildSkillHeaderBlock, buildSystemPrompt, canOpenSkill, captureSnapshot, codePresenceRule, cognipeerSink, composeToolSets, countApproxTokens, countMessagesTokens, createAgent, createAgentNode, createAskUserQuestionTool, createBindSkillToolsTool, createCodeGuardrail, createContextSummarizeNode, createContextTools, createGetToolResponseTool, createGuardrail, createJsonGuardrail, createManageTodoTool, createOpenSkillTool, createProvider, createRegexGuardrail, createResolverNode, createSkillRegistryRef, createSkillTools, createSmartAgent, createTool, createToolLimitFinalizeNode, createToolsNode, customCallbackRule, customSink, dedupeToolsByName, defaultTokenCounter, emptyUsage, evaluateGuardrails, fileSink, formatZodError, fromLangchainModel, fromLangchainTools, fromNativeProvider, generateSpanId, generateTraceId, getModelCapabilities, getNestedValue, getResolvedSmartConfig, getTokenCounter, hasToolResponseRecoveryReference, httpSink, isSmartModel, jsonSchemaRule, normalizeSmartAgentOptions, otlpSink, readMemoryFacts, regexRule, resolveAvailableSkills, resolveMemoryStore, resolveReasoning, resolveStrategy, resolveToolApprovalState, resolveUserQuestionState, restoreSnapshot, runSmartAgentEvalHarness, setTokenCounter, startStreamingSession, traceSessionToOtlp, validateReasoningConfig, withTools, writeSummaryFactsToMemory };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map