@dudousxd/nestjs-agent-core 0.4.0 → 0.6.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,9 @@ var DefaultRolesPolicy = class {
   }
 };
 
+// src/agent-loop.ts
+import { createHash } from "node:crypto";
+
 // src/diagnostics.ts
 import { emit } from "@dudousxd/nestjs-diagnostics";
 function publishAgentRunStarted(payload) {
@@ -358,6 +362,20 @@ function publishAgentRetrieved(payload) {
   emit("agent", "retrieved", payload);
 }
 __name(publishAgentRetrieved, "publishAgentRetrieved");
+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) {
@@ -436,7 +454,7 @@ var ToolTimeoutError = class ToolTimeoutError2 extends Error {
     this.name = "ToolTimeoutError";
   }
 };
-function withTimeout(work, ms, toolName) {
+function withToolTimeout(work, ms, toolName) {
   return new Promise((resolve, reject) => {
     const timer = setTimeout(() => reject(new ToolTimeoutError(toolName, ms)), ms);
     work.then((value) => {
@@ -448,7 +466,7 @@ function withTimeout(work, ms, toolName) {
     });
   });
 }
-__name(withTimeout, "withTimeout");
+__name(withToolTimeout, "withToolTimeout");
 function parseFollowUps(text, count) {
   const source = text.match(/\[[\s\S]*\]/)?.[0] ?? text;
   try {
@@ -553,6 +571,19 @@ async function runAgentLoop(deps, input, hooks) {
       agentName: input.agentName
     } : {}
   });
+  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;
@@ -586,13 +617,25 @@ ${buildContextBlock(passages)}`;
         kind: "step-start"
       }));
     });
-    const tools = await deps.registry.definitionsFor(input.actor, deps.rolesPolicy, deps.toolAllowList);
-    const turn = await hooks.step(`llm:${i}`, () => deps.model.runTurn({
-      system,
-      messages: modelMessages,
-      tools,
-      sink: writer
-    }));
+    let turn;
+    if (hooks.dispatchLlm) {
+      turn = await hooks.dispatchLlm(i, {
+        ...input.agentName !== void 0 ? {
+          agentName: input.agentName
+        } : {},
+        system,
+        messages: modelMessages,
+        actor: input.actor
+      });
+    } else {
+      const tools = await deps.registry.definitionsFor(input.actor, deps.rolesPolicy, deps.toolAllowList);
+      turn = await hooks.step(`llm:${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));
     const toolCallsWithKind = turn.toolCalls.map((call) => ({
@@ -767,6 +810,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,
@@ -777,10 +821,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
             } : {}
@@ -812,18 +858,43 @@ ${buildContextBlock(passages)}`;
           status: "auto_executed"
         }));
       }
-      const startedAt = Date.now();
+      const startedAt2 = Date.now();
       try {
-        const invocation = hooks.step(`tool:${call.id}`, () => deps.registry.invoke(call.name, call.input, ctx, deps.rolesPolicy));
-        const output = deps.toolTimeoutMs !== void 0 ? await withTimeout(invocation, deps.toolTimeoutMs, call.name) : await invocation;
-        const executionMs = Date.now() - startedAt;
+        let output;
+        if (hooks.dispatchTool) {
+          const stepCtx = {
+            actor: input.actor,
+            threadId: input.threadId,
+            runId: hooks.runId,
+            requestId: hooks.runId,
+            ...input.agentName !== void 0 ? {
+              agentName: input.agentName
+            } : {},
+            ...input.pageContext !== void 0 ? {
+              pageContext: input.pageContext
+            } : {}
+          };
+          const envelope = {
+            toolName: call.name,
+            input: call.input,
+            ctx: stepCtx,
+            ...deps.toolTimeoutMs !== void 0 ? {
+              timeoutMs: deps.toolTimeoutMs
+            } : {}
+          };
+          output = await hooks.dispatchTool(call, envelope);
+        } else {
+          const invocation = hooks.step(`tool:${call.id}`, () => 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;
         await hooks.step(`persist:toolexec:${call.id}`, () => deps.store.updateToolCall({
           toolCallId: call.id,
           status: "executed",
           output,
           executionMs,
           ...toolType === "action" ? {
-            executedByRef: input.actor.id
+            executedByRef: deciderRef
           } : {}
         }));
         results.push({
@@ -839,7 +910,10 @@ ${buildContextBlock(passages)}`;
           durationMs: executionMs
         });
       } catch (error) {
-        const executionMs = Date.now() - startedAt;
+        if (hooks.isControlFlowError?.(error) === true) {
+          throw error;
+        }
+        const executionMs = Date.now() - startedAt2;
         const message = error instanceof Error ? error.message : String(error);
         await hooks.step(`persist:toolfail:${call.id}`, () => deps.store.updateToolCall({
           toolCallId: call.id,
@@ -887,6 +961,13 @@ ${buildContextBlock(passages)}`;
   if (thread !== null && (thread.title === "" || thread.title === "New chat")) {
     await hooks.step("persist:title", () => deps.store.setTitle(input.threadId, deriveTitle(input.userText)));
   }
+  await hooks.step("persist:run:end", async () => {
+    await deps.store.recordRunEnd?.({
+      runId: hooks.runId,
+      status: "completed",
+      durationMs: Date.now() - startedAt
+    });
+  });
   await writer.end();
   publishAgentRunFinished({
     runId: hooks.runId,
@@ -903,8 +984,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,
@@ -928,6 +1011,7 @@ export {
   ToolInputInvalidError,
   ToolNotFoundError,
   ToolRegistry,
+  agentDiagnosticKey,
   bucketByActor,
   bucketByModel,
   bucketByThread,
@@ -946,6 +1030,7 @@ export {
   publishAgentRunStarted,
   publishAgentToolCall,
   runAgentLoop,
-  seedModelPrices
+  seedModelPrices,
+  withToolTimeout
 };
 //# sourceMappingURL=index.js.map