@dudousxd/nestjs-agent-core 0.24.0 → 0.25.0

package/dist/index.cjs

@@ -36,6 +36,7 @@ __export(src_exports, {
   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,
@@ -109,6 +110,7 @@ __export(src_exports, {
   encodeStreamEvent: () => encodeStreamEvent,
   estimateCost: () => estimateCost,
   estimateMessageTokens: () => estimateMessageTokens,
+  exhaustedWindow: () => exhaustedWindow,
   extractJson: () => extractJson,
   filterToolsByAllowList: () => filterToolsByAllowList,
   filterToolsByCanUse: () => filterToolsByCanUse,
@@ -146,6 +148,7 @@ __export(src_exports, {
   publishAgentToolCall: () => publishAgentToolCall,
   publishAgentToolRetry: () => publishAgentToolRetry,
   questionOptions: () => questionOptions,
+  quotaPeriodRange: () => quotaPeriodRange,
   readElicitationInput: () => readElicitationInput,
   readElicitationQuestions: () => readElicitationQuestions,
   releaseGatedFrames: () => releaseGatedFrames,
@@ -208,6 +211,7 @@ 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");
@@ -466,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) {
@@ -4770,6 +4811,7 @@ __name(runAgentLoop, "runAgentLoop");
   AGENT_OPTIONS,
   AGENT_PRICING_STORE,
   AGENT_PROMPT_CONTRIBUTORS,
+  AGENT_QUOTA_PROVIDER,
   AGENT_QUOTA_STORE,
   AGENT_REGISTRY,
   AGENT_RETRIEVER,
@@ -4843,6 +4885,7 @@ __name(runAgentLoop, "runAgentLoop");
   encodeStreamEvent,
   estimateCost,
   estimateMessageTokens,
+  exhaustedWindow,
   extractJson,
   filterToolsByAllowList,
   filterToolsByCanUse,
@@ -4880,6 +4923,7 @@ __name(runAgentLoop, "runAgentLoop");
   publishAgentToolCall,
   publishAgentToolRetry,
   questionOptions,
+  quotaPeriodRange,
   readElicitationInput,
   readElicitationQuestions,
   releaseGatedFrames,