@dudousxd/nestjs-agent 0.2.0 → 0.3.0

package/dist/durable/index.js

@@ -215,7 +215,7 @@ AgentDurableModule = _ts_decorate3([
 ], AgentDurableModule);
 
 // src/agent.module.ts
-import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER5, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY4, AGENT_DURABLE_RUNNER as AGENT_DURABLE_RUNNER2, AGENT_MODEL as AGENT_MODEL2, AGENT_OPTIONS as AGENT_OPTIONS3, AGENT_PROMPT_CONTRIBUTORS as AGENT_PROMPT_CONTRIBUTORS3, AGENT_QUOTA_STORE as AGENT_QUOTA_STORE2, AGENT_REGISTRY as AGENT_REGISTRY4, AGENT_ROLES_POLICY as AGENT_ROLES_POLICY2, AGENT_RUNNER as AGENT_RUNNER2, AGENT_SINK as AGENT_SINK2, AGENT_STORE as AGENT_STORE5, AGENT_TOOL_REGISTRY as AGENT_TOOL_REGISTRY3, AgentRegistry as AgentRegistry4, DefaultRolesPolicy, ToolRegistry as ToolRegistry3 } from "@dudousxd/nestjs-agent-core";
+import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER5, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY4, AGENT_DURABLE_RUNNER as AGENT_DURABLE_RUNNER2, AGENT_MODEL as AGENT_MODEL2, AGENT_OPTIONS as AGENT_OPTIONS3, AGENT_PROMPT_CONTRIBUTORS as AGENT_PROMPT_CONTRIBUTORS3, AGENT_QUOTA_STORE as AGENT_QUOTA_STORE3, AGENT_REGISTRY as AGENT_REGISTRY4, AGENT_ROLES_POLICY as AGENT_ROLES_POLICY2, AGENT_RUNNER as AGENT_RUNNER2, AGENT_SINK as AGENT_SINK2, AGENT_STORE as AGENT_STORE5, AGENT_TOOL_REGISTRY as AGENT_TOOL_REGISTRY3, AgentRegistry as AgentRegistry4, DefaultRolesPolicy, ToolRegistry as ToolRegistry3 } from "@dudousxd/nestjs-agent-core";
 import { Global as Global2, Module as Module2 } from "@nestjs/common";
 import { DiscoveryModule, RouterModule } from "@nestjs/core";
 
