@dudousxd/nestjs-agent-core 0.15.1 → 0.15.2

package/dist/index.cjs

@@ -154,6 +154,7 @@ __export(index_exports, {
   skillInputSchema: () => skillInputSchema,
   skillToolDefinition: () => skillToolDefinition,
   skillWriteVerdict: () => skillWriteVerdict,
+  stampToolKinds: () => stampToolKinds,
   staticSkillProvider: () => staticSkillProvider,
   summarizeWithModel: () => summarizeWithModel,
   tenantScope: () => tenantScope,
@@ -2615,6 +2616,16 @@ async function structureAnswer(schema, deps, input, hooks, messages, step) {
   throw new StructuredOutputError(issues, text, maxAttempts);
 }
 __name(structureAnswer, "structureAnswer");
+function stampToolKinds(result, deps) {
+  return {
+    ...result,
+    toolCalls: result.toolCalls.map((call) => call.kind === void 0 ? {
+      ...call,
+      kind: declaredKind(deps, call.name)
+    } : call)
+  };
+}
+__name(stampToolKinds, "stampToolKinds");
 function declaredKind(deps, name) {
   if (deps.ask === true && name === ASK_TOOL_NAME) {
     return "ask";
@@ -2815,7 +2826,7 @@ async function claimToolCall(turn, call) {
   const { deps, input, hooks, messageId } = turn;
   const persisted = await hooks.step(`persist:toolcall:${call.id}`, async () => {
     const spec = deps.registry.spec(call.name);
-    const kind = declaredKind(deps, call.name);
+    const kind = call.kind ?? declaredKind(deps, call.name);
     const awaitsHuman = kind === "action" || kind === "ask";
     await deps.store.recordToolCall({
       toolCallId: call.id,
@@ -3479,12 +3490,12 @@ ${block}`;
           writer
         }) : void 0;
         const buffer = gateMode === "whole" ? createFrameBuffer() : void 0;
-        const result = await traceLlmTurn(hooks.runId, i, () => deps.model.runTurn({
+        const result = stampToolKinds(await traceLlmTurn(hooks.runId, i, () => deps.model.runTurn({
           system: prompt.system,
           messages: prompt.messages,
           tools,
           sink: incremental?.writer ?? buffer?.writer ?? writer
-        }));
+        })), deps);
         if (incremental !== void 0) {
           await incremental.settled();
           const refusal = incremental.rejection();
@@ -3543,7 +3554,7 @@ ${block}`;
     const costUsd = resolveCostUsd(turn.usage, turn.costUsd, priceByModel.get(resolvedModelId));
     const toolCallsWithKind = turn.toolCalls.map((call) => ({
       ...call,
-      kind: declaredKind(deps, call.name)
+      kind: call.kind ?? declaredKind(deps, call.name)
     }));
     await hooks.step(`persist:usage:${i}`, () => deps.store.recordUsage({
       threadId: input.threadId,
@@ -3980,6 +3991,7 @@ __name(runAgentLoop, "runAgentLoop");
   skillInputSchema,
   skillToolDefinition,
   skillWriteVerdict,
+  stampToolKinds,
   staticSkillProvider,
   summarizeWithModel,
   tenantScope,