@dudousxd/nestjs-agent-dashboard 0.3.1 → 0.4.1

package/dist/server/index.cjs

@@ -22,28 +22,74 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/server/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AGENT_ACTOR_DIRECTORY: () => AGENT_ACTOR_DIRECTORY,
   AGENT_GOVERNANCE_QUERIES: () => AGENT_GOVERNANCE_QUERIES,
+  AGENT_PRICING_STORE: () => AGENT_PRICING_STORE,
   AgentApiController: () => AgentApiController,
   AgentApiModule: () => AgentApiModule,
   AgentDashboardModule: () => AgentDashboardModule,
   AgentUiController: () => AgentUiController,
   DASHBOARD_API_PATH: () => DASHBOARD_API_PATH,
   DASHBOARD_BASE_PATH: () => DASHBOARD_BASE_PATH,
-  DashboardService: () => DashboardService
+  DashboardService: () => DashboardService,
+  agentDashboardMountPaths: () => agentDashboardMountPaths,
+  parsePriceInput: () => parsePriceInput
 });
 module.exports = __toCommonJS(index_exports);
 
 // src/server/agent-api.controller.ts
-var import_common2 = require("@nestjs/common");
+var import_common3 = require("@nestjs/common");
 
 // src/server/dashboard.service.ts
 var import_node_diagnostics_channel = require("diagnostics_channel");
 var import_nestjs_diagnostics = require("@dudousxd/nestjs-diagnostics");
-var import_common = require("@nestjs/common");
+var import_common2 = require("@nestjs/common");
 var import_rxjs = require("rxjs");
 
+// src/server/parse-price-input.ts
+var import_common = require("@nestjs/common");
+function parsePriceInput(body) {
+  if (typeof body !== "object" || body === null) {
+    throw new import_common.BadRequestException("Expected a JSON object body.");
+  }
+  const { modelId, inputPricePer1m, outputPricePer1m, cacheWritePricePer1m, cacheReadPricePer1m } = body;
+  if (typeof modelId !== "string" || modelId.trim().length === 0) {
+    throw new import_common.BadRequestException('"modelId" must be a non-empty string.');
+  }
+  if (!isFiniteNonNegative(inputPricePer1m)) {
+    throw new import_common.BadRequestException('"inputPricePer1m" must be a non-negative number.');
+  }
+  if (!isFiniteNonNegative(outputPricePer1m)) {
+    throw new import_common.BadRequestException('"outputPricePer1m" must be a non-negative number.');
+  }
+  if (cacheWritePricePer1m !== void 0 && !isFiniteNonNegative(cacheWritePricePer1m)) {
+    throw new import_common.BadRequestException('"cacheWritePricePer1m" must be a non-negative number when present.');
+  }
+  if (cacheReadPricePer1m !== void 0 && !isFiniteNonNegative(cacheReadPricePer1m)) {
+    throw new import_common.BadRequestException('"cacheReadPricePer1m" must be a non-negative number when present.');
+  }
+  return {
+    modelId,
+    inputPricePer1m,
+    outputPricePer1m,
+    ...cacheWritePricePer1m !== void 0 ? {
+      cacheWritePricePer1m
+    } : {},
+    ...cacheReadPricePer1m !== void 0 ? {
+      cacheReadPricePer1m
+    } : {}
+  };
+}
+__name(parsePriceInput, "parsePriceInput");
+function isFiniteNonNegative(value) {
+  return typeof value === "number" && Number.isFinite(value) && value >= 0;
+}
+__name(isFiniteNonNegative, "isFiniteNonNegative");
+
 // src/server/tokens.ts
 var AGENT_GOVERNANCE_QUERIES = Symbol.for("@dudousxd/nestjs-agent:governance-queries");
