plugin-agent-orchestrator 1.0.27 → 1.0.32

package/dist/server/plugin.js

@@ -42,22 +42,119 @@ __export(plugin_exports, {
 module.exports = __toCommonJS(plugin_exports);
 var import_server = require("@nocobase/server");
 var import_path = __toESM(require("path"));
-var import_delegate_task = require("./tools/delegate-task");
 var import_external_rag_search = require("./tools/external-rag-search");
-var import_orchestrator_plan = require("./tools/orchestrator-plan");
 var import_tracing = require("./resources/tracing");
-var import_agent_loop = require("./resources/agent-loop");
-var import_RunEventBus = require("./services/RunEventBus");
+var import_agent_monitor = require("./resources/agent-monitor");
 var import_plugin = __toESM(require("./skill-hub/plugin"));
-var import_AgentLoopService = require("./services/AgentLoopService");
+var import_NativeSubAgentObserver = require("./services/NativeSubAgentObserver");
 var import_ctx_utils = require("./utils/ctx-utils");
 var import_ai_manager = require("./utils/ai-manager");
+function normalizeOptionalString(value) {
+  return typeof value === "string" ? value.trim() : "";
+}
+function readModelValue(record, key) {
+  const model = record;
+  return typeof (model == null ? void 0 : model.get) === "function" ? model.get(key) : model == null ? void 0 : model[key];
+}
+function buildAgentMemoryContextKey(values) {
+  const userPart = values.scope === "public" ? "public" : String(values.userId || "");
+  const agentPart = values.aiEmployeeUsername || "*";
+  return `${values.scope}:${userPart}:${agentPart}`;
+}
+async function validateAgentMemoryContextValues(ctx) {
+  var _a, _b, _c, _d, _e, _f, _g, _h;
+  const actionName = (_a = ctx.action) == null ? void 0 : _a.actionName;
+  const values = ((_c = (_b = ctx.action) == null ? void 0 : _b.params) == null ? void 0 : _c.values) || {};
+  let nextValues = values;
+  let currentId = (_e = (_d = ctx.action) == null ? void 0 : _d.params) == null ? void 0 : _e.filterByTk;
+  if (actionName === "update" && ((_g = (_f = ctx.action) == null ? void 0 : _f.params) == null ? void 0 : _g.filterByTk)) {
+    const existing = await ctx.db.getRepository("agentMemoryContexts").findOne({
+      filter: { id: ctx.action.params.filterByTk }
+    });
+    currentId = readModelValue(existing, "id") || currentId;
+    nextValues = {
+      ...((_h = existing == null ? void 0 : existing.toJSON) == null ? void 0 : _h.call(existing)) || existing || {},
+      ...values
+    };
+  }
+  const scope = normalizeOptionalString(nextValues.scope);
+  const userId = nextValues.userId;
+  const aiEmployeeUsername = normalizeOptionalString(nextValues.aiEmployeeUsername);
+  if (!["public", "user", "agent_user"].includes(scope)) {
+    ctx.throw(400, "scope must be one of: public, user, agent_user.");
+    return;
+  }
+  if (scope === "public" && userId != null && userId !== "") {
+    ctx.throw(400, 'scope="public" requires userId to be empty.');
+    return;
+  }
+  if ((scope === "user" || scope === "agent_user") && (userId == null || userId === "")) {
+    ctx.throw(400, `scope="${scope}" requires userId.`);
+    return;
+  }
+  if (scope === "agent_user" && !aiEmployeeUsername) {
+    ctx.throw(400, 'scope="agent_user" requires aiEmployeeUsername.');
+    return;
+  }
+  const normalizedValues = {
+    ...values,
+    scope,
+    userId: scope === "public" ? null : userId,
+    aiEmployeeUsername,
+    contextKey: buildAgentMemoryContextKey({
+      scope,
+      userId: scope === "public" ? null : userId,
+      aiEmployeeUsername
+    })
+  };
+  const repo = ctx.db.getRepository("agentMemoryContexts");
+  const duplicate = await repo.findOne({
+    filter: { contextKey: normalizedValues.contextKey }
+  }) || await repo.findOne({
+    filter: {
+      scope,
+      userId: normalizedValues.userId,
+      aiEmployeeUsername
+    }
+  });
+  const duplicateId = readModelValue(duplicate, "id");
+  if (duplicateId && String(duplicateId) !== String(currentId || "")) {
+    ctx.throw(400, "An agent memory context already exists for this scope, user, and AI employee.");
+    return;
+  }
+  ctx.action.params.values = normalizedValues;
+}
+async function resolveTracingRetentionDays(plugin) {
+  var _a, _b;
+  const envDays = Number(process.env.ORCHESTRATOR_LOG_RETENTION_DAYS);
+  if (Number.isFinite(envDays) && envDays > 0) return envDays;
+  try {
+    const defaultProfile = await plugin.db.getRepository("agentHarnessProfiles").findOne({
+      filter: {
+        tag: "default",
+        enabled: true
+      }
+    });
+    const settings = (0, import_ctx_utils.asObject)(readModelValue(defaultProfile, "settings"));
+    const profileDays = Number(settings.tracingRetentionDays);
+    if (Number.isFinite(profileDays) && profileDays > 0) {
+      return profileDays;
+    }
+  } catch (error) {
+    (_b = (_a = plugin.app.logger) == null ? void 0 : _a.warn) == null ? void 0 : _b.call(_a, "[AgentOrchestrator] Failed to load tracing retention policy", error);
+  }
+  return 30;
+}
 class PluginAgentOrchestratorServer extends import_server.Plugin {
   skillHub;
-  agentLoopService;
+  nativeObserver;
+  installNativeObserver = () => {
+    var _a;
+    (_a = this.nativeObserver) == null ? void 0 : _a.install();
+  };
   async afterAdd() {
     this.skillHub = new import_plugin.default(this);
-    this.agentLoopService = new import_AgentLoopService.AgentLoopService(this);
+    this.nativeObserver = new import_NativeSubAgentObserver.NativeSubAgentObserver(this);
   }
   async beforeLoad() {
     this.db.import({ directory: import_path.default.resolve(__dirname, "collections") });
@@ -68,17 +165,18 @@ class PluginAgentOrchestratorServer extends import_server.Plugin {
     });
   }
   async load() {
+    var _a, _b;
     await this.skillHub.load();
     this.app.acl.registerSnippet({
       name: `pm.${this.name}`,
       actions: [
         "orchestratorConfig:*",
         "orchestratorTracing:*",
-        "agentLoops:*",
+        "agentMonitor:*",
+        "agentMemoryContexts:*",
         "agentLoopRuns:*",
         "agentLoopSteps:*",
         "agentLoopEvents:*",
-        "agentLoopEventsStream:*",
         "agentHarnessProfiles:*",
         "agentExecutionSpans:*",
         "skillDefinitions:*",
@@ -97,6 +195,7 @@ class PluginAgentOrchestratorServer extends import_server.Plugin {
     this.app.acl.allow("skillHub", "test", "loggedIn");
     this.app.acl.allow("skillHub", "download", "loggedIn");
     this.app.acl.allow("skillHub", "listTemplates", "loggedIn");
+    this.app.acl.allow("agentMonitor", ["list", "get"], "loggedIn");
     this.app.resourceManager.use(
       async (ctx, next) => {
         const { resourceName, actionName } = ctx.action || {};
@@ -111,78 +210,27 @@ class PluginAgentOrchestratorServer extends import_server.Plugin {
       },
       { tag: "orchestrator-skill-executions-scope", after: "acl" }
     );
+    this.app.resourceManager.use(
+      async (ctx, next) => {
+        const { resourceName, actionName } = ctx.action || {};
+        if (resourceName === "agentMemoryContexts" && (actionName === "create" || actionName === "update")) {
+          await validateAgentMemoryContextValues(ctx);
+        }
+        await next();
+      },
+      { tag: "orchestrator-agent-memory-context-policy", after: "acl" }
+    );
     const toolsManager = (0, import_ai_manager.getAIToolsManager)(this.app);
-    toolsManager.registerTools((0, import_orchestrator_plan.createOrchestratorPlanTools)(this, this.agentLoopService));
     toolsManager.registerTools((0, import_external_rag_search.createExternalRagSearchTool)(this));
-    toolsManager.registerDynamicTools((0, import_delegate_task.createDelegateToolsProvider)(this));
-    (0, import_agent_loop.registerAgentLoopResource)(this, this.agentLoopService);
-    this.app.resource({
-      name: "agentLoopEventsStream",
-      actions: {
-        async stream(ctx, next) {
-          var _a, _b, _c;
-          const runId = ((_a = ctx.action.params) == null ? void 0 : _a.runId) || ((_b = ctx.query) == null ? void 0 : _b.runId) || ((_c = ctx.request.query) == null ? void 0 : _c.runId);
-          if (!runId) {
-            ctx.throw(400, "runId query parameter is required.");
-            return;
-          }
-          if (!(0, import_ctx_utils.isAdminUser)(ctx)) {
-            const userId = (0, import_ctx_utils.currentUserId)(ctx);
-            const run = await ctx.db.getRepository("agentLoopRuns").findOne({
-              filter: { id: runId }
-            });
-            if (!run) {
-              ctx.throw(404, "Run not found.");
-              return;
-            }
-            const ownerId = run.get ? run.get("userId") : run.userId;
-            if (!userId || String(ownerId) !== String(userId)) {
-              ctx.throw(403, "You cannot stream events for this run.");
-              return;
-            }
-          }
-          ctx.type = "text/event-stream";
-          ctx.set("Cache-Control", "no-cache");
-          ctx.set("Connection", "keep-alive");
-          ctx.set("X-Accel-Buffering", "no");
-          const unsubscribe = (0, import_RunEventBus.getRunEventBus)().subscribe(runId, (event) => {
-            try {
-              ctx.res.write(`data: ${JSON.stringify(event)}
-
-`);
-            } catch {
-              unsubscribe();
-            }
-          });
-          const keepalive = setInterval(() => {
-            try {
-              ctx.res.write(": keepalive\n\n");
-            } catch {
-              clearInterval(keepalive);
-              unsubscribe();
-            }
-          }, 15e3);
-          ctx.req.on("close", () => {
-            clearInterval(keepalive);
-            unsubscribe();
-          });
-          ctx.req.on("error", () => {
-            clearInterval(keepalive);
-            unsubscribe();
-          });
-          ctx.res.writeHead(200);
-          ctx.res.write(": connected\n\n");
-          await next();
-        }
-      }
-    });
+    (0, import_agent_monitor.registerAgentMonitorResource)(this);
+    this.installNativeObserver();
+    (_b = (_a = this.app).on) == null ? void 0 : _b.call(_a, "afterStart", this.installNativeObserver);
     (0, import_tracing.registerTracingResource)(this);
     this.app.cronJobManager.addJob({
       cronTime: "0 30 2 * * *",
       onTick: async () => {
         try {
-          const days = Number(process.env.ORCHESTRATOR_LOG_RETENTION_DAYS || 30);
-          if (!Number.isFinite(days) || days <= 0) return;
+          const days = await resolveTracingRetentionDays(this);
           const cutoff = new Date(Date.now() - days * 864e5);
           const repo = this.db.getRepository("orchestratorLogs");
           const spansRepo = this.db.getRepository("agentExecutionSpans");
@@ -207,10 +255,16 @@ class PluginAgentOrchestratorServer extends import_server.Plugin {
   async afterEnable() {
   }
   async afterDisable() {
+    var _a;
+    (_a = this.nativeObserver) == null ? void 0 : _a.uninstall();
   }
   async remove() {
   }
   async beforeStop() {
+    var _a, _b, _c, _d, _e;
+    (_b = (_a = this.app).off) == null ? void 0 : _b.call(_a, "afterStart", this.installNativeObserver);
+    (_d = (_c = this.app).removeListener) == null ? void 0 : _d.call(_c, "afterStart", this.installNativeObserver);
+    (_e = this.nativeObserver) == null ? void 0 : _e.uninstall();
     await this.skillHub.beforeStop();
   }
 }