@dudousxd/nestjs-agent-core 0.38.1 → 0.39.0

package/dist/ag-ui/index.d.cts

@@ -1,4 +1,4 @@
-import { A as AgentStreamEvent } from '../stream-events-rpd3d6gh.cjs';
+import { A as AgentStreamEvent } from '../stream-events-CbVEowYb.cjs';
 import '@standard-schema/spec';
 
 /**

package/dist/ag-ui/index.d.ts

@@ -1,4 +1,4 @@
-import { A as AgentStreamEvent } from '../stream-events-rpd3d6gh.js';
+import { A as AgentStreamEvent } from '../stream-events-CbVEowYb.js';
 import '@standard-schema/spec';
 
 /**

package/dist/genui/index.d.cts

@@ -1,8 +1,8 @@
 import { a as Catalog, J as JsonSchema, G as GenuiValidation, b as GenuiIssue } from '../catalog-CrzetM3_.cjs';
 export { A as AjvLike, c as COMPONENT_NAME, d as CatalogOptions, C as ComponentDefinition, e as JsonSchemaValidator, P as PropsSchema, f as ajvValidator, g as builtinJsonSchemaValidator, h as defineCatalog, i as defineComponent, j as formatIssues, k as isStandardSchema, t as toJsonSchema, l as toSnakeCase, m as toolNameFor, v as validateProps, n as validatePropsSync } from '../catalog-CrzetM3_.cjs';
 import { StandardSchemaV1, StandardJSONSchemaV1 } from '@standard-schema/spec';
-import { T as ToolHandler } from '../tool-Dp8HA_f4.cjs';
-import { a as Actor, T as ToolSpec, b as ToolPresentation } from '../stream-events-rpd3d6gh.cjs';
+import { T as ToolHandler } from '../tool-DZYLEKnl.cjs';
+import { a as Actor, T as ToolSpec, b as ToolPresentation } from '../stream-events-CbVEowYb.cjs';
 
 /**
  * One node of a composed UI: a catalog component, its props, and — for components declared with

package/dist/genui/index.d.ts

@@ -1,8 +1,8 @@
 import { a as Catalog, J as JsonSchema, G as GenuiValidation, b as GenuiIssue } from '../catalog-CrzetM3_.js';
 export { A as AjvLike, c as COMPONENT_NAME, d as CatalogOptions, C as ComponentDefinition, e as JsonSchemaValidator, P as PropsSchema, f as ajvValidator, g as builtinJsonSchemaValidator, h as defineCatalog, i as defineComponent, j as formatIssues, k as isStandardSchema, t as toJsonSchema, l as toSnakeCase, m as toolNameFor, v as validateProps, n as validatePropsSync } from '../catalog-CrzetM3_.js';
 import { StandardSchemaV1, StandardJSONSchemaV1 } from '@standard-schema/spec';
-import { T as ToolHandler } from '../tool-hhssZ_HW.js';
-import { a as Actor, T as ToolSpec, b as ToolPresentation } from '../stream-events-rpd3d6gh.js';
+import { T as ToolHandler } from '../tool-_iq4xRyk.js';
+import { a as Actor, T as ToolSpec, b as ToolPresentation } from '../stream-events-CbVEowYb.js';
 
 /**
  * One node of a composed UI: a catalog component, its props, and — for components declared with

package/dist/guardrails/index.d.cts

@@ -1,6 +1,6 @@
-import { I as InputProcessor, O as OutputProcessor } from '../processors-CAVevaYJ.cjs';
-import { T as ToolHandler } from '../tool-Dp8HA_f4.cjs';
-import { a as Actor } from '../stream-events-rpd3d6gh.cjs';
+import { I as InputProcessor, O as OutputProcessor } from '../processors-6p9nrqEi.cjs';
+import { T as ToolHandler } from '../tool-DZYLEKnl.cjs';
+import { a as Actor } from '../stream-events-CbVEowYb.cjs';
 import '@standard-schema/spec';
 
 /**

package/dist/guardrails/index.d.ts

@@ -1,6 +1,6 @@
-import { I as InputProcessor, O as OutputProcessor } from '../processors-C-FP4nDy.js';
-import { T as ToolHandler } from '../tool-hhssZ_HW.js';
-import { a as Actor } from '../stream-events-rpd3d6gh.js';
+import { I as InputProcessor, O as OutputProcessor } from '../processors-Bp5XmF6V.js';
+import { T as ToolHandler } from '../tool-_iq4xRyk.js';
+import { a as Actor } from '../stream-events-CbVEowYb.js';
 import '@standard-schema/spec';
 
 /**

package/dist/index.cjs

@@ -140,9 +140,11 @@ __export(src_exports, {
   filterToolsByEnabled: () => filterToolsByEnabled,
   filterToolsByRole: () => filterToolsByRole,
   findCatalogModel: () => findCatalogModel,
+  findPersona: () => findPersona,
   gateFollowUps: () => gateFollowUps,
   gateTail: () => gateTail,
   hashConfirmToken: () => hashConfirmToken,
+  intersectAllowLists: () => intersectAllowLists,
   invokeWithTransientRetry: () => invokeWithTransientRetry,
   isChatQueueStore: () => isChatQueueStore,
   isControlFlowSignal: () => isControlFlowSignal,
@@ -160,6 +162,7 @@ __export(src_exports, {
   observeTurnFrames: () => observeTurnFrames,
   offerMemories: () => offerMemories,
   offerSkills: () => offerSkills,
+  personaCatalogEntry: () => personaCatalogEntry,
   publishAgentDelegated: () => publishAgentDelegated,
   publishAgentMemoryResolved: () => publishAgentMemoryResolved,
   publishAgentMemoryWritten: () => publishAgentMemoryWritten,
@@ -189,6 +192,7 @@ __export(src_exports, {
   resolveGateLookback: () => resolveGateLookback,
   resolveMemoryDigest: () => resolveMemoryDigest,
   resolveOutputGateMode: () => resolveOutputGateMode,
+  resolvePersonaAlias: () => resolvePersonaAlias,
   resolveSkillCatalog: () => resolveSkillCatalog,
   resolveToolTransientRetryNumbers: () => resolveToolTransientRetryNumbers,
   rollupThreadUsage: () => rollupThreadUsage,
@@ -565,6 +569,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
     } : {},
@@ -957,6 +964,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);
@@ -2896,7 +2951,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);
@@ -2910,6 +2965,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);
@@ -3261,16 +3319,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);
@@ -3281,6 +3333,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;
@@ -3522,6 +3635,9 @@ function processorContext(input, step) {
     step,
     ...input.agentName !== void 0 ? {
       agentName: input.agentName
+    } : {},
+    ...input.persona !== void 0 ? {
+      persona: input.persona
     } : {}
   };
 }
@@ -3685,6 +3801,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
     } : {},
@@ -3732,6 +3851,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({
@@ -4077,6 +4199,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 = {
@@ -4087,6 +4210,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
         } : {}
@@ -4095,6 +4221,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
         } : {},
@@ -4117,7 +4246,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
@@ -4248,7 +4379,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
@@ -4544,14 +4675,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);
@@ -4592,6 +4725,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
       } : {}
@@ -4614,6 +4750,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);
@@ -4750,13 +4889,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
@@ -4971,6 +5113,9 @@ ${block}`;
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
       } : {},
+      ...input.persona !== void 0 ? {
+        persona: input.persona
+      } : {},
       ...turn.reasoning !== void 0 ? {
         reasoning: turn.reasoning
       } : {},
@@ -5065,6 +5210,9 @@ ${block}`;
       } : {},
       ...memoryDigest !== void 0 ? {
         memory: memoryDigest
+      } : {},
+      ...persona !== void 0 ? {
+        persona
       } : {}
     };
     const parallel = hooks.parallel;
@@ -5197,7 +5345,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);
@@ -5302,6 +5453,9 @@ var InMemoryAgentStore = class {
       } : {},
       ...source.model != null ? {
         model: source.model
+      } : {},
+      ...source.persona != null ? {
+        persona: source.persona
       } : {}
     };
     this.threads.set(id, row);
@@ -5353,6 +5507,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) {
@@ -5366,6 +5523,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;
@@ -5470,6 +5631,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
       } : {},
@@ -5608,6 +5772,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
       } : {},
@@ -6024,7 +6191,8 @@ var InMemoryAgentStore = class {
       defaultAgent: row.defaultAgent ?? null,
       ...row.model != null ? {
         model: row.model
-      } : {}
+      } : {},
+      persona: row.persona ?? null
     };
   }
 };
@@ -6697,9 +6865,11 @@ var SqlTokenStreamSink = class {
   filterToolsByEnabled,
   filterToolsByRole,
   findCatalogModel,
+  findPersona,
   gateFollowUps,
   gateTail,
   hashConfirmToken,
+  intersectAllowLists,
   invokeWithTransientRetry,
   isChatQueueStore,
   isControlFlowSignal,
@@ -6717,6 +6887,7 @@ var SqlTokenStreamSink = class {
   observeTurnFrames,
   offerMemories,
   offerSkills,
+  personaCatalogEntry,
   publishAgentDelegated,
   publishAgentMemoryResolved,
   publishAgentMemoryWritten,
@@ -6746,6 +6917,7 @@ var SqlTokenStreamSink = class {
   resolveGateLookback,
   resolveMemoryDigest,
   resolveOutputGateMode,
+  resolvePersonaAlias,
   resolveSkillCatalog,
   resolveToolTransientRetryNumbers,
   rollupThreadUsage,