@dudousxd/nestjs-agent-core 0.6.0 → 0.8.0

package/dist/index.cjs

@@ -40,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,
@@ -70,6 +71,8 @@ __export(index_exports, {
   publishAgentToolCall: () => publishAgentToolCall,
   runAgentLoop: () => runAgentLoop,
   seedModelPrices: () => seedModelPrices,
+  traceLlmTurn: () => traceLlmTurn,
+  traceToolExecution: () => traceToolExecution,
   withToolTimeout: () => withToolTimeout
 });
 module.exports = __toCommonJS(index_exports);
@@ -400,6 +403,7 @@ 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");
@@ -435,6 +439,12 @@ 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",
@@ -576,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);
@@ -660,9 +703,16 @@ async function runAgentLoop(deps, input, hooks) {
   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}
@@ -702,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));
@@ -744,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;
       }
@@ -957,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;
@@ -1075,6 +1140,7 @@ __name(runAgentLoop, "runAgentLoop");
   AGENT_ROLES_POLICY,
   AGENT_RUNNER,
   AGENT_SINK,
+  AGENT_SPAN_EVENTS,
   AGENT_STORE,
   AGENT_TOOL_REGISTRY,
   AgentRegistry,
@@ -1105,6 +1171,8 @@ __name(runAgentLoop, "runAgentLoop");
   publishAgentToolCall,
   runAgentLoop,
   seedModelPrices,
+  traceLlmTurn,
+  traceToolExecution,
   withToolTimeout
 });
 //# sourceMappingURL=index.cjs.map