plugin-agent-orchestrator 1.0.20 → 1.0.22

package/dist/server/services/TokenTracker.js

@@ -0,0 +1,173 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var TokenTracker_exports = {};
+__export(TokenTracker_exports, {
+  TokenTracker: () => TokenTracker,
+  extractTokenUsage: () => extractTokenUsage
+});
+module.exports = __toCommonJS(TokenTracker_exports);
+const PRICE_PER_1K_INPUT = Number(process.env.ORCHESTRATOR_PRICE_PER_1K_INPUT || 3e-3);
+const PRICE_PER_1K_OUTPUT = Number(process.env.ORCHESTRATOR_PRICE_PER_1K_OUTPUT || 0.015);
+function estimateCost(inputTokens, outputTokens) {
+  return inputTokens / 1e3 * PRICE_PER_1K_INPUT + outputTokens / 1e3 * PRICE_PER_1K_OUTPUT;
+}
+function extractTokenUsage(finalState) {
+  if (!(finalState == null ? void 0 : finalState.messages) || !Array.isArray(finalState.messages)) return null;
+  let totalInput = 0;
+  let totalOutput = 0;
+  let totalAll = 0;
+  for (const msg of finalState.messages) {
+    if (msg == null ? void 0 : msg.usage_metadata) {
+      const um = msg.usage_metadata;
+      totalInput += um.input_tokens || 0;
+      totalOutput += um.output_tokens || 0;
+      totalAll += um.total_tokens || 0;
+    }
+  }
+  if (totalAll === 0) return null;
+  return {
+    inputTokens: totalInput,
+    outputTokens: totalOutput,
+    totalTokens: totalAll,
+    cost: estimateCost(totalInput, totalOutput)
+  };
+}
+class TokenTracker {
+  constructor(plugin) {
+    this.plugin = plugin;
+  }
+  get db() {
+    return this.plugin.db;
+  }
+  get app() {
+    return this.plugin.app;
+  }
+  /**
+   * Track token usage for a single execution span.
+   * Updates the span record and accumulates into the parent run.
+   *
+   * @param spanId - The agentExecutionSpans.id to update
+   * @param usage - Parsed token usage from extractTokenUsage()
+   * @param runId - Optional agentLoopRuns.id to accumulate totals
+   */
+  async trackSpan(spanId, usage, runId) {
+    var _a, _b;
+    if (!spanId) return;
+    try {
+      const repo = this.db.getRepository("agentExecutionSpans");
+      if (!repo) return;
+      await repo.update({
+        filterByTk: spanId,
+        values: {
+          inputTokens: usage.inputTokens,
+          outputTokens: usage.outputTokens,
+          totalTokens: usage.totalTokens,
+          cost: usage.cost
+        }
+      });
+      if (runId != null) {
+        await this.accumulateToRun(runId);
+      }
+    } catch (e) {
+      (_b = (_a = this.app.log) == null ? void 0 : _a.warn) == null ? void 0 : _b.call(_a, "[TokenTracker] Failed to track span tokens", e);
+    }
+  }
+  /**
+   * Recalculate total token/cost for a run by summing all its spans.
+   */
+  async accumulateToRun(runId) {
+    var _a, _b, _c, _d, _e;
+    try {
+      const spansRepo = this.db.getRepository("agentExecutionSpans");
+      if (!spansRepo) return;
+      const spans = await spansRepo.find({
+        filter: { "metadata.agentLoopRunId": String(runId) }
+      });
+      let totalInput = 0;
+      let totalOutput = 0;
+      let totalCost = 0;
+      for (const span of spans) {
+        totalInput += Number(((_a = span.get) == null ? void 0 : _a.call(span, "inputTokens")) || span.inputTokens || 0);
+        totalOutput += Number(((_b = span.get) == null ? void 0 : _b.call(span, "outputTokens")) || span.outputTokens || 0);
+        totalCost += Number(((_c = span.get) == null ? void 0 : _c.call(span, "cost")) || span.cost || 0);
+      }
+      const runsRepo = this.db.getRepository("agentLoopRuns");
+      if (!runsRepo) return;
+      await runsRepo.update({
+        filterByTk: runId,
+        values: {
+          totalInputTokens: totalInput,
+          totalOutputTokens: totalOutput,
+          totalTokens: totalInput + totalOutput,
+          totalCost
+        }
+      });
+    } catch (e) {
+      (_e = (_d = this.app.log) == null ? void 0 : _d.warn) == null ? void 0 : _e.call(_d, "[TokenTracker] Failed to accumulate run totals", e);
+    }
+  }
+  /**
+   * Check if a run has exceeded its budget limits.
+   */
+  async checkBudget(runId) {
+    var _a, _b, _c, _d, _e, _f;
+    try {
+      const repo = this.db.getRepository("agentLoopRuns");
+      if (!repo) return { allowed: true };
+      const run = await repo.findOne({ filter: { id: runId } });
+      if (!run) return { allowed: true };
+      const budgetMaxTokens = Number(((_a = run.get) == null ? void 0 : _a.call(run, "budgetMaxTokens")) ?? 0);
+      const budgetMaxCost = Number(((_b = run.get) == null ? void 0 : _b.call(run, "budgetMaxCost")) ?? 0);
+      if (budgetMaxTokens <= 0 && budgetMaxCost <= 0) return { allowed: true };
+      const totalTokens = Number(((_c = run.get) == null ? void 0 : _c.call(run, "totalTokens")) || 0);
+      const totalCost = Number(((_d = run.get) == null ? void 0 : _d.call(run, "totalCost")) || 0);
+      if (budgetMaxTokens > 0 && totalTokens >= budgetMaxTokens) {
+        return {
+          allowed: false,
+          reason: `Budget exceeded: ${totalTokens}/${budgetMaxTokens} tokens used. Maximum allowed tokens for this run: ${budgetMaxTokens}.`
+        };
+      }
+      if (budgetMaxCost > 0 && totalCost >= budgetMaxCost) {
+        return {
+          allowed: false,
+          reason: `Budget exceeded: $${totalCost.toFixed(4)}/$${budgetMaxCost.toFixed(
+            4
+          )} spent. Maximum allowed cost for this run: $${budgetMaxCost}.`
+        };
+      }
+      return { allowed: true };
+    } catch (e) {
+      (_f = (_e = this.app.log) == null ? void 0 : _e.warn) == null ? void 0 : _f.call(_e, "[TokenTracker] Budget check failed, allowing", e);
+      return { allowed: true };
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TokenTracker,
+  extractTokenUsage
+});

package/dist/server/tools/agent-loop.js

@@ -31,6 +31,7 @@ __export(agent_loop_exports, {
 module.exports = __toCommonJS(agent_loop_exports);
 var import_zod = require("zod");
 var import_ExecutionSpanService = require("../services/ExecutionSpanService");
+var import_ctx_utils = require("../utils/ctx-utils");
 const stepSchema = import_zod.z.object({
   id: import_zod.z.string().optional(),
   key: import_zod.z.string().optional(),
@@ -58,46 +59,6 @@ function toolResult(status, payload) {
     content: typeof payload === "string" ? payload : JSON.stringify(payload)
   };
 }
-function valuesFromCtx(ctx) {
-  var _a, _b;
-  return ((_b = (_a = ctx == null ? void 0 : ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) || {};
-}
-function currentUserId(ctx) {
-  var _a, _b, _c, _d;
-  return ((_b = (_a = ctx == null ? void 0 : ctx.state) == null ? void 0 : _a.currentUser) == null ? void 0 : _b.id) || ((_d = (_c = ctx == null ? void 0 : ctx.auth) == null ? void 0 : _c.user) == null ? void 0 : _d.id);
-}
-function resolveSessionId(ctx, args) {
-  var _a, _b, _c;
-  const values = valuesFromCtx(ctx);
-  return (args == null ? void 0 : args.sessionId) || values.sessionId || ((_b = (_a = ctx == null ? void 0 : ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.sessionId) || ((_c = ctx == null ? void 0 : ctx.state) == null ? void 0 : _c.sessionId);
-}
-function resolveMessageId(ctx, args) {
-  var _a, _b;
-  const values = valuesFromCtx(ctx);
-  return (args == null ? void 0 : args.messageId) || values.messageId || ((_b = (_a = ctx == null ? void 0 : ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.messageId);
-}
-function normalizeEmployeeUsername(raw) {
-  if (!raw) return null;
-  if (typeof raw === "string") return raw;
-  return raw.username || raw.aiEmployeeUsername || raw.name || null;
-}
-async function resolveLeaderUsername(ctx, plugin, args) {
-  var _a, _b, _c, _d, _e, _f;
-  const values = valuesFromCtx(ctx);
-  const direct = normalizeEmployeeUsername(
-    (args == null ? void 0 : args.leaderUsername) || (ctx == null ? void 0 : ctx._currentAIEmployee) || ((_a = ctx == null ? void 0 : ctx.state) == null ? void 0 : _a.currentAIEmployee) || ((_c = (_b = ctx == null ? void 0 : ctx.runtime) == null ? void 0 : _b.context) == null ? void 0 : _c.currentAIEmployee) || values.aiEmployee
-  );
-  if (direct) return direct;
-  const sessionId = resolveSessionId(ctx, args);
-  if (!sessionId) return void 0;
-  try {
-    const repo = ((_e = (_d = ctx == null ? void 0 : ctx.db) == null ? void 0 : _d.getRepository) == null ? void 0 : _e.call(_d, "aiConversations")) || plugin.db.getRepository("aiConversations");
-    const conversation = await repo.findOne({ filter: { sessionId } });
-    return normalizeEmployeeUsername((conversation == null ? void 0 : conversation.aiEmployeeUsername) || ((_f = conversation == null ? void 0 : conversation.get) == null ? void 0 : _f.call(conversation, "aiEmployeeUsername")));
-  } catch {
-    return void 0;
-  }
-}
 function setLoopTraceContext(ctx, snapshot, step) {
   if (!ctx || !(snapshot == null ? void 0 : snapshot.run)) return;
   const existing = (0, import_ExecutionSpanService.getOrchestratorTraceContext)(ctx) || {};
@@ -136,19 +97,20 @@ function createAgentLoopTools(plugin, service) {
         })
       },
       invoke: async (ctx, args) => {
+        var _a;
         try {
-          const leaderUsername = await resolveLeaderUsername(ctx, plugin, args);
+          const leaderUsername = await (0, import_ctx_utils.resolveLeaderUsername)(ctx, plugin, args);
           const snapshot = await service.createRun({
             goal: args.goal,
             leaderUsername,
-            sessionId: resolveSessionId(ctx, args),
-            messageId: resolveMessageId(ctx, args),
-            userId: currentUserId(ctx),
+            sessionId: (0, import_ctx_utils.resolveSessionId)(ctx, args),
+            messageId: (0, import_ctx_utils.resolveMessageId)(ctx, args),
+            userId: (0, import_ctx_utils.currentUserId)(ctx),
             policy: args.policy,
             metadata: args.metadata,
             plan: planFromArgs(args.plan)
           });
-          setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+          setLoopTraceContext(ctx, snapshot, (_a = snapshot.nextSteps) == null ? void 0 : _a[0]);
           return toolResult("success", snapshot);
         } catch (error) {
           return toolResult("error", (error == null ? void 0 : error.message) || String(error));
@@ -161,19 +123,20 @@ function createAgentLoopTools(plugin, service) {
       defaultPermission: "ALLOW",
       introduction: {
         title: "Agent Loop - Status",
-        about: "Read the current run, steps, and next executable step."
+        about: "Read the current run, steps, and next executable steps."
       },
       definition: {
         name: "agent_loop_status",
-        description: "Fetch an agent loop run. Call this before executing the next step so subsequent skill/tool calls can be linked to the current loop step.",
+        description: "Fetch an agent loop run. Call this to check which steps are ready for execution.",
         schema: import_zod.z.object({
           runId: import_zod.z.union([import_zod.z.string(), import_zod.z.number()])
         })
       },
       invoke: async (ctx, args) => {
+        var _a;
         try {
           const snapshot = await service.getRunSnapshot(args.runId);
-          setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+          setLoopTraceContext(ctx, snapshot, (_a = snapshot.nextSteps) == null ? void 0 : _a[0]);
           return toolResult("success", snapshot);
         } catch (error) {
           return toolResult("error", (error == null ? void 0 : error.message) || String(error));
@@ -203,34 +166,35 @@ function createAgentLoopTools(plugin, service) {
         })
       },
       invoke: async (ctx, args) => {
+        var _a, _b, _c;
         try {
           let snapshot;
           if (args.status === "running") {
             snapshot = await service.startStep(args.stepId, {
-              userId: currentUserId(ctx),
+              userId: (0, import_ctx_utils.currentUserId)(ctx),
               agentExecutionSpanId: args.agentExecutionSpanId
             });
             const currentStep = snapshot.steps.find((step) => String(step.id) === String(args.stepId));
             setLoopTraceContext(ctx, snapshot, currentStep);
           } else if (args.status === "succeeded") {
             snapshot = await service.completeStep(args.stepId, args.output === void 0 ? {} : args.output, {
-              userId: currentUserId(ctx),
+              userId: (0, import_ctx_utils.currentUserId)(ctx),
               skillExecutionId: args.skillExecutionId,
               agentExecutionSpanId: args.agentExecutionSpanId,
               metadata: args.metadata
             });
-            setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+            setLoopTraceContext(ctx, snapshot, (_a = snapshot.nextSteps) == null ? void 0 : _a[0]);
           } else if (args.status === "failed") {
             snapshot = await service.failStep(args.stepId, args.error || "Step failed.", {
-              userId: currentUserId(ctx),
+              userId: (0, import_ctx_utils.currentUserId)(ctx),
               metadata: args.metadata
             });
-            setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+            setLoopTraceContext(ctx, snapshot, (_b = snapshot.nextSteps) == null ? void 0 : _b[0]);
           } else {
             snapshot = await service.skipStep(args.stepId, args.reason || "Skipped.", {
-              userId: currentUserId(ctx)
+              userId: (0, import_ctx_utils.currentUserId)(ctx)
             });
-            setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+            setLoopTraceContext(ctx, snapshot, (_c = snapshot.nextSteps) == null ? void 0 : _c[0]);
           }
           return toolResult("success", snapshot);
         } catch (error) {
@@ -257,14 +221,15 @@ function createAgentLoopTools(plugin, service) {
         })
       },
       invoke: async (ctx, args) => {
+        var _a;
         try {
           await service.replan(args.runId, args.plan, {
             reason: args.reason,
             mode: args.mode || "replace_pending",
-            userId: currentUserId(ctx)
+            userId: (0, import_ctx_utils.currentUserId)(ctx)
           });
           const snapshot = await service.getRunSnapshot(args.runId);
-          setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+          setLoopTraceContext(ctx, snapshot, (_a = snapshot.nextSteps) == null ? void 0 : _a[0]);
           return toolResult("success", snapshot);
         } catch (error) {
           return toolResult("error", (error == null ? void 0 : error.message) || String(error));
@@ -289,12 +254,13 @@ function createAgentLoopTools(plugin, service) {
         })
       },
       invoke: async (ctx, args) => {
+        var _a;
         try {
           const snapshot = await service.requestApproval(args.stepId, args.approval || {}, {
             reason: args.reason,
-            userId: currentUserId(ctx)
+            userId: (0, import_ctx_utils.currentUserId)(ctx)
           });
-          setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+          setLoopTraceContext(ctx, snapshot, (_a = snapshot.nextSteps) == null ? void 0 : _a[0]);
           return toolResult("success", snapshot);
         } catch (error) {
           return toolResult("error", (error == null ? void 0 : error.message) || String(error));
@@ -320,15 +286,16 @@ function createAgentLoopTools(plugin, service) {
         })
       },
       invoke: async (ctx, args) => {
+        var _a;
         try {
           const snapshot = await service.resumeRun(args.runId, {
             stepId: args.stepId,
             approved: args.approved,
             editedInput: args.editedInput,
-            userId: currentUserId(ctx),
+            userId: (0, import_ctx_utils.currentUserId)(ctx),
             ctx
           });
-          setLoopTraceContext(ctx, snapshot, snapshot.nextStep);
+          setLoopTraceContext(ctx, snapshot, (_a = snapshot.nextSteps) == null ? void 0 : _a[0]);
           return toolResult("success", snapshot);
         } catch (error) {
           return toolResult("error", (error == null ? void 0 : error.message) || String(error));
@@ -360,7 +327,7 @@ function createAgentLoopTools(plugin, service) {
             status: args.status || "succeeded",
             summary: args.summary,
             evidence: args.evidence,
-            userId: currentUserId(ctx)
+            userId: (0, import_ctx_utils.currentUserId)(ctx)
           });
           setLoopTraceContext(ctx, snapshot);
           return toolResult("success", snapshot);
@@ -389,7 +356,7 @@ function createAgentLoopTools(plugin, service) {
         try {
           const snapshot = await service.cancelRun(args.runId, {
             reason: args.reason,
-            userId: currentUserId(ctx)
+            userId: (0, import_ctx_utils.currentUserId)(ctx)
           });
           setLoopTraceContext(ctx, snapshot);
           return toolResult("success", snapshot);

package/dist/server/tools/delegate-task.js

@@ -33,6 +33,8 @@ module.exports = __toCommonJS(delegate_task_exports);
 var import_zod = require("zod");
 var import_crypto = require("crypto");
 var import_ExecutionSpanService = require("../services/ExecutionSpanService");
+var import_ctx_utils = require("../utils/ctx-utils");
+var import_logging = require("../utils/logging");
 const ORCHESTRATOR_DEPTH_KEY = "__orchestratorDepth";
 const ORCHESTRATOR_PATH_KEY = "__orchestratorPath";
 const MAX_DISPATCH_CONCURRENCY = 5;
@@ -232,7 +234,7 @@ ${subAgentList}`
           leaderUsername,
           subAgentUsername: reportedSub,
           toolName,
-          task: truncateText(args.tasks ?? [], 2e3),
+          task: (0, import_ctx_utils.trimText)(args.tasks ?? [], 2e3),
           result: "",
           status: "error",
           depth: ctx[ORCHESTRATOR_DEPTH_KEY] ?? 0,
@@ -251,7 +253,7 @@ ${subAgentList}`
           leaderUsername,
           subAgentUsername: reportedSub,
           toolName,
-          task: truncateText(args.tasks ?? [], 2e3),
+          task: (0, import_ctx_utils.trimText)(args.tasks ?? [], 2e3),
           result: "",
           status: "error",
           depth: ctx[ORCHESTRATOR_DEPTH_KEY] ?? 0,
@@ -324,25 +326,11 @@ ${subAgentList}`
     }
   };
 }
-function captureCtxSnapshot(ctx) {
-  var _a, _b, _c, _d;
-  let userId;
-  try {
-    userId = ((_b = (_a = ctx.auth) == null ? void 0 : _a.user) == null ? void 0 : _b.id) || ((_d = (_c = ctx.state) == null ? void 0 : _c.currentUser) == null ? void 0 : _d.id);
-  } catch {
-  }
-  return { userId };
-}
-function normalizeEmployeeUsername(raw) {
-  if (!raw) return null;
-  if (typeof raw === "string") return raw;
-  return raw.username || raw.aiEmployeeUsername || raw.name || null;
-}
 async function resolveCallingEmployee(ctx, plugin) {
   var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
   const values = ((_b = (_a = ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) || {};
   const raw = ctx._currentAIEmployee || ((_c = ctx.state) == null ? void 0 : _c.currentAIEmployee) || ((_e = (_d = ctx.runtime) == null ? void 0 : _d.context) == null ? void 0 : _e.currentAIEmployee) || values.aiEmployee;
-  const direct = normalizeEmployeeUsername(raw);
+  const direct = (0, import_ctx_utils.normalizeEmployeeUsername)(raw);
   if (direct) return direct;
   const sessionId = values.sessionId || ((_g = (_f = ctx.action) == null ? void 0 : _f.params) == null ? void 0 : _g.sessionId);
   if (!sessionId) return null;
@@ -351,20 +339,12 @@ async function resolveCallingEmployee(ctx, plugin) {
     const conversation = await repo.findOne({
       filter: { sessionId }
     });
-    return normalizeEmployeeUsername((conversation == null ? void 0 : conversation.aiEmployeeUsername) || ((_j = conversation == null ? void 0 : conversation.get) == null ? void 0 : _j.call(conversation, "aiEmployeeUsername")));
+    return (0, import_ctx_utils.normalizeEmployeeUsername)((conversation == null ? void 0 : conversation.aiEmployeeUsername) || ((_j = conversation == null ? void 0 : conversation.get) == null ? void 0 : _j.call(conversation, "aiEmployeeUsername")));
   } catch (e) {
     plugin.app.log.warn(`[AgentOrchestrator] Failed to resolve AI employee for session "${sessionId}"`, e);
     return null;
   }
 }
-function truncateText(value, maxLen) {
-  const text = typeof value === "string" ? value : value == null ? "" : JSON.stringify(value);
-  return text.length > maxLen ? `${text.slice(0, maxLen)}
-...[truncated]` : text;
-}
-function nowIso() {
-  return (/* @__PURE__ */ new Date()).toISOString();
-}
 function hasModelSettings(value) {
   return Boolean((value == null ? void 0 : value.llmService) && (value == null ? void 0 : value.model));
 }
@@ -568,7 +548,7 @@ async function invokeDelegateTask(ctx, plugin, options) {
     rootRunId: providedRootRunId,
     parentSpanId: providedParentSpanId
   } = options;
-  const ctxSnapshot = captureCtxSnapshot(ctx);
+  const ctxSnapshot = (0, import_ctx_utils.captureCtxSnapshot)(ctx);
   const currentDepth = ctx[ORCHESTRATOR_DEPTH_KEY] ?? 0;
   const currentPath = ctx[ORCHESTRATOR_PATH_KEY] ?? [leaderUsername];
   if (currentPath.includes(subAgentUsername)) {
@@ -637,53 +617,15 @@ async function invokeDelegateTask(ctx, plugin, options) {
   });
 }
 async function logDelegation(ctx, plugin, data) {
-  var _a, _b, _c, _d, _e, _f;
-  try {
-    const logsRepo = plugin.db.getRepository("orchestratorLogs");
-    if (!logsRepo) {
-      plugin.app.log.warn("[AgentOrchestrator] orchestratorLogs repository not found \u2014 skipping log");
-      return;
-    }
-    let userId = (_a = data.snapshot) == null ? void 0 : _a.userId;
-    if (userId == null) {
-      try {
-        userId = ((_c = (_b = ctx.auth) == null ? void 0 : _b.user) == null ? void 0 : _c.id) || ((_e = (_d = ctx.state) == null ? void 0 : _d.currentUser) == null ? void 0 : _e.id);
-      } catch {
-      }
-    }
-    const values = {
-      leaderUsername: data.leaderUsername,
-      subAgentUsername: data.subAgentUsername,
-      toolName: data.toolName,
-      task: truncateText(data.task, 1e4),
-      context: truncateText(data.context || "", 1e4),
-      result: truncateText(data.result || "", 5e4),
-      status: data.status,
-      depth: data.depth,
-      durationMs: data.durationMs,
-      error: truncateText(data.error || "", 1e4),
-      trace: data.trace || [],
-      messages: data.messages || [],
-      userId,
-      updatedAt: /* @__PURE__ */ new Date()
-    };
-    if (data.id) {
-      await logsRepo.update({
-        filterByTk: data.id,
-        values
-      });
-      return { id: data.id };
+  var _a, _b, _c, _d, _e;
+  let userId = (_a = data.snapshot) == null ? void 0 : _a.userId;
+  if (userId == null) {
+    try {
+      userId = ((_c = (_b = ctx.auth) == null ? void 0 : _b.user) == null ? void 0 : _c.id) || ((_e = (_d = ctx.state) == null ? void 0 : _d.currentUser) == null ? void 0 : _e.id);
+    } catch {
     }
-    const record = await logsRepo.create({
-      values: {
-        ...values,
-        createdAt: /* @__PURE__ */ new Date()
-      }
-    });
-    return ((_f = record == null ? void 0 : record.toJSON) == null ? void 0 : _f.call(record)) || record;
-  } catch (e) {
-    plugin.app.log.warn("[AgentOrchestrator] Failed to log delegation event", e);
   }
+  return (0, import_logging.logDelegation)(ctx, plugin, { ...data, userId });
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/server/tools/orchestrator-plan.js

@@ -30,6 +30,7 @@ __export(orchestrator_plan_exports, {
 });
 module.exports = __toCommonJS(orchestrator_plan_exports);
 var import_zod = require("zod");
+var import_ctx_utils = require("../utils/ctx-utils");
 const stepSchema = import_zod.z.object({
   id: import_zod.z.string().optional(),
   key: import_zod.z.string().optional(),
@@ -58,46 +59,6 @@ function toolResult(status, payload) {
     content: typeof payload === "string" ? payload : JSON.stringify(payload)
   };
 }
-function valuesFromCtx(ctx) {
-  var _a, _b, _c;
-  return ((_b = (_a = ctx == null ? void 0 : ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) || ((_c = ctx == null ? void 0 : ctx.request) == null ? void 0 : _c.body) || {};
-}
-function currentUserId(ctx) {
-  var _a, _b, _c, _d;
-  return ((_b = (_a = ctx == null ? void 0 : ctx.state) == null ? void 0 : _a.currentUser) == null ? void 0 : _b.id) || ((_d = (_c = ctx == null ? void 0 : ctx.auth) == null ? void 0 : _c.user) == null ? void 0 : _d.id);
-}
-function resolveSessionId(ctx, args) {
-  var _a, _b, _c;
-  const values = valuesFromCtx(ctx);
-  return (args == null ? void 0 : args.sessionId) || values.sessionId || ((_b = (_a = ctx == null ? void 0 : ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.sessionId) || ((_c = ctx == null ? void 0 : ctx.state) == null ? void 0 : _c.sessionId);
-}
-function resolveMessageId(ctx, args) {
-  var _a, _b;
-  const values = valuesFromCtx(ctx);
-  return (args == null ? void 0 : args.messageId) || values.messageId || ((_b = (_a = ctx == null ? void 0 : ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.messageId);
-}
-function normalizeEmployeeUsername(raw) {
-  if (!raw) return null;
-  if (typeof raw === "string") return raw;
-  return raw.username || raw.aiEmployeeUsername || raw.name || null;
-}
-async function resolveLeaderUsername(ctx, plugin, args) {
-  var _a, _b, _c, _d, _e, _f;
-  const values = valuesFromCtx(ctx);
-  const direct = normalizeEmployeeUsername(
-    (args == null ? void 0 : args.leaderUsername) || (ctx == null ? void 0 : ctx._currentAIEmployee) || ((_a = ctx == null ? void 0 : ctx.state) == null ? void 0 : _a.currentAIEmployee) || ((_c = (_b = ctx == null ? void 0 : ctx.runtime) == null ? void 0 : _b.context) == null ? void 0 : _c.currentAIEmployee) || values.aiEmployee
-  );
-  if (direct) return direct;
-  const sessionId = resolveSessionId(ctx, args);
-  if (!sessionId) return void 0;
-  try {
-    const repo = ((_e = (_d = ctx == null ? void 0 : ctx.db) == null ? void 0 : _d.getRepository) == null ? void 0 : _e.call(_d, "aiConversations")) || plugin.db.getRepository("aiConversations");
-    const conversation = await repo.findOne({ filter: { sessionId } });
-    return normalizeEmployeeUsername((conversation == null ? void 0 : conversation.aiEmployeeUsername) || ((_f = conversation == null ? void 0 : conversation.get) == null ? void 0 : _f.call(conversation, "aiEmployeeUsername")));
-  } catch {
-    return void 0;
-  }
-}
 function summarizePlan(steps) {
   return (steps || []).map((step) => ({
     id: step.id,
@@ -159,21 +120,23 @@ function createOrchestratorPlanTools(plugin, service) {
           plannerModel: import_zod.z.string().optional(),
           policy: policySchema,
           metadata: import_zod.z.any().optional(),
-          plan: import_zod.z.array(stepSchema).optional().describe("Draft plan steps. Tool/sub_agent steps should include target. Dependencies must reference planKey values.")
+          plan: import_zod.z.array(stepSchema).optional().describe(
+            "Draft plan steps. Tool/sub_agent steps should include target. Dependencies must reference planKey values."
+          )
         })
       },
       invoke: async (ctx, args) => {
         try {
-          const leaderUsername = await resolveLeaderUsername(ctx, plugin, args);
+          const leaderUsername = await (0, import_ctx_utils.resolveLeaderUsername)(ctx, plugin, args);
           const targetAgent = inferTargetAgent(args);
           const harnessTag = await resolveHarnessTag(plugin, leaderUsername, targetAgent, args);
           const detail = await service.planGoal({
             goal: args.goal,
             runId: args.runId,
             leaderUsername,
-            sessionId: resolveSessionId(ctx, args),
-            messageId: resolveMessageId(ctx, args),
-            userId: currentUserId(ctx),
+            sessionId: (0, import_ctx_utils.resolveSessionId)(ctx, args),
+            messageId: (0, import_ctx_utils.resolveMessageId)(ctx, args),
+            userId: (0, import_ctx_utils.currentUserId)(ctx),
             policy: args.policy,
             metadata: args.metadata,
             plan: Array.isArray(args.plan) ? args.plan : void 0,
@@ -220,7 +183,7 @@ function createOrchestratorPlanTools(plugin, service) {
       invoke: async (ctx, args) => {
         try {
           const detail = await service.approvePlanAndExecute(args.runId, {
-            userId: currentUserId(ctx),
+            userId: (0, import_ctx_utils.currentUserId)(ctx),
             ctx,
             reason: args.reason
           });
@@ -275,7 +238,7 @@ function createOrchestratorPlanTools(plugin, service) {
             "success",
             await service.cancelRun(args.runId, {
               reason: args.reason,
-              userId: currentUserId(ctx)
+              userId: (0, import_ctx_utils.currentUserId)(ctx)
             })
           );
         } catch (error) {

package/dist/server/types.js

@@ -0,0 +1,24 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var types_exports = {};
+module.exports = __toCommonJS(types_exports);