plugin-ai-api 1.1.1 → 1.1.2

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

@@ -1,168 +0,0 @@
-/**
- * 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);
-const USER_SUMMARY_FIELDS = ["id", "username", "nickname", "email"];
-async function getByUser(ctx, next) {
-  const { userId } = ctx.action.params;
-  if (!userId) {
-    ctx.throw(400, "userId is required");
-  }
-  const member = await ctx.db.getRepository("aiApiGroupMembers").findOne({
-    filter: { userId },
-    appends: ["group"]
-  });
-  if (member && member.get("group")) {
-    ctx.body = member.get("group");
-    return next();
-  }
-  const defaultGroup = await ctx.db.getRepository("aiApiUsageGroups").findOne({
-    filter: { isDefault: true }
-  });
-  if (!defaultGroup) {
-    ctx.throw(404, "Default group not found");
-  }
-  ctx.body = defaultGroup;
-  return next();
-}
-async function listUnassignedUsers(ctx, next) {
-  const { page = 1, pageSize = 20, keyword } = ctx.action.params;
-  const memberRows = await ctx.db.getRepository("aiApiGroupMembers").find({
-    fields: ["userId"],
-    pageSize: 1e4
-  });
-  const assignedUserIds = memberRows.map((row) => {
-    var _a;
-    const value = ((_a = row.get) == null ? void 0 : _a.call(row, "userId")) ?? row.userId;
-    return String(value);
-  });
-  const where = {};
-  if (keyword) {
-    where.$or = [
-      { username: { $includes: keyword } },
-      { email: { $includes: keyword } },
-      { nickname: { $includes: keyword } }
-    ];
-  }
-  if (assignedUserIds.length > 0) {
-    where.id = { $notIn: assignedUserIds };
-  }
-  const repository = ctx.db.getRepository("users");
-  const [rows, count] = await repository.findAndCount({
-    filter: where,
-    fields: USER_SUMMARY_FIELDS,
-    page: Number(page),
-    pageSize: Number(pageSize)
-  });
-  ctx.body = { rows, count };
-  return next();
-}
-async function searchUsers(ctx, next) {
-  const { keyword, pageSize = 20 } = ctx.action.params;
-  const where = {};
-  if (keyword) {
-    where.$or = [
-      { username: { $includes: keyword } },
-      { email: { $includes: keyword } },
-      { nickname: { $includes: keyword } }
-    ];
-  }
-  const [rows, count] = await ctx.db.getRepository("users").findAndCount({
-    filter: where,
-    fields: USER_SUMMARY_FIELDS,
-    page: 1,
-    pageSize: Number(pageSize)
-  });
-  ctx.body = { rows, count };
-  return next();
-}
-async function addMember(ctx, next) {
-  const { groupId, userId } = ctx.action.params.values || ctx.action.params || {};
-  if (!groupId || !userId) {
-    ctx.throw(400, "groupId and userId are required");
-  }
-  const group = await ctx.db.getRepository("aiApiUsageGroups").findOne({
-    filterByTk: groupId
-  });
-  if (!group) {
-    ctx.throw(404, "Usage group not found");
-  }
-  if (group.get("isDefault")) {
-    ctx.throw(400, "Cannot add members to the default group: users without another group use it automatically");
-  }
-  const existing = await ctx.db.getRepository("aiApiGroupMembers").findOne({
-    filter: { userId }
-  });
-  if (existing) {
-    ctx.throw(409, "User already belongs to a group");
-  }
-  const member = await ctx.db.getRepository("aiApiGroupMembers").create({
-    values: { groupId, userId }
-  });
-  ctx.body = member;
-  return next();
-}
-async function removeMember(ctx, next) {
-  const { groupId, userId } = ctx.action.params.values || ctx.action.params || {};
-  if (!groupId || !userId) {
-    ctx.throw(400, "groupId and userId are required");
-  }
-  const member = await ctx.db.getRepository("aiApiGroupMembers").findOne({
-    filter: { groupId, userId }
-  });
-  if (!member) {
-    ctx.throw(404, "Member not found");
-  }
-  await ctx.db.getRepository("aiApiGroupMembers").destroy({
-    filterByTk: member.get("id")
-  });
-  ctx.body = member;
-  return next();
-}
-var ai_api_usage_groups_default = {
-  name: "aiApiUsageGroups",
-  actions: {
-    getByUser: {
-      handler: getByUser
-    },
-    listUnassignedUsers: {
-      handler: listUnassignedUsers
-    },
-    searchUsers: {
-      handler: searchUsers
-    },
-    addMember: {
-      handler: addMember
-    },
-    removeMember: {
-      handler: removeMember
-    }
-  }
-};

package/src/server/collections/ai-api-user-permissions.ts

@@ -1,46 +0,0 @@
-/**
- * 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.
- */
-
-import { defineCollection } from '@nocobase/database';
-
-export default defineCollection({
-  name: 'aiApiUserPermissions',
-  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: 'allowedLlmServices',
-      type: 'json',
-      defaultValue: [],
-      comment: 'LLM service names/titles this user may use. Empty means the user is denied every service.',
-    },
-    { name: 'allowAllModels', type: 'boolean', defaultValue: true },
-    {
-      name: 'allowedModels',
-      type: 'json',
-      defaultValue: [],
-      comment: 'Array of "serviceName/modelId" this user may use (when allowAllModels=false)',
-    },
-  ],
-  indexes: [
-    {
-      fields: ['userId'],
-      unique: true,
-    },
-  ],
-});

package/src/server/collections/ai-api-user-quota-buckets.ts