@@ -368,7 +368,7 @@ AgentDepsFactory2 = _ts_decorate4([
 ], AgentDepsFactory2);
 
 // src/agent.service.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE3 } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY2, AGENT_QUOTA_STORE as AGENT_QUOTA_STORE2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE3 } from "@dudousxd/nestjs-agent-core";
 import { BadRequestException, ForbiddenException, Inject as Inject3, Injectable as Injectable4, NotFoundException } from "@nestjs/common";
 function _ts_decorate5(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -394,10 +394,12 @@ var AgentService = class {
   runner;
   store;
   deps;
-  constructor(runner, store, deps) {
+  quota;
+  constructor(runner, store, deps, quota) {
     this.runner = runner;
     this.store = store;
     this.deps = deps;
+    this.quota = quota;
   }
   async chat(params) {
     const agentName = params.agentName ?? this.deps.defaultAgentName();
@@ -407,7 +409,10 @@ var AgentService = class {
         throw new BadRequestException("regenerate requires an existing threadId");
       }
       const created = await this.store.createThread({
-        actor: params.actor
+        actor: params.actor,
+        ...params.transient === true ? {
+          transient: true
+        } : {}
       });
       threadId = created.id;
     } else if (params.regenerate === true) {
@@ -470,8 +475,53 @@ var AgentService = class {
     await this.assertOwnsThread(actor, threadId);
     return this.store.forkThread(threadId, fromMessageId);
   }
+  async renameThread(actor, threadId, title) {
+    const trimmed = title.trim();
+    if (trimmed.length === 0) {
+      throw new BadRequestException("title must not be empty");
+    }
+    if (trimmed.length > 200) {
+      throw new BadRequestException("title must be at most 200 characters");
+    }
+    await this.assertOwnsThread(actor, threadId);
+    return this.store.setTitle(threadId, trimmed);
+  }
+  async promoteThread(actor, threadId) {
+    await this.assertOwnsThread(actor, threadId);
+    return this.store.promoteThread(threadId);
+  }
+  /**
+  * Drop a message and everything after it — the "edit and resend" / "delete from here" primitive.
+  * The client then sends a fresh turn on the truncated thread. Ownership-gated like the other
+  * thread-scoped mutations.
+  */
+  async truncateThreadFrom(actor, threadId, messageId) {
+    await this.assertOwnsThread(actor, threadId);
+    return this.store.truncateFrom(threadId, messageId);
+  }
+  /**
+  * The day's usage for the badge: tokens + summed USD cost from the store, and the configured
+  * limit from the quota store (null → unlimited). `withinLimit` comes from the quota store so it
+  * can never drift from what enforcement uses.
+  */
   async quotaToday(actorRef) {
-    return this.store.quotaToday(actorRef, utcDay());
+    const day = utcDay();
+    const { usedTokens, costUsd } = await this.store.quotaToday(actorRef, day);
+    if (this.quota === void 0) {
+      return {
+        usedTokens,
+        costUsd,
+        limitTokens: null,
+        withinLimit: true
+      };
+    }
+    const state = await this.quota.check(actorRef, day);
+    return {
+      usedTokens: state.usedTokens,
+      costUsd,
+      limitTokens: state.limitTokens,
+      withinLimit: state.withinLimit
+    };
   }
   /**
   * Authorization seam for thread-scoped endpoints: the caller must own the thread. A missing
@@ -522,11 +572,13 @@ AgentService = _ts_decorate5([
   _ts_param3(0, Inject3(AGENT_RUNNER)),
   _ts_param3(1, Inject3(AGENT_STORE3)),
   _ts_param3(2, Inject3(AGENT_DEPS_FACTORY2)),
+  _ts_param3(3, Inject3(AGENT_QUOTA_STORE2)),
   _ts_metadata4("design:type", Function),
   _ts_metadata4("design:paramtypes", [
     typeof AgentRunner === "undefined" ? Object : AgentRunner,
     typeof AgentStore === "undefined" ? Object : AgentStore,
-    typeof AgentDepsFactory === "undefined" ? Object : AgentDepsFactory
+    typeof AgentDepsFactory === "undefined" ? Object : AgentDepsFactory,
+    Object
   ])
 ], AgentService);
 
@@ -576,6 +628,9 @@ var ChatController = class {
       } : {},
       ...body.regenerate === true ? {
         regenerate: true
+      } : {},
+      ...body.transient === true ? {
+        transient: true
       } : {}
     });
     await this.pipe(res, runId, threadId);
@@ -732,7 +787,7 @@ QuotaController = _ts_decorate7([
 
 // src/controller/threads.controller.ts
 import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER3 } from "@dudousxd/nestjs-agent-core";
-import { Controller as Controller3, Delete, Get as Get3, Inject as Inject6, Param as Param2, Post as Post2, Req as Req3 } from "@nestjs/common";
+import { Body as Body2, Controller as Controller3, Delete, Get as Get3, Inject as Inject6, Param as Param2, Patch, Post as Post2, Req as Req3 } from "@nestjs/common";
 function _ts_decorate8(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
@@ -779,6 +834,27 @@ var ThreadsController = class {
     const actor = await this.actorResolver.resolve(req);
     return this.agent.forkThread(actor, id, messageId);
   }
+  async rename(req, id, body) {
+    const actor = await this.actorResolver.resolve(req);
+    await this.agent.renameThread(actor, id, body.title);
+    return {
+      ok: true
+    };
+  }
+  async promote(req, id) {
+    const actor = await this.actorResolver.resolve(req);
+    await this.agent.promoteThread(actor, id);
+    return {
+      ok: true
+    };
+  }
+  async truncateFrom(req, id, messageId) {
+    const actor = await this.actorResolver.resolve(req);
+    await this.agent.truncateThreadFrom(actor, id, messageId);
+    return {
+      ok: true
+    };
+  }
 };
 _ts_decorate8([
   Get3(),
@@ -824,6 +900,43 @@ _ts_decorate8([
   ]),
   _ts_metadata7("design:returntype", Promise)
 ], ThreadsController.prototype, "fork", null);
+_ts_decorate8([
+  Patch(":id"),
+  _ts_param6(0, Req3()),
+  _ts_param6(1, Param2("id")),
+  _ts_param6(2, Body2()),
+  _ts_metadata7("design:type", Function),
+  _ts_metadata7("design:paramtypes", [
+    typeof Request === "undefined" ? Object : Request,
+    String,
+    typeof RenameThreadBody === "undefined" ? Object : RenameThreadBody
+  ]),
+  _ts_metadata7("design:returntype", Promise)
+], ThreadsController.prototype, "rename", null);
+_ts_decorate8([
+  Post2(":id/promote"),
+  _ts_param6(0, Req3()),
+  _ts_param6(1, Param2("id")),
+  _ts_metadata7("design:type", Function),
+  _ts_metadata7("design:paramtypes", [
+    typeof Request === "undefined" ? Object : Request,
+    String
+  ]),
+  _ts_metadata7("design:returntype", Promise)
+], ThreadsController.prototype, "promote", null);
+_ts_decorate8([
+  Delete(":id/from/:messageId"),
+  _ts_param6(0, Req3()),
+  _ts_param6(1, Param2("id")),
+  _ts_param6(2, Param2("messageId")),
+  _ts_metadata7("design:type", Function),
+  _ts_metadata7("design:paramtypes", [
+    typeof Request === "undefined" ? Object : Request,
+    String,
+    String
+  ]),
+  _ts_metadata7("design:returntype", Promise)
+], ThreadsController.prototype, "truncateFrom", null);
 ThreadsController = _ts_decorate8([
   Controller3("threads"),
   _ts_param6(1, Inject6(AGENT_ACTOR_RESOLVER3)),
@@ -836,7 +949,7 @@ ThreadsController = _ts_decorate8([
 
 // src/controller/tool-call.controller.ts
 import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER4 } from "@dudousxd/nestjs-agent-core";
-import { Body as Body2, Controller as Controller4, Inject as Inject7, Post as Post3, Req as Req4 } from "@nestjs/common";
+import { Body as Body3, Controller as Controller4, Inject as Inject7, Post as Post3, Req as Req4 } from "@nestjs/common";
 function _ts_decorate9(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
@@ -882,7 +995,7 @@ var ToolCallController = class {
 _ts_decorate9([
   Post3("approve"),
   _ts_param7(0, Req4()),
-  _ts_param7(1, Body2()),
+  _ts_param7(1, Body3()),
   _ts_metadata8("design:type", Function),
   _ts_metadata8("design:paramtypes", [
     typeof Request === "undefined" ? Object : Request,
@@ -893,7 +1006,7 @@ _ts_decorate9([
 _ts_decorate9([
   Post3("reject"),
   _ts_param7(0, Req4()),
-  _ts_param7(1, Body2()),
+  _ts_param7(1, Body3()),
   _ts_metadata8("design:type", Function),
   _ts_metadata8("design:paramtypes", [
     typeof Request === "undefined" ? Object : Request,
@@ -1491,7 +1604,7 @@ function sharedProviders(durable, includeStore) {
       // An explicit `quota` wins; otherwise `quotaLimitTokens` binds the built-in ledger-backed
       // store; otherwise quotas are off (undefined). The ledger store reads the same usage rows
       // the loop already writes, so enforcement needs no extra shared state across replicas.
-      provide: AGENT_QUOTA_STORE2,
+      provide: AGENT_QUOTA_STORE3,
       useFactory: /* @__PURE__ */ __name((o, store) => o.quota ?? (o.quotaLimitTokens !== void 0 ? new LedgerQuotaStore(store, o.quotaLimitTokens) : void 0), "useFactory"),
       inject: [
         AGENT_OPTIONS3,
@@ -1556,7 +1669,7 @@ function exportsFor(includeStore) {
     ...includeStore ? [
       AGENT_STORE5
     ] : [],
-    AGENT_QUOTA_STORE2,
+    AGENT_QUOTA_STORE3,
     AGENT_PROMPT_CONTRIBUTORS3,
     AGENT_DEPS_FACTORY4,
     AgentDepsFactory2,