plugin-ai-api 1.0.15 → 1.0.21

package/dist/server/billing.js

@@ -0,0 +1,331 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var billing_exports = {};
+__export(billing_exports, {
+  AiApiQuotaError: () => AiApiQuotaError,
+  calculateLlmCost: () => calculateLlmCost,
+  finalizeLlmBilling: () => finalizeLlmBilling,
+  markLlmProviderAttempted: () => markLlmProviderAttempted,
+  prepareLlmBilling: () => prepareLlmBilling
+});
+module.exports = __toCommonJS(billing_exports);
+var import_dayjs = __toESM(require("dayjs"));
+var import_utc = __toESM(require("dayjs/plugin/utc"));
+var import_timezone = __toESM(require("dayjs/plugin/timezone"));
+import_dayjs.default.extend(import_utc.default);
+import_dayjs.default.extend(import_timezone.default);
+const PRICE_SCALE = 10;
+const COST_SCALE = 8;
+const PRICE_TO_COST_DIVISOR = 100000000n;
+class AiApiQuotaError extends Error {
+  constructor(code, message) {
+    super(message);
+    this.code = code;
+    this.name = "AiApiQuotaError";
+  }
+}
+function stateOf(ctx) {
+  return ctx.state;
+}
+function valueOf(model, name) {
+  if (!model) return void 0;
+  if (typeof model.get === "function") return model.get(name);
+  return model[name];
+}
+function decimalString(value, scale) {
+  const source = String(value ?? "0").trim();
+  const match = source.match(/^(-?)(\d+)(?:\.(\d+))?$/);
+  if (!match) throw new Error(`Invalid decimal value: ${source}`);
+  const fraction = (match[3] ?? "").padEnd(scale, "0").slice(0, scale);
+  return `${match[1]}${match[2]}.${fraction}`;
+}
+function decimalUnits(value, scale) {
+  return BigInt(decimalString(value, scale).replace(".", ""));
+}
+function formatUnits(value, scale) {
+  const sign = value < 0n ? "-" : "";
+  const digits = (value < 0n ? -value : value).toString().padStart(scale + 1, "0");
+  if (scale === 0) return `${sign}${digits}`;
+  return `${sign}${digits.slice(0, -scale)}.${digits.slice(-scale)}`;
+}
+function divideRounded(value, divisor) {
+  return (value + divisor / 2n) / divisor;
+}
+function calculateCostUnits(inputTokens, outputTokens, price) {
+  const input = divideRounded(
+    BigInt(inputTokens) * decimalUnits(price.inputPricePerMillionTokens, PRICE_SCALE),
+    PRICE_TO_COST_DIVISOR
+  );
+  const output = divideRounded(
+    BigInt(outputTokens) * decimalUnits(price.outputPricePerMillionTokens, PRICE_SCALE),
+    PRICE_TO_COST_DIVISOR
+  );
+  const fixed = divideRounded(decimalUnits(price.fixedCostPerRequest, PRICE_SCALE), 100n);
+  return input + output + fixed;
+}
+function calculateLlmCost(inputTokens, outputTokens, price) {
+  return formatUnits(calculateCostUnits(inputTokens, outputTokens, price), COST_SCALE);
+}
+function normalizePositiveInteger(value, fallback) {
+  const parsed = Number(value);
+  return Number.isSafeInteger(parsed) && parsed > 0 ? parsed : fallback;
+}
+function estimateInputTokens(ctx) {
+  const body = ctx.request.body ?? {};
+  const input = body.messages ?? body.prompt ?? "";
+  return Math.max(1, Math.ceil(JSON.stringify(input).length / 4));
+}
+function getPeriodBounds(periodType, timezone2) {
+  const unit = periodType === "daily" ? "day" : "month";
+  try {
+    const start = (0, import_dayjs.default)().tz(timezone2 || "UTC").startOf(unit);
+    return { start: start.utc().toDate(), end: start.add(1, unit).utc().toDate() };
+  } catch {
+    const start = (0, import_dayjs.default)().utc().startOf(unit);
+    return { start: start.toDate(), end: start.add(1, unit).toDate() };
+  }
+}
+function exceedsIntegerLimit(current, added, limit) {
+  if (limit === null || limit === void 0 || limit === "") return false;
+  return current + added > BigInt(String(limit));
+}
+function exceedsDecimalLimit(current, added, limit) {
+  if (limit === null || limit === void 0 || limit === "") return false;
+  return current + added > decimalUnits(limit, COST_SCALE);
+}
+async function findPrice(ctx, service, modelId) {
+  const now = /* @__PURE__ */ new Date();
+  const price = await ctx.db.getRepository("aiApiModelPrices").findOne({
+    filter: {
+      llmService: valueOf(service, "name"),
+      model: modelId,
+      enabled: true,
+      effectiveFrom: { $lte: now },
+      $or: [{ effectiveTo: null }, { effectiveTo: { $gt: now } }]
+    },
+    sort: "-effectiveFrom"
+  });
+  if (!price) return void 0;
+  return {
+    id: valueOf(price, "id"),
+    currency: valueOf(price, "currency"),
+    inputPricePerMillionTokens: decimalString(valueOf(price, "inputPricePerMillionTokens"), PRICE_SCALE),
+    outputPricePerMillionTokens: decimalString(valueOf(price, "outputPricePerMillionTokens"), PRICE_SCALE),
+    fixedCostPerRequest: decimalString(valueOf(price, "fixedCostPerRequest"), PRICE_SCALE)
+  };
+}
+async function prepareLlmBilling(ctx, resolved) {
+  var _a;
+  const userId = (_a = stateOf(ctx).currentUser) == null ? void 0 : _a.id;
+  const serviceName = valueOf(resolved.service, "name");
+  const provider = valueOf(resolved.service, "provider");
+  const price = await findPrice(ctx, resolved.service, resolved.modelId);
+  const billing = {
+    resolution: { service: serviceName, provider, model: resolved.modelId },
+    price
+  };
+  stateOf(ctx).aiApiLlmBilling = billing;
+  const config = await ctx.db.getRepository("aiApiConfig").findOne();
+  if (!valueOf(config, "quotaEnabled") || userId === void 0 || userId === null) return;
+  const policy = await ctx.db.getRepository("aiApiUserQuotaPolicies").findOne({
+    filter: { userId, enabled: true },
+    sort: "-updatedAt"
+  });
+  if (!policy) return;
+  const rejectUnpriced = valueOf(policy, "rejectUnpricedModel");
+  if (!price && rejectUnpriced) {
+    throw new AiApiQuotaError(
+      "model_price_not_configured",
+      `Pricing is not configured for '${serviceName}/${resolved.modelId}'.`
+    );
+  }
+  const policyCurrency = valueOf(policy, "currency");
+  if (price && policyCurrency !== price.currency) {
+    throw new AiApiQuotaError(
+      "quota_currency_mismatch",
+      `Quota currency '${policyCurrency}' does not match model price currency '${price.currency}'.`
+    );
+  }
+  const body = ctx.request.body ?? {};
+  const estimatedInputTokens = estimateInputTokens(ctx);
+  const defaultOutput = normalizePositiveInteger(valueOf(config, "defaultReservationOutputTokens"), 4096);
+  const estimatedOutputTokens = normalizePositiveInteger(body.max_completion_tokens ?? body.max_tokens, defaultOutput);
+  const reservedTokens = estimatedInputTokens + estimatedOutputTokens;
+  const reservedCost = price ? calculateLlmCost(estimatedInputTokens, estimatedOutputTokens, price) : "0.00000000";
+  const period = getPeriodBounds(valueOf(policy, "periodType"), valueOf(policy, "timezone"));
+  const Bucket = ctx.db.getModel("aiApiUserQuotaBuckets");
+  const reservation = await ctx.db.sequelize.transaction(async (transaction) => {
+    const [bucket] = await Bucket.findOrCreate({
+      where: { policyId: valueOf(policy, "id"), periodStart: period.start },
+      defaults: {
+        userId,
+        periodEnd: period.end,
+        requestCount: 0,
+        totalTokens: 0,
+        cost: "0.00000000",
+        reservedRequests: 0,
+        reservedTokens: 0,
+        reservedCost: "0.00000000"
+      },
+      transaction
+    });
+    await bucket.reload({ transaction, lock: transaction.LOCK.UPDATE });
+    const requestCount = BigInt(String(bucket.get("requestCount") ?? 0));
+    const reservedRequests = BigInt(String(bucket.get("reservedRequests") ?? 0));
+    if (exceedsIntegerLimit(requestCount + reservedRequests, 1n, valueOf(policy, "requestLimit"))) {
+      throw new AiApiQuotaError("request_quota_exceeded", "The request quota for this user has been exceeded.");
+    }
+    const totalTokens = BigInt(String(bucket.get("totalTokens") ?? 0));
+    const alreadyReservedTokens = BigInt(String(bucket.get("reservedTokens") ?? 0));
+    if (exceedsIntegerLimit(
+      totalTokens + alreadyReservedTokens,
+      BigInt(reservedTokens),
+      valueOf(policy, "totalTokenLimit")
+    )) {
+      throw new AiApiQuotaError("token_quota_exceeded", "The token quota for this user has been exceeded.");
+    }
+    const cost = decimalUnits(bucket.get("cost"), COST_SCALE);
+    const alreadyReservedCost = decimalUnits(bucket.get("reservedCost"), COST_SCALE);
+    if (exceedsDecimalLimit(
+      cost + alreadyReservedCost,
+      decimalUnits(reservedCost, COST_SCALE),
+      valueOf(policy, "costLimit")
+    )) {
+      throw new AiApiQuotaError("cost_quota_exceeded", "The cost quota for this user has been exceeded.");
+    }
+    await bucket.update(
+      {
+        reservedRequests: formatUnits(reservedRequests + 1n, 0),
+        reservedTokens: formatUnits(alreadyReservedTokens + BigInt(reservedTokens), 0),
+        reservedCost: formatUnits(alreadyReservedCost + decimalUnits(reservedCost, COST_SCALE), COST_SCALE)
+      },
+      { transaction }
+    );
+    return {
+      bucketId: bucket.get("id"),
+      policyId: valueOf(policy, "id"),
+      estimatedInputTokens,
+      estimatedOutputTokens,
+      reservedTokens,
+      reservedCost,
+      missingUsageBehavior: valueOf(policy, "missingUsageBehavior") === "allow" ? "allow" : "use_reserved"
+    };
+  });
+  billing.reservation = reservation;
+}
+function markLlmProviderAttempted(ctx) {
+  const state = stateOf(ctx);
+  state.aiApiLlmBilling = { ...state.aiApiLlmBilling ?? {}, providerAttempted: true };
+}
+function usageNumbers(usage) {
+  if (!usage || usage.prompt_tokens === null || usage.completion_tokens === null) return void 0;
+  return {
+    input: usage.prompt_tokens,
+    output: usage.completion_tokens,
+    total: usage.total_tokens ?? usage.prompt_tokens + usage.completion_tokens
+  };
+}
+async function finalizeLlmBilling(ctx, providerUsage, succeeded) {
+  var _a, _b, _c, _d, _e, _f;
+  const billing = stateOf(ctx).aiApiLlmBilling;
+  if (!billing) return {};
+  let numbers = usageNumbers(providerUsage);
+  let costStatus;
+  if (numbers) {
+    costStatus = billing.price ? "calculated" : "unpriced";
+  } else if (succeeded && ((_a = billing.reservation) == null ? void 0 : _a.missingUsageBehavior) === "use_reserved") {
+    numbers = {
+      input: billing.reservation.estimatedInputTokens,
+      output: billing.reservation.estimatedOutputTokens,
+      total: billing.reservation.reservedTokens
+    };
+    costStatus = billing.price ? "estimated" : "unpriced";
+  } else {
+    costStatus = billing.price ? "usage_unavailable" : "unpriced";
+  }
+  const cost = numbers && billing.price ? calculateLlmCost(numbers.input, numbers.output, billing.price) : void 0;
+  const reservation = billing.reservation;
+  if (reservation) {
+    const Bucket = ctx.db.getModel("aiApiUserQuotaBuckets");
+    await ctx.db.sequelize.transaction(async (transaction) => {
+      const bucket = await Bucket.findByPk(reservation.bucketId, { transaction, lock: transaction.LOCK.UPDATE });
+      if (!bucket) return;
+      const reservedRequests = BigInt(String(bucket.get("reservedRequests") ?? 0));
+      const reservedTokens = BigInt(String(bucket.get("reservedTokens") ?? 0));
+      const reservedCost = decimalUnits(bucket.get("reservedCost"), COST_SCALE);
+      const requestCount = BigInt(String(bucket.get("requestCount") ?? 0));
+      const totalTokens = BigInt(String(bucket.get("totalTokens") ?? 0));
+      const currentCost = decimalUnits(bucket.get("cost"), COST_SCALE);
+      await bucket.update(
+        {
+          reservedRequests: formatUnits(reservedRequests > 0n ? reservedRequests - 1n : 0n, 0),
+          reservedTokens: formatUnits(
+            reservedTokens >= BigInt(reservation.reservedTokens) ? reservedTokens - BigInt(reservation.reservedTokens) : 0n,
+            0
+          ),
+          reservedCost: formatUnits(
+            reservedCost >= decimalUnits(reservation.reservedCost, COST_SCALE) ? reservedCost - decimalUnits(reservation.reservedCost, COST_SCALE) : 0n,
+            COST_SCALE
+          ),
+          requestCount: formatUnits(requestCount + (billing.providerAttempted ? 1n : 0n), 0),
+          totalTokens: formatUnits(totalTokens + BigInt((numbers == null ? void 0 : numbers.total) ?? 0), 0),
+          cost: formatUnits(currentCost + decimalUnits(cost ?? "0", COST_SCALE), COST_SCALE)
+        },
+        { transaction }
+      );
+    });
+  }
+  return {
+    usage: numbers ? { prompt_tokens: numbers.input, completion_tokens: numbers.output, total_tokens: numbers.total } : providerUsage,
+    estimatedCost: cost,
+    currency: (_b = billing.price) == null ? void 0 : _b.currency,
+    costStatus,
+    modelPriceId: (_c = billing.price) == null ? void 0 : _c.id,
+    quotaPolicyId: reservation == null ? void 0 : reservation.policyId,
+    inputPricePerMillionTokens: (_d = billing.price) == null ? void 0 : _d.inputPricePerMillionTokens,
+    outputPricePerMillionTokens: (_e = billing.price) == null ? void 0 : _e.outputPricePerMillionTokens,
+    fixedCostPerRequest: (_f = billing.price) == null ? void 0 : _f.fixedCostPerRequest
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AiApiQuotaError,
+  calculateLlmCost,
+  finalizeLlmBilling,
+  markLlmProviderAttempted,
+  prepareLlmBilling
+});