@@ -1,24 +0,0 @@
-import { defineCollection } from '@nocobase/database';
-
-export default 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/src/server/collections/ai-api-user-quota-policies.ts

@@ -1,33 +0,0 @@
-import { defineCollection } from '@nocobase/database';
-
-export default 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' },
-    { name: 'contextOverflowBehavior', type: 'string', allowNull: false, defaultValue: 'reject' },
-  ],
-  indexes: [
-    {
-      fields: ['userId'],
-      unique: true,
-    },
-  ],
-});

package/src/server/resource/ai-api-usage-groups.ts

@@ -1,171 +0,0 @@
-/**
- * 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.
- */
-
-import { Context } from '@nocobase/actions';
-
-const USER_SUMMARY_FIELDS = ['id', 'username', 'nickname', 'email'];
-
-async function getByUser(ctx: Context, next: () => Promise<void>) {
-  const { userId } = ctx.action.params;
-  if (!userId) {
-    ctx.throw(400, 'userId is required');
-  }
-  const member = await ctx.db.getRepository('aiApiGroupMembers').findOne({
-    filter: { userId },
-    appends: ['group'],
-  });
-
-  if (member && member.get('group')) {
-    ctx.body = member.get('group');
-    return next();
-  }
-
-  const defaultGroup = await ctx.db.getRepository('aiApiUsageGroups').findOne({
-    filter: { isDefault: true },
-  });
-
-  if (!defaultGroup) {
-    ctx.throw(404, 'Default group not found');
-  }
-
-  ctx.body = defaultGroup;
-  return next();
-}
-
-async function listUnassignedUsers(ctx: Context, next: () => Promise<void>) {
-  const { page = 1, pageSize = 20, keyword } = ctx.action.params;
-
-  const memberRows = (await ctx.db.getRepository('aiApiGroupMembers').find({
-    fields: ['userId'],
-    pageSize: 10000,
-  })) as unknown[];
-  const assignedUserIds = memberRows.map((row) => {
-    const value = (row as Record<string, unknown>).get?.('userId') ?? (row as Record<string, unknown>).userId;
-    return String(value);
-  });
-
-  const where: Record<string, unknown> = {};
-  if (keyword) {
-    where.$or = [
-      { username: { $includes: keyword } },
-      { email: { $includes: keyword } },
-      { nickname: { $includes: keyword } },
-    ];
-  }
-  if (assignedUserIds.length > 0) {
-    where.id = { $notIn: assignedUserIds };
-  }
-
-  const repository = ctx.db.getRepository('users');
-  const [rows, count] = await repository.findAndCount({
-    filter: where,
-    fields: USER_SUMMARY_FIELDS,
-    page: Number(page),
-    pageSize: Number(pageSize),
-  });
-
-  ctx.body = { rows, count };
-  return next();
-}
-
-async function searchUsers(ctx: Context, next: () => Promise<void>) {
-  const { keyword, pageSize = 20 } = ctx.action.params;
-
-  const where: Record<string, unknown> = {};
-  if (keyword) {
-    where.$or = [
-      { username: { $includes: keyword } },
-      { email: { $includes: keyword } },
-      { nickname: { $includes: keyword } },
-    ];
-  }
-
-  const [rows, count] = await ctx.db.getRepository('users').findAndCount({
-    filter: where,
-    fields: USER_SUMMARY_FIELDS,
-    page: 1,
-    pageSize: Number(pageSize),
-  });
-
-  ctx.body = { rows, count };
-  return next();
-}
-
-async function addMember(ctx: Context, next: () => Promise<void>) {
-  const { groupId, userId } = ctx.action.params.values || ctx.action.params || {};
-  if (!groupId || !userId) {
-    ctx.throw(400, 'groupId and userId are required');
-  }
-
-  const group = await ctx.db.getRepository('aiApiUsageGroups').findOne({
-    filterByTk: groupId,
-  });
-  if (!group) {
-    ctx.throw(404, 'Usage group not found');
-  }
-  // The default group is a fallback, not a membership target: an explicit row would
-  // pin the user there and hide them from listUnassignedUsers forever.
-  if (group.get('isDefault')) {
-    ctx.throw(400, 'Cannot add members to the default group: users without another group use it automatically');
-  }
-
-  const existing = await ctx.db.getRepository('aiApiGroupMembers').findOne({
-    filter: { userId },
-  });
-  if (existing) {
-    ctx.throw(409, 'User already belongs to a group');
-  }
-
-  const member = await ctx.db.getRepository('aiApiGroupMembers').create({
-    values: { groupId, userId },
-  });
-  ctx.body = member;
-  return next();
-}
-
-async function removeMember(ctx: Context, next: () => Promise<void>) {
-  const { groupId, userId } = ctx.action.params.values || ctx.action.params || {};
-  if (!groupId || !userId) {
-    ctx.throw(400, 'groupId and userId are required');
-  }
-
-  const member = await ctx.db.getRepository('aiApiGroupMembers').findOne({
-    filter: { groupId, userId },
-  });
-  if (!member) {
-    ctx.throw(404, 'Member not found');
-  }
-
-  await ctx.db.getRepository('aiApiGroupMembers').destroy({
-    filterByTk: member.get('id'),
-  });
-  ctx.body = member;
-  return next();
-}
-
-export default {
-  name: 'aiApiUsageGroups',
-  actions: {
-    getByUser: {
-      handler: getByUser,
-    },
-    listUnassignedUsers: {
-      handler: listUnassignedUsers,
-    },
-    searchUsers: {
-      handler: searchUsers,
-    },
-    addMember: {
-      handler: addMember,
-    },
-    removeMember: {
-      handler: removeMember,
-    },
-  },
-};