@dudousxd/nestjs-agent-core 0.5.0 → 0.7.0

package/dist/index.cjs

@@ -23,8 +23,10 @@ var index_exports = {};
 __export(index_exports, {
   AGENT_ACTOR_DIRECTORY: () => AGENT_ACTOR_DIRECTORY,
   AGENT_ACTOR_RESOLVER: () => AGENT_ACTOR_RESOLVER,
+  AGENT_APPROVAL_PORT: () => AGENT_APPROVAL_PORT,
   AGENT_ATTACHMENT_STAGING: () => AGENT_ATTACHMENT_STAGING,
   AGENT_DEPS_FACTORY: () => AGENT_DEPS_FACTORY,
+  AGENT_DIAGNOSTIC_EVENTS: () => AGENT_DIAGNOSTIC_EVENTS,
   AGENT_DURABLE_RUNNER: () => AGENT_DURABLE_RUNNER,
   AGENT_EMBEDDING_PROVIDER: () => AGENT_EMBEDDING_PROVIDER,
   AGENT_GOVERNANCE_QUERIES: () => AGENT_GOVERNANCE_QUERIES,
@@ -38,6 +40,7 @@ __export(index_exports, {
   AGENT_ROLES_POLICY: () => AGENT_ROLES_POLICY,
   AGENT_RUNNER: () => AGENT_RUNNER,
   AGENT_SINK: () => AGENT_SINK,
+  AGENT_SPAN_EVENTS: () => AGENT_SPAN_EVENTS,
   AGENT_STORE: () => AGENT_STORE,
   AGENT_TOOL_REGISTRY: () => AGENT_TOOL_REGISTRY,
   AgentRegistry: () => AgentRegistry,
@@ -48,6 +51,7 @@ __export(index_exports, {
   ToolInputInvalidError: () => ToolInputInvalidError,
   ToolNotFoundError: () => ToolNotFoundError,
   ToolRegistry: () => ToolRegistry,
+  agentDiagnosticKey: () => agentDiagnosticKey,
   bucketByActor: () => bucketByActor,
   bucketByModel: () => bucketByModel,
   bucketByThread: () => bucketByThread,
@@ -67,6 +71,8 @@ __export(index_exports, {
   publishAgentToolCall: () => publishAgentToolCall,
   runAgentLoop: () => runAgentLoop,
   seedModelPrices: () => seedModelPrices,
+  traceLlmTurn: () => traceLlmTurn,
+  traceToolExecution: () => traceToolExecution,
   withToolTimeout: () => withToolTimeout
 });
 module.exports = __toCommonJS(index_exports);
@@ -91,6 +97,7 @@ var AGENT_DEPS_FACTORY = Symbol.for("@dudousxd/nestjs-agent:deps-factory");
 var AGENT_PROMPT_CONTRIBUTORS = Symbol.for("@dudousxd/nestjs-agent:prompt-contributors");
 var AGENT_ACTOR_DIRECTORY = Symbol.for("@dudousxd/nestjs-agent:actor-directory");
 var AGENT_ATTACHMENT_STAGING = Symbol.for("@dudousxd/nestjs-agent:attachment-staging");
+var AGENT_APPROVAL_PORT = Symbol.for("@dudousxd/nestjs-agent:approval-port");
 
 // src/spi/token-stream-sink.ts
 var AgentStreamError = class extends Error {
@@ -394,6 +401,10 @@ var DefaultRolesPolicy = class {
   }
 };
 
+// src/agent-loop.ts
+var import_node_crypto = require("crypto");
+var import_nestjs_diagnostics2 = require("@dudousxd/nestjs-diagnostics");
+
 // src/diagnostics.ts
 var import_nestjs_diagnostics = require("@dudousxd/nestjs-diagnostics");
 function publishAgentRunStarted(payload) {
@@ -428,6 +439,26 @@ function publishAgentRetrieved(payload) {
   (0, import_nestjs_diagnostics.emit)("agent", "retrieved", payload);
 }
 __name(publishAgentRetrieved, "publishAgentRetrieved");
+var AGENT_SPAN_EVENTS = [
+  "llm.turn",
+  "tool.execution",
+  "retrieval",
+  "follow-ups"
+];
+var AGENT_DIAGNOSTIC_EVENTS = [
+  "run.started",
+  "message",
+  "tool-call",
+  "quota.exceeded",
+  "run.finished",
+  "run.failed",
+  "delegated",
+  "retrieved"
+];
+function agentDiagnosticKey(event) {
+  return `agent:${event}`;
+}
+__name(agentDiagnosticKey, "agentDiagnosticKey");
 
 // src/agent-loop.ts
 function resolveCostUsd(usage, reportedCostUsd, price) {
@@ -555,6 +586,39 @@ async function generateFollowUps(model, messages, count) {
   };
 }
 __name(generateFollowUps, "generateFollowUps");
+async function spanned(event, runId, payload, run, summarize) {
+  let value;
+  await (0, import_nestjs_diagnostics2.trace)("agent", event, async () => {
+    value = await run();
+    return summarize(value);
+  }, payload, {
+    traceId: runId
+  });
+  return value;
+}
+__name(spanned, "spanned");
+function traceLlmTurn(runId, step, run) {
+  return spanned("llm.turn", runId, {
+    runId,
+    step
+  }, run, (turn) => ({
+    ...turn.modelId !== void 0 ? {
+      modelId: turn.modelId
+    } : {},
+    inputTokens: turn.usage.inputTokens,
+    outputTokens: turn.usage.outputTokens,
+    textLength: turn.text.length,
+    toolCalls: turn.toolCalls.length
+  }));
+}
+__name(traceLlmTurn, "traceLlmTurn");
+function traceToolExecution(runId, call, run) {
+  return spanned("tool.execution", runId, {
+    runId,
+    ...call
+  }, run, () => ({}));
+}
+__name(traceToolExecution, "traceToolExecution");
 async function runAgentLoop(deps, input, hooks) {
   const maxSteps = deps.maxSteps ?? 8;
   let system = await resolveSystemPrompt(deps, input);
@@ -625,21 +689,30 @@ async function runAgentLoop(deps, input, hooks) {
   });
   const startedAt = await hooks.step("run:started-at", () => Promise.resolve(Date.now()));
   await hooks.step("persist:run:start", async () => {
+    const promptHash = (0, import_node_crypto.createHash)("sha256").update(system).digest("hex");
     await deps.store.recordRunStart?.({
       runId: hooks.runId,
       threadId: input.threadId,
       actorRef: input.actor.id,
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
-      } : {}
+      } : {},
+      promptHash
     });
   });
   let injectedPassages;
   if (deps.retriever !== void 0) {
     const retriever = deps.retriever;
-    const passages = await hooks.step("retrieve", () => retriever.retrieve(input.userText, {
-      topK: deps.retrievalTopK ?? 5
-    }));
+    const topK = deps.retrievalTopK ?? 5;
+    const passages = await hooks.step("retrieve", () => spanned("retrieval", hooks.runId, {
+      runId: hooks.runId,
+      queryLength: input.userText.length,
+      topK
+    }, () => retriever.retrieve(input.userText, {
+      topK
+    }), (retrieved) => ({
+      count: retrieved.length
+    })));
     if (passages.length > 0) {
       injectedPassages = passages;
       system = `${system}
@@ -679,12 +752,12 @@ ${buildContextBlock(passages)}`;
       });
     } else {
       const tools = await deps.registry.definitionsFor(input.actor, deps.rolesPolicy, deps.toolAllowList);
-      turn = await hooks.step(`llm:${i}`, () => deps.model.runTurn({
+      turn = await hooks.step(`llm:${i}`, () => traceLlmTurn(hooks.runId, i, () => deps.model.runTurn({
         system,
         messages: modelMessages,
         tools,
         sink: writer
-      }));
+      })));
     }
     const resolvedModelId = turn.modelId ?? deps.modelId ?? "unknown";
     const costUsd = resolveCostUsd(turn.usage, turn.costUsd, priceByModel.get(resolvedModelId));
@@ -721,13 +794,24 @@ ${buildContextBlock(passages)}`;
     let followUps;
     if (isFinalTurn && deps.followUpsCount !== void 0 && deps.followUpsCount > 0) {
       const count = deps.followUpsCount;
-      const generated = await hooks.step(`followups:${i}`, () => generateFollowUps(deps.model, [
+      const generated = await hooks.step(`followups:${i}`, () => spanned("follow-ups", hooks.runId, {
+        runId: hooks.runId,
+        step: i,
+        count
+      }, () => generateFollowUps(deps.model, [
         ...modelMessages,
         {
           role: "assistant",
           content: turn.text
         }
-      ], count));
+      ], count), (result) => ({
+        followUps: result.followUps.length,
+        inputTokens: result.usage.inputTokens,
+        outputTokens: result.usage.outputTokens,
+        ...result.modelId !== void 0 ? {
+          modelId: result.modelId
+        } : {}
+      })));
       if (generated.followUps.length > 0) {
         followUps = generated.followUps;
       }
@@ -860,6 +944,7 @@ ${buildContextBlock(passages)}`;
         });
         continue;
       }
+      let deciderRef = input.actor.id;
       if (toolType === "action") {
         await hooks.step(`persist:toolcall:${call.id}`, () => deps.store.recordToolCall({
           toolCallId: call.id,
@@ -870,10 +955,12 @@ ${buildContextBlock(passages)}`;
           status: "pending_approval"
         }));
         const decision = await hooks.awaitApproval(call, ctx);
+        deciderRef = decision.executedByRef ?? input.actor.id;
         if (!decision.approved) {
           await hooks.step(`persist:toolreject:${call.id}`, () => deps.store.updateToolCall({
             toolCallId: call.id,
             status: "rejected",
+            executedByRef: deciderRef,
             ...decision.reason !== void 0 ? {
               error: decision.reason
             } : {}
@@ -931,7 +1018,11 @@ ${buildContextBlock(passages)}`;
           };
           output = await hooks.dispatchTool(call, envelope);
         } else {
-          const invocation = hooks.step(`tool:${call.id}`, () => deps.registry.invoke(call.name, call.input, ctx, deps.rolesPolicy));
+          const invocation = hooks.step(`tool:${call.id}`, () => traceToolExecution(hooks.runId, {
+            toolCallId: call.id,
+            toolName: call.name,
+            toolType
+          }, () => deps.registry.invoke(call.name, call.input, ctx, deps.rolesPolicy)));
           output = deps.toolTimeoutMs !== void 0 ? await withToolTimeout(invocation, deps.toolTimeoutMs, call.name) : await invocation;
         }
         const executionMs = Date.now() - startedAt2;
@@ -941,7 +1032,7 @@ ${buildContextBlock(passages)}`;
           output,
           executionMs,
           ...toolType === "action" ? {
-            executedByRef: input.actor.id
+            executedByRef: deciderRef
           } : {}
         }));
         results.push({
@@ -1032,8 +1123,10 @@ __name(runAgentLoop, "runAgentLoop");
 0 && (module.exports = {
   AGENT_ACTOR_DIRECTORY,
   AGENT_ACTOR_RESOLVER,
+  AGENT_APPROVAL_PORT,
   AGENT_ATTACHMENT_STAGING,
   AGENT_DEPS_FACTORY,
+  AGENT_DIAGNOSTIC_EVENTS,
   AGENT_DURABLE_RUNNER,
   AGENT_EMBEDDING_PROVIDER,
   AGENT_GOVERNANCE_QUERIES,
@@ -1047,6 +1140,7 @@ __name(runAgentLoop, "runAgentLoop");
   AGENT_ROLES_POLICY,
   AGENT_RUNNER,
   AGENT_SINK,
+  AGENT_SPAN_EVENTS,
   AGENT_STORE,
   AGENT_TOOL_REGISTRY,
   AgentRegistry,
@@ -1057,6 +1151,7 @@ __name(runAgentLoop, "runAgentLoop");
   ToolInputInvalidError,
   ToolNotFoundError,
   ToolRegistry,
+  agentDiagnosticKey,
   bucketByActor,
   bucketByModel,
   bucketByThread,
@@ -1076,6 +1171,8 @@ __name(runAgentLoop, "runAgentLoop");
   publishAgentToolCall,
   runAgentLoop,
   seedModelPrices,
+  traceLlmTurn,
+  traceToolExecution,
   withToolTimeout
 });
 //# sourceMappingURL=index.cjs.map