package/dist/server/collections/ai-api-config.js

@@ -62,6 +62,24 @@ var ai_api_config_default = (0, import_database.defineCollection)({
       defaultValue: 60,
       comment: "Max requests per user per minute"
     },
+    {
+      name: "maxRequestBodyMb",
+      type: "integer",
+      defaultValue: 10,
+      comment: "Max request body size in MB. Raise this to accept inline base64 images in vision requests."
+    },
+    {
+      name: "quotaEnabled",
+      type: "boolean",
+      defaultValue: false,
+      comment: "Enable per-user request, token, and cost quotas for direct LLM mode"
+    },
+    {
+      name: "defaultReservationOutputTokens",
+      type: "integer",
+      defaultValue: 4096,
+      comment: "Output tokens reserved when a request does not specify a maximum"
+    },
     {
       name: "options",
       type: "jsonb",

package/dist/server/collections/ai-api-model-metadata.js

@@ -0,0 +1,83 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_api_model_metadata_exports = {};
+__export(ai_api_model_metadata_exports, {
+  default: () => ai_api_model_metadata_default
+});
+module.exports = __toCommonJS(ai_api_model_metadata_exports);
+var import_database = require("@nocobase/database");
+var ai_api_model_metadata_default = (0, import_database.defineCollection)({
+  name: "aiApiModelMetadata",
+  autoGenId: true,
+  fields: [
+    { name: "llmService", type: "string", allowNull: false, index: true },
+    { name: "model", type: "string", allowNull: false, index: true },
+    {
+      name: "contextWindow",
+      type: "integer",
+      allowNull: true,
+      comment: "Override for the model context window (input+output token capacity). Returned as context_window / context_length."
+    },
+    {
+      name: "maxCompletionTokens",
+      type: "integer",
+      allowNull: true,
+      comment: "Override for the maximum output tokens. Returned as max_completion_tokens."
+    },
+    {
+      name: "ownedByOverride",
+      type: "string",
+      allowNull: true,
+      comment: "Override for the owned_by field in the OpenAI model object."
+    },
+    {
+      name: "displayName",
+      type: "string",
+      allowNull: true,
+      comment: "Friendly display name returned as display_name / name in the model object."
+    },
+    {
+      name: "description",
+      type: "text",
+      allowNull: true,
+      comment: "Human-readable description returned as description in the model object."
+    },
+    {
+      name: "enabled",
+      type: "boolean",
+      defaultValue: true,
+      index: true,
+      comment: "When false, the model is hidden from /v1/models and reported as active:false."
+    }
+  ],
+  indexes: [
+    {
+      fields: ["llmService", "model"],
+      unique: true
+    }
+  ]
+});

package/dist/server/collections/ai-api-model-prices.js

@@ -0,0 +1,55 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_api_model_prices_exports = {};
+__export(ai_api_model_prices_exports, {
+  default: () => ai_api_model_prices_default
+});
+module.exports = __toCommonJS(ai_api_model_prices_exports);
+var import_database = require("@nocobase/database");
+var ai_api_model_prices_default = (0, import_database.defineCollection)({
+  name: "aiApiModelPrices",
+  autoGenId: true,
+  fields: [
+    { name: "llmService", type: "string", allowNull: false, index: true },
+    { name: "provider", type: "string", allowNull: false, index: true },
+    { name: "model", type: "string", allowNull: false, index: true },
+    { name: "currency", type: "string", allowNull: false, defaultValue: "USD" },
+    { name: "inputPricePerMillionTokens", type: "decimal", precision: 20, scale: 10, allowNull: false },
+    { name: "outputPricePerMillionTokens", type: "decimal", precision: 20, scale: 10, allowNull: false },
+    { name: "fixedCostPerRequest", type: "decimal", precision: 20, scale: 10, allowNull: false, defaultValue: 0 },
+    { name: "effectiveFrom", type: "datetimeTz", allowNull: false, index: true },
+    { name: "effectiveTo", type: "datetimeTz", allowNull: true, index: true },
+    { name: "enabled", type: "boolean", defaultValue: true, index: true },
+    { name: "notes", type: "text", allowNull: true }
+  ],
+  indexes: [
+    {
+      fields: ["llmService", "model", "effectiveFrom"],
+      unique: true
+    }
+  ]
+});

package/dist/server/collections/ai-api-usage-records.js

@@ -52,6 +52,9 @@ var ai_api_usage_records_default = (0, import_database.defineCollection)({
     { name: "endpoint", type: "string" },
     { name: "mode", type: "string", allowNull: true },
     { name: "model", type: "string", allowNull: true, index: true },
+    { name: "resolvedService", type: "string", allowNull: true, index: true },
+    { name: "resolvedProvider", type: "string", allowNull: true, index: true },
+    { name: "resolvedModel", type: "string", allowNull: true, index: true },
     { name: "status", type: "string", index: true },
     { name: "httpStatus", type: "integer", allowNull: true },
     { name: "errorCode", type: "string", allowNull: true },
@@ -61,6 +64,12 @@ var ai_api_usage_records_default = (0, import_database.defineCollection)({
     { name: "totalTokens", type: "integer", allowNull: true },
     { name: "estimatedCost", type: "decimal", allowNull: true, precision: 20, scale: 8 },
     { name: "currency", type: "string", allowNull: true },
+    { name: "costStatus", type: "string", allowNull: true, index: true },
+    { name: "modelPriceId", type: "bigInt", allowNull: true, index: true },
+    { name: "quotaPolicyId", type: "bigInt", allowNull: true, index: true },
+    { name: "inputPricePerMillionTokens", type: "decimal", allowNull: true, precision: 20, scale: 10 },
+    { name: "outputPricePerMillionTokens", type: "decimal", allowNull: true, precision: 20, scale: 10 },
+    { name: "fixedCostPerRequest", type: "decimal", allowNull: true, precision: 20, scale: 10 },
     { name: "providerRequestId", type: "string", allowNull: true },
     { name: "requestMetadata", type: "jsonb", defaultValue: {} },
     { name: "responseMetadata", type: "jsonb", defaultValue: {} },

package/dist/server/collections/ai-api-user-quota-buckets.js

@@ -0,0 +1,54 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_api_user_quota_buckets_exports = {};
+__export(ai_api_user_quota_buckets_exports, {
+  default: () => ai_api_user_quota_buckets_default
+});
+module.exports = __toCommonJS(ai_api_user_quota_buckets_exports);
+var import_database = require("@nocobase/database");
+var ai_api_user_quota_buckets_default = (0, import_database.defineCollection)({
+  name: "aiApiUserQuotaBuckets",
+  autoGenId: true,
+  fields: [
+    { name: "policyId", type: "bigInt", allowNull: false, index: true },
+    { name: "userId", type: "bigInt", allowNull: false, index: true },
+    { name: "periodStart", type: "datetimeTz", allowNull: false, index: true },
+    { name: "periodEnd", type: "datetimeTz", allowNull: false, index: true },
+    { name: "requestCount", type: "bigInt", allowNull: false, defaultValue: 0 },
+    { name: "totalTokens", type: "bigInt", allowNull: false, defaultValue: 0 },
+    { name: "cost", type: "decimal", precision: 20, scale: 8, allowNull: false, defaultValue: 0 },
+    { name: "reservedRequests", type: "bigInt", allowNull: false, defaultValue: 0 },
+    { name: "reservedTokens", type: "bigInt", allowNull: false, defaultValue: 0 },
+    { name: "reservedCost", type: "decimal", precision: 20, scale: 8, allowNull: false, defaultValue: 0 }
+  ],
+  indexes: [
+    {
+      fields: ["policyId", "periodStart"],
+      unique: true
+    }
+  ]
+});

package/dist/server/collections/ai-api-user-quota-policies.js

@@ -0,0 +1,62 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_api_user_quota_policies_exports = {};
+__export(ai_api_user_quota_policies_exports, {
+  default: () => ai_api_user_quota_policies_default
+});
+module.exports = __toCommonJS(ai_api_user_quota_policies_exports);
+var import_database = require("@nocobase/database");
+var ai_api_user_quota_policies_default = (0, import_database.defineCollection)({
+  name: "aiApiUserQuotaPolicies",
+  autoGenId: true,
+  fields: [
+    { name: "userId", type: "bigInt", allowNull: false, index: true },
+    {
+      name: "user",
+      type: "belongsTo",
+      target: "users",
+      targetKey: "id",
+      foreignKey: "userId",
+      constraints: false
+    },
+    { name: "enabled", type: "boolean", defaultValue: true, index: true },
+    { name: "periodType", type: "string", allowNull: false, defaultValue: "monthly" },
+    { name: "timezone", type: "string", allowNull: false, defaultValue: "UTC" },
+    { name: "requestLimit", type: "bigInt", allowNull: true },
+    { name: "totalTokenLimit", type: "bigInt", allowNull: true },
+    { name: "costLimit", type: "decimal", precision: 20, scale: 8, allowNull: true },
+    { name: "currency", type: "string", allowNull: false, defaultValue: "USD" },
+    { name: "rejectUnpricedModel", type: "boolean", defaultValue: true },
+    { name: "missingUsageBehavior", type: "string", allowNull: false, defaultValue: "use_reserved" }
+  ],
+  indexes: [
+    {
+      fields: ["userId"],
+      unique: true
+    }
+  ]
+});