plugin-ai-api 1.0.24 → 1.0.28

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

@@ -0,0 +1,74 @@
+/**
+ * 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_usage_groups_exports = {};
+__export(ai_api_usage_groups_exports, {
+  default: () => ai_api_usage_groups_default
+});
+module.exports = __toCommonJS(ai_api_usage_groups_exports);
+var import_database = require("@nocobase/database");
+var ai_api_usage_groups_default = (0, import_database.defineCollection)({
+  name: "aiApiUsageGroups",
+  autoGenId: true,
+  fields: [
+    { name: "name", type: "string", allowNull: false },
+    { name: "isDefault", type: "boolean", defaultValue: false, index: true },
+    { name: "quotaMode", type: "string", allowNull: false, defaultValue: "per_user" },
+    { name: "rateLimitPerMinute", type: "integer", allowNull: false, defaultValue: 60 },
+    { 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" },
+    { name: "contextOverflowBehavior", type: "string", allowNull: false, defaultValue: "reject" },
+    // Model access: empty lists mean "no narrowing" — the group inherits the full
+    // global configuration. Non-empty lists narrow what members may use.
+    {
+      name: "allowedLlmServices",
+      type: "json",
+      defaultValue: [],
+      comment: "Empty means all globally enabled services; otherwise narrows to these services."
+    },
+    { name: "allowAllModels", type: "boolean", defaultValue: true },
+    {
+      name: "allowedModels",
+      type: "json",
+      defaultValue: [],
+      comment: 'Array of "serviceName/modelId" members may use (when allowAllModels=false).'
+    }
+  ],
+  indexes: [
+    {
+      fields: ["isDefault"],
+      unique: true,
+      where: { isDefault: true }
+    }
+  ]
+});

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

@@ -62,11 +62,13 @@ var ai_api_usage_records_default = (0, import_database.defineCollection)({
     { name: "inputTokens", type: "integer", allowNull: true },
     { name: "outputTokens", type: "integer", allowNull: true },
     { name: "totalTokens", type: "integer", allowNull: true },
+    { name: "promptCacheTokens", 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: "groupId", 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 },

package/dist/server/middleware/rate-limit.js

@@ -30,6 +30,7 @@ __export(rate_limit_exports, {
 });
 module.exports = __toCommonJS(rate_limit_exports);
 var import_openai_format = require("../utils/openai-format");
+var import_request_cache = require("../utils/request-cache");
 function createRateLimitMiddleware(limiter) {
   return async (ctx) => {
     var _a, _b, _c;
@@ -37,13 +38,13 @@ function createRateLimitMiddleware(limiter) {
     if (userId === void 0 || userId === null) return true;
     let limit = 60;
     try {
-      const config = await ctx.db.getRepository("aiApiConfig").findOne();
-      const configLimit = config == null ? void 0 : config.rateLimitPerMinute;
-      if (configLimit && configLimit > 0) {
-        limit = configLimit;
+      const group = await (0, import_request_cache.resolveRequestUserGroup)(ctx, userId);
+      const groupLimit = group == null ? void 0 : group.rateLimitPerMinute;
+      if (groupLimit && groupLimit > 0) {
+        limit = groupLimit;
       }
-    } catch (configErr) {
-      (_c = (_b = ctx.app) == null ? void 0 : _b.logger) == null ? void 0 : _c.warn("[ai-api] Rate limit config read failed, using default (60/min)", configErr);
+    } catch (err) {
+      (_c = (_b = ctx.app) == null ? void 0 : _b.logger) == null ? void 0 : _c.warn("[ai-api] Rate limit group resolution failed, using default (60/min)", err);
     }
     const result = limiter.check(userId, limit);
     if (!result.allowed) {

package/dist/server/migrations/20260813000000-add-prompt-cache-tokens.js

@@ -0,0 +1,69 @@
+/**
+ * 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 add_prompt_cache_tokens_exports = {};
+__export(add_prompt_cache_tokens_exports, {
+  default: () => AddPromptCacheTokensToUsageRecords
+});
+module.exports = __toCommonJS(add_prompt_cache_tokens_exports);
+var import_server = require("@nocobase/server");
+class AddPromptCacheTokensToUsageRecords extends import_server.Migration {
+  on = "beforeLoad";
+  async up() {
+    const collection = this.db.getCollection("aiApiUsageRecords");
+    if (!collection) return;
+    const field = collection.getField("promptCacheTokens");
+    if (!field) {
+      collection.addField("promptCacheTokens", { type: "integer", allowNull: true });
+    }
+    if (await collection.existsInDb()) {
+      const tableName = collection.getTableNameWithSchema();
+      const exists = await this.tableColumnExists(tableName, "promptCacheTokens");
+      if (!exists) {
+        await this.queryInterface.addColumn(tableName, "promptCacheTokens", {
+          type: this.db.sequelize.getDialect() === "sqlite" ? "INTEGER" : "INTEGER",
+          allowNull: true
+        });
+      }
+    }
+  }
+  async down() {
+    const collection = this.db.getCollection("aiApiUsageRecords");
+    if (!collection) return;
+    if (await collection.existsInDb()) {
+      const tableName = collection.getTableNameWithSchema();
+      const exists = await this.tableColumnExists(tableName, "promptCacheTokens");
+      if (exists) {
+        await this.queryInterface.removeColumn(tableName, "promptCacheTokens");
+      }
+    }
+    collection.removeField("promptCacheTokens");
+  }
+  async tableColumnExists(tableName, columnName) {
+    const columns = await this.queryInterface.describeTable(tableName);
+    return Object.prototype.hasOwnProperty.call(columns, columnName);
+  }
+}

package/dist/server/migrations/20260815000000-add-usage-groups.js

@@ -0,0 +1,149 @@
+/**
+ * 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 add_usage_groups_exports = {};
+__export(add_usage_groups_exports, {
+  default: () => AddUsageGroups
+});
+module.exports = __toCommonJS(add_usage_groups_exports);
+var import_server = require("@nocobase/server");
+class AddUsageGroups extends import_server.Migration {
+  // Must run afterSync: plugin collections are only registered once the app has
+  // loaded, and the new group tables must exist before seeding.
+  on = "afterSync";
+  async up() {
+    const groupCollection = this.getGroupCollection();
+    if (!groupCollection) {
+      throw new Error("AI API usage groups migration could not resolve the aiApiUsageGroups collection.");
+    }
+    const configCollection = this.db.getCollection("aiApiConfig");
+    let rateLimitPerMinute = 60;
+    let quotaEnabled = false;
+    if (configCollection && await configCollection.existsInDb()) {
+      const tableName = configCollection.getTableNameWithSchema();
+      const columns = await this.queryInterface.describeTable(tableName);
+      if (columns.rateLimitPerMinute) {
+        const legacyRateLimit = await this.readLegacyRateLimit(configCollection.quotedTableName());
+        if (legacyRateLimit && legacyRateLimit > 0) {
+          rateLimitPerMinute = legacyRateLimit;
+        }
+      }
+      const config = await this.db.getRepository("aiApiConfig").findOne();
+      if (config) {
+        quotaEnabled = !!config.get("quotaEnabled");
+      }
+    }
+    const existingDefault = await this.db.getRepository("aiApiUsageGroups").findOne({
+      filter: { isDefault: true }
+    });
+    if (!existingDefault) {
+      await this.db.getRepository("aiApiUsageGroups").create({
+        values: {
+          name: "Default",
+          isDefault: true,
+          quotaMode: "per_user",
+          rateLimitPerMinute,
+          enabled: quotaEnabled,
+          periodType: "monthly",
+          timezone: "UTC",
+          requestLimit: null,
+          totalTokenLimit: null,
+          costLimit: null,
+          currency: "USD",
+          rejectUnpricedModel: true,
+          missingUsageBehavior: "use_reserved",
+          contextOverflowBehavior: "reject"
+        }
+      });
+    }
+    if (configCollection && await configCollection.existsInDb()) {
+      const tableName = configCollection.getTableNameWithSchema();
+      const columns = await this.queryInterface.describeTable(tableName);
+      if (columns.rateLimitPerMinute) {
+        await this.queryInterface.removeColumn(tableName, "rateLimitPerMinute");
+      }
+    }
+    await this.migrateLegacyQuotaPolicies();
+  }
+  async down() {
+    const configCollection = this.db.getCollection("aiApiConfig");
+    if (configCollection && await configCollection.existsInDb()) {
+      const tableName = configCollection.getTableNameWithSchema();
+      const columns = await this.queryInterface.describeTable(tableName);
+      if (!columns.rateLimitPerMinute) {
+        await this.queryInterface.addColumn(tableName, "rateLimitPerMinute", {
+          type: "INTEGER",
+          allowNull: false,
+          defaultValue: 60
+        });
+      }
+    }
+  }
+  getGroupCollection() {
+    return this.db.getCollection("aiApiUsageGroups");
+  }
+  async readLegacyRateLimit(quotedTable) {
+    const qi = this.db.sequelize.getQueryInterface();
+    const [rows] = await this.db.sequelize.query(
+      `SELECT ${qi.quoteIdentifier("rateLimitPerMinute")} FROM ${quotedTable} LIMIT 1`
+    );
+    const first = rows == null ? void 0 : rows[0];
+    const value = Number(first == null ? void 0 : first.rateLimitPerMinute);
+    return Number.isFinite(value) ? value : void 0;
+  }
+  async migrateLegacyQuotaPolicies() {
+    const defaultGroup = await this.db.getRepository("aiApiUsageGroups").findOne({
+      filter: { isDefault: true }
+    });
+    if (!defaultGroup) return;
+    const policyCollection = this.db.getCollection("aiApiUserQuotaPolicies");
+    const memberCollection = this.db.getCollection("aiApiGroupMembers");
+    const bucketCollection = this.db.getCollection("aiApiGroupQuotaBuckets");
+    if (!policyCollection || !memberCollection || !bucketCollection) return;
+    const policies = await this.db.getRepository("aiApiUserQuotaPolicies").find({
+      pageSize: 1e3
+    });
+    for (const policy of policies) {
+      const userId = this.valueOf(policy, "userId");
+      if (userId === void 0 || userId === null) continue;
+      const existingMember = await this.db.getRepository("aiApiGroupMembers").findOne({
+        filter: { userId }
+      });
+      if (!existingMember) {
+        await this.db.getRepository("aiApiGroupMembers").create({
+          values: { groupId: this.valueOf(defaultGroup, "id"), userId }
+        });
+      }
+    }
+  }
+  valueOf(model, name) {
+    if (!model) return void 0;
+    if (typeof model.get === "function") {
+      return model.get(name);
+    }
+    return model[name];
+  }
+}

package/dist/server/migrations/20260816000000-migrate-user-permissions-to-groups.js

@@ -0,0 +1,169 @@
+/**
+ * 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 migrate_user_permissions_to_groups_exports = {};
+__export(migrate_user_permissions_to_groups_exports, {
+  default: () => MigrateUserPermissionsToGroups
+});
+module.exports = __toCommonJS(migrate_user_permissions_to_groups_exports);
+var import_server = require("@nocobase/server");
+class MigrateUserPermissionsToGroups extends import_server.Migration {
+  // Must run afterSync: the new allowedLlmServices/allowAllModels/allowedModels
+  // columns on aiApiUsageGroups only exist once collections have been synced.
+  on = "afterSync";
+  async up() {
+    const collection = this.db.getCollection("aiApiUserPermissions");
+    if (!collection || !await collection.existsInDb()) return;
+    const repository = this.db.getRepository("aiApiUserPermissions");
+    const rows = [];
+    for (let page = 1; ; page += 1) {
+      const batch = await repository.find({ page, pageSize: 1e3, sort: "id" });
+      rows.push(...batch);
+      if (batch.length < 1e3) break;
+    }
+    if (!rows.length) return;
+    const migratable = [];
+    const deniedUserIds = [];
+    for (const row of rows) {
+      const userId = this.valueOf(row, "userId");
+      if (userId === void 0 || userId === null) continue;
+      const enabled = this.valueOf(row, "enabled") !== false;
+      const allowedLlmServices = this.stringArray(this.valueOf(row, "allowedLlmServices"));
+      const allowAllModels = this.valueOf(row, "allowAllModels") !== false;
+      const allowedModels = this.stringArray(this.valueOf(row, "allowedModels"));
+      const hasEffectiveGrant = allowedLlmServices.length > 0 && (allowAllModels || allowedModels.length > 0);
+      if (!enabled || !hasEffectiveGrant) {
+        deniedUserIds.push(userId);
+        continue;
+      }
+      migratable.push({ userId, sig: { allowedLlmServices, allowAllModels, allowedModels } });
+    }
+    if (deniedUserIds.length) {
+      this.app.logger.warn(
+        `[ai-api] ${deniedUserIds.length} aiApiUserPermissions row(s) denied all access (enabled=false or empty grant) and cannot be migrated to usage groups. These users now inherit the default group's access. Affected userIds: ${deniedUserIds.join(", ")}. Use role permissions to block API access entirely.`
+      );
+    }
+    if (!migratable.length) return;
+    const defaultGroup = await this.db.getRepository("aiApiUsageGroups").findOne({
+      filter: { isDefault: true }
+    });
+    const quotaValues = this.quotaValuesFrom(defaultGroup);
+    const bySignature = /* @__PURE__ */ new Map();
+    for (const { userId, sig } of migratable) {
+      const key = JSON.stringify({
+        allowedLlmServices: [...sig.allowedLlmServices].sort(),
+        allowAllModels: sig.allowAllModels,
+        allowedModels: [...sig.allowedModels].sort()
+      });
+      const entry = bySignature.get(key);
+      if (entry) {
+        entry.userIds.push(userId);
+      } else {
+        bySignature.set(key, { sig, userIds: [userId] });
+      }
+    }
+    let index = 0;
+    for (const { sig, userIds } of bySignature.values()) {
+      index += 1;
+      const group = await this.db.getRepository("aiApiUsageGroups").create({
+        values: {
+          ...quotaValues,
+          name: `Migrated permissions ${index}`,
+          isDefault: false,
+          quotaMode: "per_user",
+          allowedLlmServices: sig.allowedLlmServices,
+          allowAllModels: sig.allowAllModels,
+          allowedModels: sig.allowedModels
+        }
+      });
+      const groupId = this.valueOf(group, "id");
+      for (const userId of userIds) {
+        const existing = await this.db.getRepository("aiApiGroupMembers").findOne({ filter: { userId } });
+        if (existing) {
+          await this.db.getRepository("aiApiGroupMembers").update({
+            filterByTk: this.valueOf(existing, "id"),
+            values: { groupId }
+          });
+        } else {
+          await this.db.getRepository("aiApiGroupMembers").create({ values: { groupId, userId } });
+        }
+      }
+    }
+    this.app.logger.info(
+      `[ai-api] Migrated ${migratable.length} aiApiUserPermissions row(s) into ${bySignature.size} usage group(s).`
+    );
+  }
+  async down() {
+    const groups = await this.db.getRepository("aiApiUsageGroups").find({
+      filter: { "name.$startsWith": "Migrated permissions" }
+    });
+    for (const group of groups) {
+      const groupId = this.valueOf(group, "id");
+      await this.db.getRepository("aiApiGroupMembers").destroy({ filter: { groupId } });
+      await this.db.getRepository("aiApiUsageGroups").destroy({ filterByTk: groupId });
+    }
+  }
+  quotaValuesFrom(defaultGroup) {
+    if (!defaultGroup) {
+      return {
+        rateLimitPerMinute: 60,
+        enabled: false,
+        periodType: "monthly",
+        timezone: "UTC",
+        requestLimit: null,
+        totalTokenLimit: null,
+        costLimit: null,
+        currency: "USD",
+        rejectUnpricedModel: true,
+        missingUsageBehavior: "use_reserved",
+        contextOverflowBehavior: "reject"
+      };
+    }
+    return {
+      rateLimitPerMinute: Number(this.valueOf(defaultGroup, "rateLimitPerMinute") ?? 60),
+      enabled: !!this.valueOf(defaultGroup, "enabled"),
+      periodType: this.valueOf(defaultGroup, "periodType") ?? "monthly",
+      timezone: this.valueOf(defaultGroup, "timezone") ?? "UTC",
+      requestLimit: this.valueOf(defaultGroup, "requestLimit") ?? null,
+      totalTokenLimit: this.valueOf(defaultGroup, "totalTokenLimit") ?? null,
+      costLimit: this.valueOf(defaultGroup, "costLimit") ?? null,
+      currency: this.valueOf(defaultGroup, "currency") ?? "USD",
+      rejectUnpricedModel: !!this.valueOf(defaultGroup, "rejectUnpricedModel"),
+      missingUsageBehavior: this.valueOf(defaultGroup, "missingUsageBehavior") ?? "use_reserved",
+      contextOverflowBehavior: this.valueOf(defaultGroup, "contextOverflowBehavior") ?? "reject"
+    };
+  }
+  stringArray(value) {
+    return Array.isArray(value) ? value.filter((item) => typeof item === "string" && item.length > 0) : [];
+  }
+  valueOf(model, name) {
+    if (!model) return void 0;
+    if (typeof model.get === "function") {
+      return model.get(name);
+    }
+    return model[name];
+  }
+}