+var AGENT_ACTOR_DIRECTORY = Symbol.for("@dudousxd/nestjs-agent:actor-directory");
+var AGENT_PRICING_STORE = Symbol.for("@dudousxd/nestjs-agent:pricing-store");
 var DASHBOARD_BASE_PATH = Symbol.for("@dudousxd/nestjs-agent-dashboard:base-path");
 var DASHBOARD_API_PATH = Symbol.for("@dudousxd/nestjs-agent-dashboard:api-path");
 
@@ -65,6 +111,7 @@ function _ts_param(paramIndex, decorator) {
   };
 }
 __name(_ts_param, "_ts_param");
+var PRICING_STORE_UNBOUND_MESSAGE = "Pricing CRUD is unavailable: no AGENT_PRICING_STORE is bound. Bind a pricing store (e.g. MikroOrmPricingStore from @dudousxd/nestjs-agent-store-mikro-orm) to enable it.";
 var AGENT_EVENTS = [
   "run.started",
   "message",
@@ -82,16 +129,21 @@ var DashboardService = class {
     __name(this, "DashboardService");
   }
   queries;
-  constructor(queries) {
+  actorDirectory;
+  pricingStore;
+  constructor(queries, actorDirectory, pricingStore) {
     this.queries = queries;
+    this.actorDirectory = actorDirectory;
+    this.pricingStore = pricingStore;
   }
   /** Spend/usage overview for a day range: by-model + by-actor spend and the daily trend, in parallel. */
   async spend(range) {
-    const [byModel, byActor, trend] = await Promise.all([
+    const [byModel, byActorRaw, trend] = await Promise.all([
       this.queries.spendByModel(range),
       this.queries.spendByActor(range),
       this.queries.usageTrend(range)
     ]);
+    const byActor = await this.withActorLabels(byActorRaw);
     return {
       byModel,
       byActor,
@@ -99,16 +151,60 @@ var DashboardService = class {
     };
   }
   /** Top threads by cost for a day range (default 10, highest cost first). */
-  topThreads(range, limit = 10) {
-    return this.queries.spendByThread(range, limit);
+  async topThreads(range, limit = 10) {
+    const rows = await this.queries.spendByThread(range, limit);
+    return this.withActorLabels(rows);
   }
   /** Most recent tool calls (status/type/thread) for the Runs & tools activity feed. */
   recentToolCalls(limit) {
     return this.queries.recentToolCalls(limit);
   }
   /** Most recent threads with rolled-up message/token counts. */
-  recentThreads(limit) {
-    return this.queries.recentThreads(limit);
+  async recentThreads(limit) {
+    const rows = await this.queries.recentThreads(limit);
+    return this.withActorLabels(rows);
+  }
+  /**
+  * Decorate actor-scoped rows with `actorLabel`, batching the distinct `actorRef`s into ONE
+  * {@link ActorDirectory.resolveDisplay} call per response. `null` for every row when no directory
+  * is bound, or for a ref the directory didn't resolve.
+  */
+  async withActorLabels(rows) {
+    if (rows.length === 0) {
+      return [];
+    }
+    if (this.actorDirectory === void 0) {
+      return rows.map((row) => ({
+        ...row,
+        actorLabel: null
+      }));
+    }
+    const refs = [
+      ...new Set(rows.map((row) => row.actorRef))
+    ];
+    const resolved = await this.actorDirectory.resolveDisplay(refs);
+    return rows.map((row) => ({
+      ...row,
+      actorLabel: resolved[row.actorRef] ?? null
+    }));
+  }
+  /** Current price row per model, for the pricing tab. 501s when no `AGENT_PRICING_STORE` is bound. */
+  async listPrices() {
+    if (this.pricingStore === void 0) {
+      throw new import_common2.NotImplementedException(PRICING_STORE_UNBOUND_MESSAGE);
+    }
+    return this.pricingStore.listCurrentPrices();
+  }
+  /**
+  * Set a model's current price (`POST <api>/pricing` body). 501s when no `AGENT_PRICING_STORE` is
+  * bound (checked BEFORE body validation, so an unbound store always reports as unimplemented rather
+  * than as a validation error); otherwise the body is minimally validated via {@link parsePriceInput}.
+  */
+  async upsertPrice(body) {
+    if (this.pricingStore === void 0) {
+      throw new import_common2.NotImplementedException(PRICING_STORE_UNBOUND_MESSAGE);
+    }
+    await this.pricingStore.upsertModelPrice(parsePriceInput(body));
   }
   /**
   * Live SSE stream of `aviary:agent:*` diagnostics events. One subscription per SSE client:
@@ -142,11 +238,17 @@ var DashboardService = class {
   }
 };
 DashboardService = _ts_decorate([
-  (0, import_common.Injectable)(),
-  _ts_param(0, (0, import_common.Inject)(AGENT_GOVERNANCE_QUERIES)),
+  (0, import_common2.Injectable)(),
+  _ts_param(0, (0, import_common2.Inject)(AGENT_GOVERNANCE_QUERIES)),
+  _ts_param(1, (0, import_common2.Optional)()),
+  _ts_param(1, (0, import_common2.Inject)(AGENT_ACTOR_DIRECTORY)),
+  _ts_param(2, (0, import_common2.Optional)()),
+  _ts_param(2, (0, import_common2.Inject)(AGENT_PRICING_STORE)),
   _ts_metadata("design:type", Function),
   _ts_metadata("design:paramtypes", [
-    typeof AgentGovernanceQueries === "undefined" ? Object : AgentGovernanceQueries
+    typeof AgentGovernanceQueries === "undefined" ? Object : AgentGovernanceQueries,
+    typeof ActorDirectory === "undefined" ? Object : ActorDirectory,
+    typeof AgentPricingStore === "undefined" ? Object : AgentPricingStore
   ])
 ], DashboardService);
 
@@ -215,15 +317,30 @@ var AgentApiController = class {
   threads(limit) {
     return this.dashboard.recentThreads(parseLimit(limit, 50));
   }
+  /**
+  * Current price row per model, for the pricing tab. 501s (via `DashboardService.listPrices`) when
+  * no `AGENT_PRICING_STORE` is bound.
+  */
+  listPrices() {
+    return this.dashboard.listPrices();
+  }
+  /**
+  * Set a model's current price. Body shape mirrors core's `ModelPriceInput`
+  * (`{ modelId, inputPricePer1m, outputPricePer1m, cacheWritePricePer1m?, cacheReadPricePer1m? }`).
+  * 501s when no `AGENT_PRICING_STORE` is bound; 400s on a malformed body.
+  */
+  upsertPrice(body) {
+    return this.dashboard.upsertPrice(body);
+  }
   /** Server-Sent Events stream of live `aviary:agent:*` events — the Live feed tails it. */
   stream() {
     return this.dashboard.streamEvents();
   }
 };
 _ts_decorate2([
-  (0, import_common2.Get)("spend"),
-  _ts_param2(0, (0, import_common2.Query)("from")),
-  _ts_param2(1, (0, import_common2.Query)("to")),
+  (0, import_common3.Get)("spend"),
+  _ts_param2(0, (0, import_common3.Query)("from")),
+  _ts_param2(1, (0, import_common3.Query)("to")),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", [
     String,
@@ -232,10 +349,10 @@ _ts_decorate2([
   _ts_metadata2("design:returntype", typeof Promise === "undefined" ? Object : Promise)
 ], AgentApiController.prototype, "spend", null);
 _ts_decorate2([
-  (0, import_common2.Get)("top-threads"),
-  _ts_param2(0, (0, import_common2.Query)("from")),
-  _ts_param2(1, (0, import_common2.Query)("to")),
-  _ts_param2(2, (0, import_common2.Query)("limit")),
+  (0, import_common3.Get)("top-threads"),
+  _ts_param2(0, (0, import_common3.Query)("from")),
+  _ts_param2(1, (0, import_common3.Query)("to")),
+  _ts_param2(2, (0, import_common3.Query)("limit")),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", [
     String,
@@ -245,8 +362,8 @@ _ts_decorate2([
   _ts_metadata2("design:returntype", typeof Promise === "undefined" ? Object : Promise)
 ], AgentApiController.prototype, "topThreads", null);
 _ts_decorate2([
-  (0, import_common2.Get)("tool-calls"),
-  _ts_param2(0, (0, import_common2.Query)("limit")),
+  (0, import_common3.Get)("tool-calls"),
+  _ts_param2(0, (0, import_common3.Query)("limit")),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", [
     String
@@ -254,8 +371,8 @@ _ts_decorate2([
   _ts_metadata2("design:returntype", typeof Promise === "undefined" ? Object : Promise)
 ], AgentApiController.prototype, "toolCalls", null);
 _ts_decorate2([
-  (0, import_common2.Get)("threads"),
-  _ts_param2(0, (0, import_common2.Query)("limit")),
+  (0, import_common3.Get)("threads"),
+  _ts_param2(0, (0, import_common3.Query)("limit")),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", [
     String
@@ -263,28 +380,69 @@ _ts_decorate2([
   _ts_metadata2("design:returntype", typeof Promise === "undefined" ? Object : Promise)
 ], AgentApiController.prototype, "threads", null);
 _ts_decorate2([
-  (0, import_common2.Sse)("stream"),
+  (0, import_common3.Get)("pricing"),
+  _ts_metadata2("design:type", Function),
+  _ts_metadata2("design:paramtypes", []),
+  _ts_metadata2("design:returntype", typeof Promise === "undefined" ? Object : Promise)
+], AgentApiController.prototype, "listPrices", null);
+_ts_decorate2([
+  (0, import_common3.Post)("pricing"),
+  _ts_param2(0, (0, import_common3.Body)()),
+  _ts_metadata2("design:type", Function),
+  _ts_metadata2("design:paramtypes", [
+    Object
+  ]),
+  _ts_metadata2("design:returntype", typeof Promise === "undefined" ? Object : Promise)
+], AgentApiController.prototype, "upsertPrice", null);
+_ts_decorate2([
+  (0, import_common3.Sse)("stream"),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", []),
   _ts_metadata2("design:returntype", typeof Observable === "undefined" ? Object : Observable)
 ], AgentApiController.prototype, "stream", null);
 AgentApiController = _ts_decorate2([
-  (0, import_common2.Controller)(),
+  (0, import_common3.Controller)(),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", [
     typeof DashboardService === "undefined" ? Object : DashboardService
   ])
 ], AgentApiController);
 
+// src/server/normalize-path.ts
+function normalizeDashboardPath(path) {
+  return `/${path.replace(/^\/+|\/+$/g, "")}`;
+}
+__name(normalizeDashboardPath, "normalizeDashboardPath");
+
+// src/server/agent-dashboard-mount-paths.ts
+function unprefixed(path) {
+  return path.replace(/^\/+/, "");
+}
+__name(unprefixed, "unprefixed");
+function agentDashboardMountPaths(options) {
+  const basePath = normalizeDashboardPath(options?.basePath ?? "/ai-gateway");
+  const apiBasePath = normalizeDashboardPath(options?.apiBasePath ?? `${basePath}/api`);
+  const base = unprefixed(basePath);
+  const api = unprefixed(apiBasePath);
+  return [
+    base,
+    `${base}/{*splat}`,
+    api,
+    `${api}/{*splat}`
+  ];
+}
+__name(agentDashboardMountPaths, "agentDashboardMountPaths");
+
 // src/server/agent-dashboard.module.ts
-var import_common4 = require("@nestjs/common");
+var import_reflect_metadata = require("reflect-metadata");
+var import_common5 = require("@nestjs/common");
 var import_core = require("@nestjs/core");
 
 // src/server/agent-ui.controller.ts
 var import_node_fs = require("fs");
 var import_node_path = require("path");
 var import_node_url = require("url");
-var import_common3 = require("@nestjs/common");
+var import_common4 = require("@nestjs/common");
 function _ts_decorate3(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);
@@ -332,7 +490,7 @@ var AgentUiController = class {
   index() {
     const indexPath = (0, import_node_path.join)(this.dir, "index.html");
     if (!(0, import_node_fs.existsSync)(indexPath)) {
-      throw new import_common3.NotFoundException("Dashboard is not built. Run the package build.");
+      throw new import_common4.NotFoundException("Dashboard is not built. Run the package build.");
     }
     const html = (0, import_node_fs.readFileSync)(indexPath, "utf8").replaceAll(`="${BUILD_BASE}/`, `="${this.basePath}/`);
     const inject = `<script>window.__AGENT_BASE__='${this.basePath}';window.__AGENT_API__='${this.apiBasePath}';</script>`;
@@ -340,40 +498,40 @@ var AgentUiController = class {
   }
   asset(file) {
     const safe = (0, import_node_path.basename)(file);
-    if (safe !== file) throw new import_common3.NotFoundException();
+    if (safe !== file) throw new import_common4.NotFoundException();
     const root = (0, import_node_path.resolve)(this.dir, "assets");
     const assetPath = (0, import_node_path.resolve)(root, safe);
     if (!assetPath.startsWith(root + import_node_path.sep) || !(0, import_node_fs.existsSync)(assetPath)) {
-      throw new import_common3.NotFoundException();
+      throw new import_common4.NotFoundException();
     }
     const type = CONTENT_TYPES[(0, import_node_path.extname)(safe)] ?? "application/octet-stream";
-    return new import_common3.StreamableFile((0, import_node_fs.readFileSync)(assetPath), {
+    return new import_common4.StreamableFile((0, import_node_fs.readFileSync)(assetPath), {
       type
     });
   }
 };
 _ts_decorate3([
-  (0, import_common3.Get)(),
-  (0, import_common3.Header)("Content-Type", "text/html; charset=utf-8"),
-  (0, import_common3.Header)("Cache-Control", "no-store, must-revalidate"),
+  (0, import_common4.Get)(),
+  (0, import_common4.Header)("Content-Type", "text/html; charset=utf-8"),
+  (0, import_common4.Header)("Cache-Control", "no-store, must-revalidate"),
   _ts_metadata3("design:type", Function),
   _ts_metadata3("design:paramtypes", []),
   _ts_metadata3("design:returntype", String)
 ], AgentUiController.prototype, "index", null);
 _ts_decorate3([
-  (0, import_common3.Get)("assets/:file"),
-  (0, import_common3.Header)("Cache-Control", "public, max-age=31536000, immutable"),
-  _ts_param3(0, (0, import_common3.Param)("file")),
+  (0, import_common4.Get)("assets/:file"),
+  (0, import_common4.Header)("Cache-Control", "public, max-age=31536000, immutable"),
+  _ts_param3(0, (0, import_common4.Param)("file")),
   _ts_metadata3("design:type", Function),
   _ts_metadata3("design:paramtypes", [
     String
   ]),
-  _ts_metadata3("design:returntype", typeof import_common3.StreamableFile === "undefined" ? Object : import_common3.StreamableFile)
+  _ts_metadata3("design:returntype", typeof import_common4.StreamableFile === "undefined" ? Object : import_common4.StreamableFile)
 ], AgentUiController.prototype, "asset", null);
 AgentUiController = _ts_decorate3([
-  (0, import_common3.Controller)(),
-  _ts_param3(0, (0, import_common3.Inject)(DASHBOARD_BASE_PATH)),
-  _ts_param3(1, (0, import_common3.Inject)(DASHBOARD_API_PATH)),
+  (0, import_common4.Controller)(),
+  _ts_param3(0, (0, import_common4.Inject)(DASHBOARD_BASE_PATH)),
+  _ts_param3(1, (0, import_common4.Inject)(DASHBOARD_API_PATH)),
   _ts_metadata3("design:type", Function),
   _ts_metadata3("design:paramtypes", [
     String,
@@ -389,27 +547,42 @@ function _ts_decorate4(decorators, target, key, desc) {
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
 __name(_ts_decorate4, "_ts_decorate");
+var GUARDS_METADATA = "__guards__";
 function normalize(path) {
-  return `/${path.replace(/^\/+|\/+$/g, "")}`;
+  return normalizeDashboardPath(path);
 }
 __name(normalize, "normalize");
-var AgentApiModule = class {
+function stampGuards(guards, ...controllers) {
+  for (const controller of controllers) {
+    Reflect.defineMetadata(GUARDS_METADATA, guards ?? [], controller);
+  }
+}
+__name(stampGuards, "stampGuards");
+var AgentApiModule = class _AgentApiModule {
   static {
     __name(this, "AgentApiModule");
   }
+  static register(options) {
+    return {
+      module: _AgentApiModule,
+      imports: [
+        ...options.imports ?? []
+      ],
+      controllers: [
+        AgentApiController
+      ],
+      providers: [
+        DashboardService,
+        ...options.guards ?? []
+      ],
+      exports: [
+        DashboardService
+      ]
+    };
+  }
 };
 AgentApiModule = _ts_decorate4([
-  (0, import_common4.Module)({
-    controllers: [
-      AgentApiController
-    ],
-    providers: [
-      DashboardService
-    ],
-    exports: [
-      DashboardService
-    ]
-  })
+  (0, import_common5.Module)({})
 ], AgentApiModule);
 var AgentDashboardModule = class _AgentDashboardModule {
   static {
@@ -418,10 +591,22 @@ var AgentDashboardModule = class _AgentDashboardModule {
   static forRoot(options = {}) {
     const basePath = normalize(options.basePath ?? "/ai-gateway");
     const apiBasePath = normalize(options.apiBasePath ?? `${basePath}/api`);
+    stampGuards(options.guards, AgentApiController, AgentUiController);
     return {
       module: _AgentDashboardModule,
       imports: [
-        AgentApiModule,
+        ...options.imports ?? [],
+        // Guards + host imports must reach the API controller's HOST module — enhancers resolve
+        // from their controller's own module, never from a parent (see AgentApiModule.register).
+        // Spread-only-when-set: exactOptionalPropertyTypes rejects an explicit `undefined`.
+        AgentApiModule.register({
+          ...options.imports ? {
+            imports: options.imports
+          } : {},
+          ...options.guards ? {
+            guards: options.guards
+          } : {}
+        }),
         import_core.RouterModule.register([
           {
             path: basePath,
@@ -444,7 +629,9 @@ var AgentDashboardModule = class _AgentDashboardModule {
         {
           provide: DASHBOARD_API_PATH,
           useValue: apiBasePath
-        }
+        },
+        // AgentUiController is hosted HERE, so its guards DI-instantiate from this module.
+        ...options.guards ?? []
       ],
       // Re-export the API module so its DashboardService reaches importers (e.g. the host's own controllers).
       exports: [
@@ -454,17 +641,21 @@ var AgentDashboardModule = class _AgentDashboardModule {
   }
 };
 AgentDashboardModule = _ts_decorate4([
-  (0, import_common4.Module)({})
+  (0, import_common5.Module)({})
 ], AgentDashboardModule);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AGENT_ACTOR_DIRECTORY,
   AGENT_GOVERNANCE_QUERIES,
+  AGENT_PRICING_STORE,
   AgentApiController,
   AgentApiModule,
   AgentDashboardModule,
   AgentUiController,
   DASHBOARD_API_PATH,
   DASHBOARD_BASE_PATH,
-  DashboardService
+  DashboardService,
+  agentDashboardMountPaths,
+  parsePriceInput
 });
 //# sourceMappingURL=index.cjs.map