@dudousxd/nestjs-agent-core 0.23.0 → 0.25.0

package/dist/guardrails/index.d.cts

@@ -1,4 +1,4 @@
-import { A as Actor, I as InputProcessor, O as OutputProcessor, T as ToolHandler } from '../tool-BdW-LV1J.cjs';
+import { A as Actor, I as InputProcessor, O as OutputProcessor, T as ToolHandler } from '../tool-fLRLs_f3.cjs';
 import '@standard-schema/spec';
 
 /**

package/dist/guardrails/index.d.ts

@@ -1,4 +1,4 @@
-import { A as Actor, I as InputProcessor, O as OutputProcessor, T as ToolHandler } from '../tool-BdW-LV1J.js';
+import { A as Actor, I as InputProcessor, O as OutputProcessor, T as ToolHandler } from '../tool-fLRLs_f3.js';
 import '@standard-schema/spec';
 
 /**

package/dist/index.cjs

@@ -32,9 +32,11 @@ __export(src_exports, {
   AGENT_GOVERNANCE_QUERIES: () => AGENT_GOVERNANCE_QUERIES,
   AGENT_MEMORY: () => AGENT_MEMORY,
   AGENT_MODEL: () => AGENT_MODEL,
+  AGENT_MODEL_CATALOG: () => AGENT_MODEL_CATALOG,
   AGENT_OPTIONS: () => AGENT_OPTIONS,
   AGENT_PRICING_STORE: () => AGENT_PRICING_STORE,
   AGENT_PROMPT_CONTRIBUTORS: () => AGENT_PROMPT_CONTRIBUTORS,
+  AGENT_QUOTA_PROVIDER: () => AGENT_QUOTA_PROVIDER,
   AGENT_QUOTA_STORE: () => AGENT_QUOTA_STORE,
   AGENT_REGISTRY: () => AGENT_REGISTRY,
   AGENT_RETRIEVER: () => AGENT_RETRIEVER,
@@ -108,11 +110,13 @@ __export(src_exports, {
   encodeStreamEvent: () => encodeStreamEvent,
   estimateCost: () => estimateCost,
   estimateMessageTokens: () => estimateMessageTokens,
+  exhaustedWindow: () => exhaustedWindow,
   extractJson: () => extractJson,
   filterToolsByAllowList: () => filterToolsByAllowList,
   filterToolsByCanUse: () => filterToolsByCanUse,
   filterToolsByEnabled: () => filterToolsByEnabled,
   filterToolsByRole: () => filterToolsByRole,
+  findCatalogModel: () => findCatalogModel,
   gateFollowUps: () => gateFollowUps,
   gateTail: () => gateTail,
   invokeWithTransientRetry: () => invokeWithTransientRetry,
@@ -144,6 +148,7 @@ __export(src_exports, {
   publishAgentToolCall: () => publishAgentToolCall,
   publishAgentToolRetry: () => publishAgentToolRetry,
   questionOptions: () => questionOptions,
+  quotaPeriodRange: () => quotaPeriodRange,
   readElicitationInput: () => readElicitationInput,
   readElicitationQuestions: () => readElicitationQuestions,
   releaseGatedFrames: () => releaseGatedFrames,
@@ -169,6 +174,7 @@ __export(src_exports, {
   skillToolDefinition: () => skillToolDefinition,
   skillWriteVerdict: () => skillWriteVerdict,
   stampToolKinds: () => stampToolKinds,
+  staticModelCatalog: () => staticModelCatalog,
   staticSkillProvider: () => staticSkillProvider,
   summarizeWithModel: () => summarizeWithModel,
   tenantScope: () => tenantScope,
@@ -183,6 +189,7 @@ __export(src_exports, {
   windowHistory: () => windowHistory,
   withAskTool: () => withAskTool,
   withMemoryTool: () => withMemoryTool,
+  withSelectedModel: () => withSelectedModel,
   withSkillTool: () => withSkillTool,
   withToolTimeout: () => withToolTimeout,
   withTurnFrames: () => withTurnFrames,
@@ -204,6 +211,8 @@ var AGENT_TOOL_REGISTRY = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:too
 var AGENT_REGISTRY = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:agent-registry");
 var AGENT_ACTOR_RESOLVER = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:actor-resolver");
 var AGENT_GOVERNANCE_QUERIES = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:governance-queries");
+var AGENT_QUOTA_PROVIDER = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:quota-provider");
+var AGENT_MODEL_CATALOG = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:model-catalog");
 var AGENT_PRICING_STORE = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:pricing-store");
 var AGENT_RETRIEVER = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:retriever");
 var AGENT_EMBEDDING_PROVIDER = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:embedding-provider");
@@ -216,6 +225,31 @@ var AGENT_SKILLS = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:skills");
 var AGENT_MEMORY = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:memory");
 var AGENT_SKILL_SOURCES = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:skill-sources");
 
+// src/spi/model-catalog.ts
+function staticModelCatalog(view) {
+  return {
+    list: /* @__PURE__ */ __name(() => view, "list")
+  };
+}
+__name(staticModelCatalog, "staticModelCatalog");
+function findCatalogModel(view, id) {
+  for (const group of view.providers) {
+    const found = group.models.find((model) => model.id === id);
+    if (found !== void 0) return found;
+  }
+  return void 0;
+}
+__name(findCatalogModel, "findCatalogModel");
+function withSelectedModel(provider, model) {
+  return {
+    runTurn: /* @__PURE__ */ __name((args) => provider.runTurn({
+      ...args,
+      model
+    }), "runTurn")
+  };
+}
+__name(withSelectedModel, "withSelectedModel");
+
 // src/spi/token-stream-sink.ts
 var AgentStreamError = class extends Error {
   static {
@@ -436,6 +470,43 @@ function mergeUi(...lists) {
 }
 __name(mergeUi, "mergeUi");
 
+// src/spi/quota-provider.ts
+function exhaustedWindow(windows) {
+  for (const window of windows) {
+    const tokensOut = window.limitTokens !== void 0 && window.usedTokens >= window.limitTokens;
+    const spendOut = window.limitUsd !== void 0 && window.usedUsd >= window.limitUsd;
+    if (tokensOut || spendOut) {
+      return {
+        period: window.period,
+        reason: `${window.period === "day" ? "Daily" : "Monthly"} ${spendOut ? "spend" : "token"} limit reached`
+      };
+    }
+  }
+  return void 0;
+}
+__name(exhaustedWindow, "exhaustedWindow");
+function quotaPeriodRange(period, now) {
+  const year = now.getUTCFullYear();
+  const month = now.getUTCMonth();
+  const day = now.toISOString().slice(0, 10);
+  if (period === "day") {
+    const next = new Date(Date.UTC(year, month, now.getUTCDate() + 1));
+    return {
+      fromDay: day,
+      toDay: day,
+      resetsAt: next.toISOString()
+    };
+  }
+  const first = new Date(Date.UTC(year, month, 1));
+  const last = new Date(Date.UTC(year, month + 1, 0));
+  return {
+    fromDay: first.toISOString().slice(0, 10),
+    toDay: last.toISOString().slice(0, 10),
+    resetsAt: new Date(Date.UTC(year, month + 1, 1)).toISOString()
+  };
+}
+__name(quotaPeriodRange, "quotaPeriodRange");
+
 // src/spi/pricing-store.ts
 async function seedModelPrices(store, prices) {
   for (const price of prices) {
@@ -4112,7 +4183,12 @@ async function invokeClaimedToolsTogether(turn, parallel, claimed) {
   return results;
 }
 __name(invokeClaimedToolsTogether, "invokeClaimedToolsTogether");
-async function runAgentLoop(deps, input, hooks) {
+async function runAgentLoop(boundDeps, input, hooks) {
+  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);
   const inputProcessors = deps.inputProcessors ?? [];
@@ -4309,6 +4385,9 @@ ${block}`;
         actor: input.actor,
         ...gated ? {
           bufferOutput: true
+        } : {},
+        ...input.model !== void 0 ? {
+          model: input.model
         } : {}
       });
     } else {
@@ -4728,9 +4807,11 @@ __name(runAgentLoop, "runAgentLoop");
   AGENT_GOVERNANCE_QUERIES,
   AGENT_MEMORY,
   AGENT_MODEL,
+  AGENT_MODEL_CATALOG,
   AGENT_OPTIONS,
   AGENT_PRICING_STORE,
   AGENT_PROMPT_CONTRIBUTORS,
+  AGENT_QUOTA_PROVIDER,
   AGENT_QUOTA_STORE,
   AGENT_REGISTRY,
   AGENT_RETRIEVER,
@@ -4804,11 +4885,13 @@ __name(runAgentLoop, "runAgentLoop");
   encodeStreamEvent,
   estimateCost,
   estimateMessageTokens,
+  exhaustedWindow,
   extractJson,
   filterToolsByAllowList,
   filterToolsByCanUse,
   filterToolsByEnabled,
   filterToolsByRole,
+  findCatalogModel,
   gateFollowUps,
   gateTail,
   invokeWithTransientRetry,
@@ -4840,6 +4923,7 @@ __name(runAgentLoop, "runAgentLoop");
   publishAgentToolCall,
   publishAgentToolRetry,
   questionOptions,
+  quotaPeriodRange,
   readElicitationInput,
   readElicitationQuestions,
   releaseGatedFrames,
@@ -4865,6 +4949,7 @@ __name(runAgentLoop, "runAgentLoop");
   skillToolDefinition,
   skillWriteVerdict,
   stampToolKinds,
+  staticModelCatalog,
   staticSkillProvider,
   summarizeWithModel,
   tenantScope,
@@ -4879,6 +4964,7 @@ __name(runAgentLoop, "runAgentLoop");
   windowHistory,
   withAskTool,
   withMemoryTool,
+  withSelectedModel,
   withSkillTool,
   withToolTimeout,
   withTurnFrames,