@dudousxd/nestjs-agent-core 0.38.1 → 0.39.0

package/dist/index.js

@@ -331,6 +331,9 @@ function queuedMessageView(message) {
     ...message.agentName !== void 0 ? {
       agentName: message.agentName
     } : {},
+    ...message.persona !== void 0 ? {
+      persona: message.persona
+    } : {},
     ...message.model !== void 0 ? {
       model: message.model
     } : {},
@@ -723,6 +726,54 @@ function filterToolsByAllowList(tools, allowedTools) {
 }
 __name(filterToolsByAllowList, "filterToolsByAllowList");
 
+// src/personas.ts
+function intersectAllowLists(first, second) {
+  if (first === void 0) {
+    return second === void 0 ? void 0 : [
+      ...second
+    ];
+  }
+  if (second === void 0) {
+    return [
+      ...first
+    ];
+  }
+  const allowed = new Set(second);
+  return first.filter((name) => allowed.has(name));
+}
+__name(intersectAllowLists, "intersectAllowLists");
+function findPersona(definition, id) {
+  if (id === void 0) {
+    return void 0;
+  }
+  return definition?.personas?.find((persona) => persona.id === id);
+}
+__name(findPersona, "findPersona");
+function personaCatalogEntry(persona) {
+  return {
+    id: persona.id,
+    label: persona.label,
+    ...persona.description !== void 0 ? {
+      description: persona.description
+    } : {}
+  };
+}
+__name(personaCatalogEntry, "personaCatalogEntry");
+function resolvePersonaAlias(definitions, name) {
+  for (const definition of definitions) {
+    for (const persona of definition.personas ?? []) {
+      if (persona.aliases?.includes(name) === true) {
+        return {
+          agent: definition.name,
+          persona: persona.id
+        };
+      }
+    }
+  }
+  return void 0;
+}
+__name(resolvePersonaAlias, "resolvePersonaAlias");
+
 // src/history.ts
 function estimateMessageTokens(message) {
   const extras = (message.toolCalls !== void 0 ? JSON.stringify(message.toolCalls).length : 0) + (message.toolResults !== void 0 ? JSON.stringify(message.toolResults).length : 0);
@@ -2662,7 +2713,7 @@ var ToolRegistry = class {
   * a call can reach here from a replayed durable step or an approval granted before the flag
   * moved, neither of which went through `definitionsFor` again) and re-parses the input via Zod.
   */
-  async invoke(name, input, ctx, policy) {
+  async invoke(name, input, ctx, policy, options = {}) {
     const entry = this.entries.get(name);
     if (entry === void 0) {
       throw new ToolNotFoundError(name);
@@ -2676,6 +2727,9 @@ var ToolRegistry = class {
     if (!await canActorUseTool(ctx.actor, entry.handler)) {
       throw new ToolForbiddenError(name);
     }
+    if (options.allowedTools !== void 0 && !options.allowedTools.includes(name)) {
+      throw new ToolForbiddenError(name);
+    }
     const validation = await entry.spec.inputSchema["~standard"].validate(input);
     if (validation.issues !== void 0) {
       throw new ToolInputInvalidError(name, validation.issues);
@@ -3027,16 +3081,10 @@ async function resolvePrompt(prompt, ctx) {
   return typeof prompt === "function" ? prompt(ctx) : prompt;
 }
 __name(resolvePrompt, "resolvePrompt");
-async function resolveSystemPrompt(deps, input) {
-  const ctx = {
-    actor: input.actor,
-    agentName: input.agentName ?? "default",
-    ...input.pageContext !== void 0 ? {
-      pageContext: input.pageContext
-    } : {}
-  };
+async function resolveSystemPrompt(deps, input, persona) {
+  const ctx = promptContext(input, persona);
   const sections = [
-    await resolvePrompt(deps.systemPrompt, ctx)
+    persona?.prompt ?? await resolvePrompt(deps.systemPrompt, ctx)
   ];
   for (const contribute of deps.promptContributors ?? []) {
     const section = await contribute(ctx);
@@ -3047,6 +3095,67 @@ async function resolveSystemPrompt(deps, input) {
   return sections.join("\n\n");
 }
 __name(resolveSystemPrompt, "resolveSystemPrompt");
+function withoutPersona(input) {
+  const { persona: _dropped, ...rest } = input;
+  return rest;
+}
+__name(withoutPersona, "withoutPersona");
+function promptContext(input, persona) {
+  return {
+    actor: input.actor,
+    agentName: input.agentName ?? "default",
+    ...input.pageContext !== void 0 ? {
+      pageContext: input.pageContext
+    } : {},
+    ...persona !== void 0 ? {
+      persona: {
+        id: persona.id,
+        label: persona.label
+      }
+    } : {}
+  };
+}
+__name(promptContext, "promptContext");
+async function resolveTurnPersona(deps, input) {
+  const persona = findPersona(deps, input.persona);
+  if (persona === void 0) {
+    return null;
+  }
+  const ref = {
+    id: persona.id,
+    label: persona.label
+  };
+  let prompt;
+  if (persona.systemPrompt !== void 0) {
+    const ctx = promptContext(input, ref);
+    const basePrompt = typeof persona.systemPrompt === "function" ? await resolvePrompt(deps.systemPrompt, ctx) : void 0;
+    prompt = await resolvePrompt(persona.systemPrompt, {
+      ...ctx,
+      ...basePrompt !== void 0 ? {
+        basePrompt
+      } : {}
+    });
+  }
+  return {
+    ...ref,
+    ...persona.allowedTools !== void 0 ? {
+      allowedTools: [
+        ...persona.allowedTools
+      ]
+    } : {},
+    ...prompt !== void 0 ? {
+      prompt
+    } : {}
+  };
+}
+__name(resolveTurnPersona, "resolveTurnPersona");
+function personaRefusal(persona, toolName) {
+  if (persona?.allowedTools === void 0 || persona.allowedTools.includes(toolName)) {
+    return null;
+  }
+  return `(tool "${toolName}" is not available to the "${persona.id}" persona)`;
+}
+__name(personaRefusal, "personaRefusal");
 function extractTask(input) {
   if (typeof input === "object" && input !== null && "task" in input) {
     const task = input.task;
@@ -3288,6 +3397,9 @@ function processorContext(input, step) {
     step,
     ...input.agentName !== void 0 ? {
       agentName: input.agentName
+    } : {},
+    ...input.persona !== void 0 ? {
+      persona: input.persona
     } : {}
   };
 }
@@ -3451,6 +3563,9 @@ function toolContext(deps, input, hooks) {
     ...input.agentName !== void 0 ? {
       agentName: input.agentName
     } : {},
+    ...input.persona !== void 0 ? {
+      persona: input.persona
+    } : {},
     ...input.pageContext !== void 0 ? {
       pageContext: input.pageContext
     } : {},
@@ -3498,6 +3613,9 @@ async function runIntake(intake, deps, input, hooks, writer, threadHasAssistant)
       ],
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
+      } : {},
+      ...input.persona !== void 0 ? {
+        persona: input.persona
       } : {}
     });
     await deps.store.recordToolCall({
@@ -3843,6 +3961,7 @@ async function invokeClaimedTool(turn, claimed) {
     if (claimed.toolType === "memory") {
       return await rememberIntoTurn(turn, claimed, startedAt);
     }
+    const invokeAllowList = turn.persona?.allowedTools === void 0 ? void 0 : intersectAllowLists(deps.toolAllowList, turn.persona.allowedTools);
     let raw;
     if (hooks.dispatchTool) {
       const stepCtx = {
@@ -3853,6 +3972,9 @@ async function invokeClaimedTool(turn, claimed) {
         ...input.agentName !== void 0 ? {
           agentName: input.agentName
         } : {},
+        ...input.persona !== void 0 ? {
+          persona: input.persona
+        } : {},
         ...input.pageContext !== void 0 ? {
           pageContext: input.pageContext
         } : {}
@@ -3861,6 +3983,9 @@ async function invokeClaimedTool(turn, claimed) {
         toolName: call.name,
         input: call.input,
         ctx: stepCtx,
+        ...invokeAllowList !== void 0 ? {
+          allowedTools: invokeAllowList
+        } : {},
         ...deps.toolTimeoutMs !== void 0 ? {
           timeoutMs: deps.toolTimeoutMs
         } : {},
@@ -3883,7 +4008,9 @@ async function invokeClaimedTool(turn, claimed) {
           return deps.registry.invoke(call.name, call.input, {
             ...toolCallContext(ctx, call.id),
             emitUi: ui2.emit
-          }, deps.rolesPolicy);
+          }, deps.rolesPolicy, invokeAllowList !== void 0 ? {
+            allowedTools: invokeAllowList
+          } : {});
         }, deps.toolTransientRetry ?? {}, {
           ...hooks.isControlFlowError !== void 0 ? {
             isControlFlowError: hooks.isControlFlowError
@@ -4014,7 +4141,7 @@ async function delegateToolCall(turn, claimed) {
   const { deps, input, hooks } = turn;
   const { call, targetAgent = call.name } = claimed;
   const task = extractTask(call.input);
-  const refusal = delegationRefusal({
+  const refusal = personaRefusal(turn.persona, call.name) ?? delegationRefusal({
     deps,
     input,
     targetAgent
@@ -4310,14 +4437,16 @@ async function invokeClaimedToolsTogether(turn, parallel, claimed) {
   return results;
 }
 __name(invokeClaimedToolsTogether, "invokeClaimedToolsTogether");
-async function runAgentLoop(boundDeps, input, hooks) {
+async function runAgentLoop(boundDeps, requested, hooks) {
+  const persona = requested.persona === void 0 ? void 0 : await hooks.step("persona:resolve", () => resolveTurnPersona(boundDeps, requested)) ?? void 0;
+  const input = persona === void 0 || persona.id !== requested.persona ? withoutPersona(requested) : requested;
   const deps = input.model === void 0 ? boundDeps : {
     ...boundDeps,
     model: withSelectedModel(boundDeps.model, input.model),
     modelId: input.model
   };
   const maxSteps = deps.maxSteps ?? 8;
-  let system = await resolveSystemPrompt(deps, input);
+  let system = await resolveSystemPrompt(deps, input, persona);
   const inputProcessors = deps.inputProcessors ?? [];
   const outputProcessors = deps.outputProcessors ?? [];
   const gateMode = resolveOutputGateMode(outputProcessors);
@@ -4358,6 +4487,9 @@ async function runAgentLoop(boundDeps, input, hooks) {
       role: "user",
       content: input.userText,
       runId: hooks.runId,
+      ...input.persona !== void 0 ? {
+        persona: input.persona
+      } : {},
       ...input.attachments !== void 0 ? {
         attachments: input.attachments
       } : {}
@@ -4380,6 +4512,9 @@ async function runAgentLoop(boundDeps, input, hooks) {
     actorId: input.actor.id,
     ...input.agentName !== void 0 ? {
       agentName: input.agentName
+    } : {},
+    ...input.persona !== void 0 ? {
+      persona: input.persona
     } : {}
   });
   const stages = await resolvePromptStages(deps, hooks);
@@ -4516,13 +4651,16 @@ ${block}`;
         } : {},
         ...input.model !== void 0 ? {
           model: input.model
+        } : {},
+        ...persona?.allowedTools !== void 0 ? {
+          personaAllowedTools: persona.allowedTools
         } : {}
       });
     } else {
       const tools = withMemoryTool({
         tools: withSkillTool({
           tools: withAskTool({
-            tools: await deps.registry.definitionsFor(input.actor, deps.rolesPolicy, deps.toolAllowList, {
+            tools: await deps.registry.definitionsFor(input.actor, deps.rolesPolicy, intersectAllowLists(deps.toolAllowList, persona?.allowedTools), {
               threadId: input.threadId,
               ...input.agentName !== void 0 ? {
                 agentName: input.agentName
@@ -4737,6 +4875,9 @@ ${block}`;
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
       } : {},
+      ...input.persona !== void 0 ? {
+        persona: input.persona
+      } : {},
       ...turn.reasoning !== void 0 ? {
         reasoning: turn.reasoning
       } : {},
@@ -4831,6 +4972,9 @@ ${block}`;
       } : {},
       ...memoryDigest !== void 0 ? {
         memory: memoryDigest
+      } : {},
+      ...persona !== void 0 ? {
+        persona
       } : {}
     };
     const parallel = hooks.parallel;
@@ -4963,7 +5107,10 @@ var InMemoryAgentStore = class {
       transient: input.transient ?? false,
       createdAt: ts,
       updatedAt: ts,
-      messages: []
+      messages: [],
+      ...input.persona !== void 0 ? {
+        persona: input.persona
+      } : {}
     };
     this.threads.set(id, row);
     return this.toSummary(row);
@@ -5068,6 +5215,9 @@ var InMemoryAgentStore = class {
       } : {},
       ...source.model != null ? {
         model: source.model
+      } : {},
+      ...source.persona != null ? {
+        persona: source.persona
       } : {}
     };
     this.threads.set(id, row);
@@ -5119,6 +5269,9 @@ var InMemoryAgentStore = class {
     if (patch.model !== void 0) {
       row.model = patch.model;
     }
+    if (patch.persona !== void 0) {
+      row.persona = patch.persona;
+    }
     row.updatedAt = this.now();
   }
   async activeRunForThread(threadId) {
@@ -5132,6 +5285,10 @@ var InMemoryAgentStore = class {
   async defaultAgentForThread(threadId) {
     return this.threads.get(threadId)?.defaultAgent ?? null;
   }
+  /** The thread's persona, projected like {@link defaultAgentForThread}. */
+  async personaForThread(threadId) {
+    return this.threads.get(threadId)?.persona ?? null;
+  }
   /** The thread's pinned model, projected like {@link defaultAgentForThread}. */
   async modelForThread(threadId) {
     return this.threads.get(threadId)?.model ?? null;
@@ -5236,6 +5393,9 @@ var InMemoryAgentStore = class {
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
       } : {},
+      ...input.persona !== void 0 ? {
+        persona: input.persona
+      } : {},
       ...input.model !== void 0 ? {
         model: input.model
       } : {},
@@ -5374,6 +5534,9 @@ var InMemoryAgentStore = class {
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
       } : {},
+      ...input.persona !== void 0 ? {
+        persona: input.persona
+      } : {},
       ...input.runId !== void 0 ? {
         runId: input.runId
       } : {},
@@ -5790,7 +5953,8 @@ var InMemoryAgentStore = class {
       defaultAgent: row.defaultAgent ?? null,
       ...row.model != null ? {
         model: row.model
-      } : {}
+      } : {},
+      persona: row.persona ?? null
     };
   }
 };
@@ -6462,9 +6626,11 @@ export {
   filterToolsByEnabled,
   filterToolsByRole,
   findCatalogModel,
+  findPersona,
   gateFollowUps,
   gateTail,
   hashConfirmToken,
+  intersectAllowLists,
   invokeWithTransientRetry,
   isChatQueueStore,
   isControlFlowSignal,
@@ -6482,6 +6648,7 @@ export {
   observeTurnFrames,
   offerMemories,
   offerSkills,
+  personaCatalogEntry,
   publishAgentDelegated,
   publishAgentMemoryResolved,
   publishAgentMemoryWritten,
@@ -6511,6 +6678,7 @@ export {
   resolveGateLookback,
   resolveMemoryDigest,
   resolveOutputGateMode,
+  resolvePersonaAlias,
   resolveSkillCatalog,
   resolveToolTransientRetryNumbers,
   rollupThreadUsage,