@dudousxd/nestjs-agent-core 0.5.0 → 0.7.0

package/dist/index.js

@@ -21,6 +21,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 {
@@ -324,6 +325,10 @@ var DefaultRolesPolicy = class {
   }
 };
 
+// src/agent-loop.ts
+import { createHash } from "node:crypto";
+import { trace } from "@dudousxd/nestjs-diagnostics";
+
 // src/diagnostics.ts
 import { emit } from "@dudousxd/nestjs-diagnostics";
 function publishAgentRunStarted(payload) {
@@ -358,6 +363,26 @@ function publishAgentRetrieved(payload) {
   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) {
@@ -485,6 +510,39 @@ async function generateFollowUps(model, messages, count) {
   };
 }
 __name(generateFollowUps, "generateFollowUps");
+async function spanned(event, runId, payload, run, summarize) {
+  let value;
+  await 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);
@@ -555,21 +613,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 = 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}
@@ -609,12 +676,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));
@@ -651,13 +718,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;
       }
@@ -790,6 +868,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,
@@ -800,10 +879,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
             } : {}
@@ -861,7 +942,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;
@@ -871,7 +956,7 @@ ${buildContextBlock(passages)}`;
           output,
           executionMs,
           ...toolType === "action" ? {
-            executedByRef: input.actor.id
+            executedByRef: deciderRef
           } : {}
         }));
         results.push({
@@ -961,8 +1046,10 @@ __name(runAgentLoop, "runAgentLoop");
 export {
   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,
@@ -976,6 +1063,7 @@ export {
   AGENT_ROLES_POLICY,
   AGENT_RUNNER,
   AGENT_SINK,
+  AGENT_SPAN_EVENTS,
   AGENT_STORE,
   AGENT_TOOL_REGISTRY,
   AgentRegistry,
@@ -986,6 +1074,7 @@ export {
   ToolInputInvalidError,
   ToolNotFoundError,
   ToolRegistry,
+  agentDiagnosticKey,
   bucketByActor,
   bucketByModel,
   bucketByThread,
@@ -1005,6 +1094,8 @@ export {
   publishAgentToolCall,
   runAgentLoop,
   seedModelPrices,
+  traceLlmTurn,
+  traceToolExecution,
   withToolTimeout
 };
 //# sourceMappingURL=index.js.map