package/dist/server/migrations/20260816100000-add-model-metadata-system-prompt.js

@@ -0,0 +1,69 @@
+/**
+ * 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 add_model_metadata_system_prompt_exports = {};
+__export(add_model_metadata_system_prompt_exports, {
+  default: () => AddSystemPromptToModelMetadata
+});
+module.exports = __toCommonJS(add_model_metadata_system_prompt_exports);
+var import_server = require("@nocobase/server");
+class AddSystemPromptToModelMetadata extends import_server.Migration {
+  on = "beforeLoad";
+  async up() {
+    const collection = this.db.getCollection("aiApiModelMetadata");
+    if (!collection) return;
+    const field = collection.getField("systemPrompt");
+    if (!field) {
+      collection.addField("systemPrompt", { type: "text", allowNull: true });
+    }
+    if (await collection.existsInDb()) {
+      const tableName = collection.getTableNameWithSchema();
+      const exists = await this.tableColumnExists(tableName, "systemPrompt");
+      if (!exists) {
+        await this.queryInterface.addColumn(tableName, "systemPrompt", {
+          type: "TEXT",
+          allowNull: true
+        });
+      }
+    }
+  }
+  async down() {
+    const collection = this.db.getCollection("aiApiModelMetadata");
+    if (!collection) return;
+    if (await collection.existsInDb()) {
+      const tableName = collection.getTableNameWithSchema();
+      const exists = await this.tableColumnExists(tableName, "systemPrompt");
+      if (exists) {
+        await this.queryInterface.removeColumn(tableName, "systemPrompt");
+      }
+    }
+    collection.removeField("systemPrompt");
+  }
+  async tableColumnExists(tableName, columnName) {
+    const columns = await this.queryInterface.describeTable(tableName);
+    return Object.prototype.hasOwnProperty.call(columns, columnName);
+  }
+}