plugin-ai-api 1.0.15 → 1.0.20

package/src/client-v2/pages/UserQuotasPage.tsx

@@ -0,0 +1,258 @@
+import React, { useCallback, useEffect, useState } from 'react';
+import {
+  Button,
+  Card,
+  Form,
+  Input,
+  InputNumber,
+  Modal,
+  Popconfirm,
+  Select,
+  Space,
+  Switch,
+  Table,
+  Tag,
+  message,
+} from 'antd';
+import type { ColumnsType } from 'antd/es/table';
+import { useFlowContext } from '@nocobase/flow-engine';
+import { useT } from '../locale';
+import { errorMessage, unwrapData } from './api';
+
+interface UserSummary {
+  id: string | number;
+  nickname?: string;
+  username?: string;
+  email?: string;
+}
+
+interface QuotaPolicy {
+  id: string | number;
+  userId: string | number;
+  user?: UserSummary;
+  enabled: boolean;
+  periodType: 'daily' | 'monthly';
+  timezone: string;
+  requestLimit?: string;
+  totalTokenLimit?: string;
+  costLimit?: string;
+  currency: string;
+  rejectUnpricedModel: boolean;
+  missingUsageBehavior: 'allow' | 'use_reserved';
+}
+
+export default function UserQuotasPage() {
+  const ctx = useFlowContext();
+  const t = useT();
+  const [form] = Form.useForm<QuotaPolicy>();
+  const [rows, setRows] = useState<QuotaPolicy[]>([]);
+  const [users, setUsers] = useState<UserSummary[]>([]);
+  const [loading, setLoading] = useState(false);
+  const [saving, setSaving] = useState(false);
+  const [editing, setEditing] = useState<QuotaPolicy>();
+  const [open, setOpen] = useState(false);
+
+  const load = useCallback(async () => {
+    setLoading(true);
+    try {
+      const [policiesResponse, usersResponse] = await Promise.all([
+        ctx.api.request({
+          url: 'aiApiUserQuotaPolicies:list',
+          method: 'get',
+          params: { pageSize: 200, appends: ['user'], sort: '-updatedAt' },
+        }),
+        ctx.api.request({ url: 'users:list', method: 'get', params: { pageSize: 200 } }),
+      ]);
+      setRows(unwrapData<QuotaPolicy[]>(policiesResponse, []));
+      setUsers(unwrapData<UserSummary[]>(usersResponse, []));
+    } catch (error) {
+      message.error(errorMessage(error));
+    } finally {
+      setLoading(false);
+    }
+  }, [ctx.api]);
+
+  useEffect(() => {
+    load();
+  }, [load]);
+
+  const showCreate = () => {
+    setEditing(undefined);
+    form.setFieldsValue({
+      enabled: true,
+      periodType: 'monthly',
+      timezone: 'UTC',
+      currency: 'USD',
+      rejectUnpricedModel: true,
+      missingUsageBehavior: 'use_reserved',
+    } as QuotaPolicy);
+    setOpen(true);
+  };
+
+  const showEdit = (record: QuotaPolicy) => {
+    setEditing(record);
+    form.setFieldsValue(record);
+    setOpen(true);
+  };
+
+  const save = async () => {
+    const values = await form.validateFields();
+    setSaving(true);
+    try {
+      await ctx.api.request({
+        url: editing ? `aiApiUserQuotaPolicies:update/${editing.id}` : 'aiApiUserQuotaPolicies:create',
+        method: 'post',
+        data: values,
+      });
+      message.success(t('Saved successfully'));
+      setOpen(false);
+      await load();
+    } catch (error) {
+      message.error(errorMessage(error));
+    } finally {
+      setSaving(false);
+    }
+  };
+
+  const remove = async (record: QuotaPolicy) => {
+    try {
+      await ctx.api.request({
+        url: `aiApiUserQuotaPolicies:destroy/${record.id}`,
+        method: 'post',
+      });
+      message.success(t('Deleted successfully'));
+      await load();
+    } catch (error) {
+      message.error(errorMessage(error));
+    }
+  };
+
+  const userLabel = (user?: UserSummary) => user?.nickname || user?.username || user?.email || String(user?.id ?? '');
+  const columns: ColumnsType<QuotaPolicy> = [
+    {
+      title: t('User'),
+      key: 'user',
+      width: 180,
+      render: (_, record) => userLabel(record.user) || String(record.userId),
+    },
+    { title: t('Period'), dataIndex: 'periodType', key: 'periodType', width: 100 },
+    {
+      title: t('Request limit'),
+      dataIndex: 'requestLimit',
+      key: 'requestLimit',
+      width: 130,
+      render: (value) => value ?? t('Unlimited'),
+    },
+    {
+      title: t('Token limit'),
+      dataIndex: 'totalTokenLimit',
+      key: 'totalTokenLimit',
+      width: 130,
+      render: (value) => value ?? t('Unlimited'),
+    },
+    {
+      title: t('Cost limit'),
+      dataIndex: 'costLimit',
+      key: 'costLimit',
+      width: 120,
+      render: (value, record) => (value == null ? t('Unlimited') : `${value} ${record.currency}`),
+    },
+    { title: t('Timezone'), dataIndex: 'timezone', key: 'timezone', width: 140 },
+    {
+      title: t('Status'),
+      dataIndex: 'enabled',
+      key: 'enabled',
+      width: 100,
+      render: (enabled: boolean) => (
+        <Tag color={enabled ? 'green' : 'default'}>{enabled ? t('Enabled') : t('Disabled')}</Tag>
+      ),
+    },
+    {
+      title: t('Actions'),
+      key: 'actions',
+      width: 150,
+      fixed: 'right',
+      render: (_, record) => (
+        <Space size={0}>
+          <Button type="link" onClick={() => showEdit(record)}>
+            {t('Edit')}
+          </Button>
+          <Popconfirm title={t('Delete this quota?')} onConfirm={() => remove(record)}>
+            <Button type="link" danger>
+              {t('Delete')}
+            </Button>
+          </Popconfirm>
+        </Space>
+      ),
+    },
+  ];
+
+  return (
+    <Card
+      title={t('User quotas')}
+      extra={
+        <Button type="primary" onClick={showCreate}>
+          {t('Add quota')}
+        </Button>
+      }
+    >
+      <Table rowKey="id" columns={columns} dataSource={rows} loading={loading} scroll={{ x: 1100 }} />
+      <Modal
+        title={editing ? t('Edit quota') : t('Add quota')}
+        open={open}
+        onCancel={() => setOpen(false)}
+        onOk={save}
+        confirmLoading={saving}
+        destroyOnClose
+      >
+        <Form form={form} layout="vertical" preserve={false}>
+          <Form.Item name="userId" label={t('User')} rules={[{ required: true }]}>
+            <Select
+              disabled={Boolean(editing)}
+              showSearch
+              optionFilterProp="label"
+              options={users.map((user) => ({ label: userLabel(user), value: user.id }))}
+            />
+          </Form.Item>
+          <Form.Item name="periodType" label={t('Period')} rules={[{ required: true }]}>
+            <Select
+              options={[
+                { label: t('Daily'), value: 'daily' },
+                { label: t('Monthly'), value: 'monthly' },
+              ]}
+            />
+          </Form.Item>
+          <Form.Item name="timezone" label={t('Timezone')} rules={[{ required: true }]}>
+            <Input placeholder="UTC" />
+          </Form.Item>
+          <Form.Item name="requestLimit" label={t('Request limit')}>
+            <InputNumber min={0} stringMode style={{ width: '100%' }} />
+          </Form.Item>
+          <Form.Item name="totalTokenLimit" label={t('Token limit')}>
+            <InputNumber min={0} stringMode style={{ width: '100%' }} />
+          </Form.Item>
+          <Form.Item name="costLimit" label={t('Cost limit')}>
+            <InputNumber min={0} stringMode style={{ width: '100%' }} />
+          </Form.Item>
+          <Form.Item name="currency" label={t('Currency')} rules={[{ required: true }]}>
+            <Input />
+          </Form.Item>
+          <Form.Item name="rejectUnpricedModel" label={t('Reject unpriced models')} valuePropName="checked">
+            <Switch />
+          </Form.Item>
+          <Form.Item name="missingUsageBehavior" label={t('Missing usage behavior')} rules={[{ required: true }]}>
+            <Select
+              options={[
+                { label: t('Use reserved estimate'), value: 'use_reserved' },
+                { label: t('Allow without token charge'), value: 'allow' },
+              ]}
+            />
+          </Form.Item>
+          <Form.Item name="enabled" label={t('Enabled')} valuePropName="checked">
+            <Switch />
+          </Form.Item>
+        </Form>
+      </Modal>
+    </Card>
+  );
+}

package/src/client-v2/pages/api.ts

@@ -0,0 +1,16 @@
+export interface ApiEnvelope<T> {
+  data?: {
+    data?: T;
+    meta?: { count?: number };
+  };
+}
+
+export function unwrapData<T>(response: unknown, fallback: T): T {
+  if (!response || typeof response !== 'object') return fallback;
+  const outer = response as ApiEnvelope<T>;
+  return outer.data?.data ?? fallback;
+}
+
+export function errorMessage(error: unknown): string {
+  return error instanceof Error ? error.message : String(error);
+}

package/src/client-v2/plugin.tsx

@@ -1,5 +1,4 @@
 import { Plugin, Application } from '@nocobase/client-v2';
-import React from 'react';
 
 export class PluginAiApiClient extends Plugin<Record<string, never>, Application> {
   async load() {
@@ -14,10 +13,29 @@ export class PluginAiApiClient extends Plugin<Record<string, never>, Application
       menuKey: 'ai-api',
       key: 'index',
       title: this.t('Configuration'),
-      
-      componentLoader: () => import('../client/AiApiConfigPage'),
+      componentLoader: () => import('./pages/GeneralPage'),
     });
 
+    this.pluginSettingsManager.addPageTabItem({
+      menuKey: 'ai-api',
+      key: 'model-pricing',
+      title: this.t('Model pricing'),
+      componentLoader: () => import('./pages/ModelPricingPage'),
+    });
+
+    this.pluginSettingsManager.addPageTabItem({
+      menuKey: 'ai-api',
+      key: 'user-quotas',
+      title: this.t('User quotas'),
+      componentLoader: () => import('./pages/UserQuotasPage'),
+    });
+
+    this.pluginSettingsManager.addPageTabItem({
+      menuKey: 'ai-api',
+      key: 'usage',
+      title: this.t('Usage'),
+      componentLoader: () => import('./pages/UsagePage'),
+    });
   }
 }
 

package/src/locale/en-US.json

@@ -6,5 +6,81 @@
   "Rate Limit": "Rate Limit",
   "Save Configuration": "Save Configuration",
   "Configuration saved": "Configuration saved",
-  "Failed to save configuration": "Failed to save configuration"
-}
+  "Failed to save configuration": "Failed to save configuration",
+  "API mode": "API mode",
+  "Direct LLM": "Direct LLM",
+  "AI Employee agent": "AI Employee agent",
+  "Default LLM service": "Default LLM service",
+  "Enable user quotas": "Enable user quotas",
+  "Default reserved output tokens": "Default reserved output tokens",
+  "Refresh": "Refresh",
+  "Model pricing": "Model pricing",
+  "User quotas": "User quotas",
+  "Usage": "Usage",
+  "LLM service": "LLM service",
+  "Model": "Model",
+  "Input price / 1M": "Input price / 1M",
+  "Output price / 1M": "Output price / 1M",
+  "Fixed request cost": "Fixed request cost",
+  "Currency": "Currency",
+  "Status": "Status",
+  "Enabled": "Enabled",
+  "Disabled": "Disabled",
+  "Actions": "Actions",
+  "Edit": "Edit",
+  "Delete": "Delete",
+  "Delete this price?": "Delete this price?",
+  "Add price": "Add price",
+  "Edit price": "Edit price",
+  "Effective from": "Effective from",
+  "Effective to": "Effective to",
+  "Notes": "Notes",
+  "Saved successfully": "Saved successfully",
+  "Deleted successfully": "Deleted successfully",
+  "User": "User",
+  "Period": "Period",
+  "Request limit": "Request limit",
+  "Token limit": "Token limit",
+  "Cost limit": "Cost limit",
+  "Timezone": "Timezone",
+  "Unlimited": "Unlimited",
+  "Add quota": "Add quota",
+  "Edit quota": "Edit quota",
+  "Delete this quota?": "Delete this quota?",
+  "Daily": "Daily",
+  "Monthly": "Monthly",
+  "Reject unpriced models": "Reject unpriced models",
+  "Missing usage behavior": "Missing usage behavior",
+  "Use reserved estimate": "Use reserved estimate",
+  "Allow without token charge": "Allow without token charge",
+  "Started at": "Started at",
+  "Requested model": "Requested model",
+  "Resolved service": "Resolved service",
+  "Resolved model": "Resolved model",
+  "Input tokens": "Input tokens",
+  "Output tokens": "Output tokens",
+  "Total tokens": "Total tokens",
+  "Cost": "Cost",
+  "Cost status": "Cost status",
+  "Request ID": "Request ID",
+  "Failed to load models": "Failed to load models",
+  "Select a model": "Select a model",
+  "Select an AI Employee": "Select an AI Employee",
+  "Usage guide": "Usage guide",
+  "OpenAI-compatible endpoint": "OpenAI-compatible endpoint",
+  "Base URL": "Base URL",
+  "Use a NocoBase API key as the Bearer token.": "Use a NocoBase API key as the Bearer token.",
+  "List available models": "List available models",
+  "Send a chat completion": "Send a chat completion",
+  "Usage filters": "Usage filters",
+  "Time range": "Time range",
+  "User ID": "User ID",
+  "Succeeded": "Succeeded",
+  "Failed": "Failed",
+  "Started": "Started",
+  "Apply filters": "Apply filters",
+  "Reset": "Reset",
+  "Requests": "Requests",
+  "Total cost": "Total cost",
+  "Usage records": "Usage records"
+}

package/src/locale/vi-VN.json

@@ -0,0 +1,86 @@
+{
+  "AI API Gateway": "Cổng AI API",
+  "Configuration": "Cấu hình",
+  "Default AI Employee": "AI Employee mặc định",
+  "Enabled LLM Services": "Dịch vụ LLM được bật",
+  "Rate Limit": "Giới hạn tốc độ",
+  "Save Configuration": "Lưu cấu hình",
+  "Configuration saved": "Đã lưu cấu hình",
+  "Failed to save configuration": "Không thể lưu cấu hình",
+  "API mode": "Chế độ API",
+  "Direct LLM": "LLM trực tiếp",
+  "AI Employee agent": "AI Employee agent",
+  "Default LLM service": "Dịch vụ LLM mặc định",
+  "Enable user quotas": "Bật quota theo người dùng",
+  "Default reserved output tokens": "Số output token giữ chỗ mặc định",
+  "Refresh": "Làm mới",
+  "Model pricing": "Bảng giá model",
+  "User quotas": "Quota người dùng",
+  "Usage": "Lịch sử sử dụng",
+  "LLM service": "Dịch vụ LLM",
+  "Model": "Model",
+  "Input price / 1M": "Giá input / 1 triệu token",
+  "Output price / 1M": "Giá output / 1 triệu token",
+  "Fixed request cost": "Chi phí cố định mỗi request",
+  "Currency": "Tiền tệ",
+  "Status": "Trạng thái",
+  "Enabled": "Đang bật",
+  "Disabled": "Đã tắt",
+  "Actions": "Thao tác",
+  "Edit": "Sửa",
+  "Delete": "Xóa",
+  "Delete this price?": "Xóa cấu hình giá này?",
+  "Add price": "Thêm giá",
+  "Edit price": "Sửa giá",
+  "Effective from": "Hiệu lực từ",
+  "Effective to": "Hiệu lực đến",
+  "Notes": "Ghi chú",
+  "Saved successfully": "Đã lưu thành công",
+  "Deleted successfully": "Đã xóa thành công",
+  "User": "Người dùng",
+  "Period": "Chu kỳ",
+  "Request limit": "Giới hạn request",
+  "Token limit": "Giới hạn token",
+  "Cost limit": "Giới hạn chi phí",
+  "Timezone": "Múi giờ",
+  "Unlimited": "Không giới hạn",
+  "Add quota": "Thêm quota",
+  "Edit quota": "Sửa quota",
+  "Delete this quota?": "Xóa quota này?",
+  "Daily": "Hàng ngày",
+  "Monthly": "Hàng tháng",
+  "Reject unpriced models": "Từ chối model chưa có giá",
+  "Missing usage behavior": "Xử lý khi thiếu token usage",
+  "Use reserved estimate": "Dùng số liệu giữ chỗ để ước tính",
+  "Allow without token charge": "Cho phép và không tính token",
+  "Started at": "Bắt đầu lúc",
+  "Requested model": "Model được yêu cầu",
+  "Resolved service": "Service đã resolve",
+  "Resolved model": "Model đã resolve",
+  "Input tokens": "Input token",
+  "Output tokens": "Output token",
+  "Total tokens": "Tổng token",
+  "Cost": "Chi phí",
+  "Cost status": "Trạng thái chi phí",
+  "Request ID": "Request ID",
+  "Failed to load models": "Không thể tải danh sách model",
+  "Select a model": "Chọn model",
+  "Select an AI Employee": "Chọn AI Employee",
+  "Usage guide": "Hướng dẫn sử dụng",
+  "OpenAI-compatible endpoint": "Endpoint tương thích OpenAI",
+  "Base URL": "Base URL",
+  "Use a NocoBase API key as the Bearer token.": "Sử dụng NocoBase API key làm Bearer token.",
+  "List available models": "Liệt kê model khả dụng",
+  "Send a chat completion": "Gửi yêu cầu chat completion",
+  "Usage filters": "Bộ lọc usage",
+  "Time range": "Khoảng thời gian",
+  "User ID": "User ID",
+  "Succeeded": "Thành công",
+  "Failed": "Thất bại",
+  "Started": "Đã bắt đầu",
+  "Apply filters": "Áp dụng bộ lọc",
+  "Reset": "Đặt lại",
+  "Requests": "Số request",
+  "Total cost": "Tổng chi phí",
+  "Usage records": "Chi tiết usage"
+}

package/src/locale/zh-CN.json

@@ -1,10 +1,86 @@
-{
-  "AI API Gateway": "AI API 网关",
-  "Configuration": "配置",
-  "Default AI Employee": "默认 AI 员工",
-  "Enabled LLM Services": "已启用 LLM 服务",
-  "Rate Limit": "速率限制",
-  "Save Configuration": "保存配置",
-  "Configuration saved": "配置已保存",
-  "Failed to save configuration": "保存配置失败"
-}
+{
+  "AI API Gateway": "AI API 网关",
+  "Configuration": "配置",
+  "Default AI Employee": "默认 AI 员工",
+  "Enabled LLM Services": "已启用 LLM 服务",
+  "Rate Limit": "速率限制",
+  "Save Configuration": "保存配置",
+  "Configuration saved": "配置已保存",
+  "Failed to save configuration": "保存配置失败",
+  "API mode": "API 模式",
+  "Direct LLM": "直接 LLM",
+  "AI Employee agent": "AI 员工代理",
+  "Default LLM service": "默认 LLM 服务",
+  "Enable user quotas": "启用用户配额",
+  "Default reserved output tokens": "默认预留输出令牌数",
+  "Refresh": "刷新",
+  "Model pricing": "模型定价",
+  "User quotas": "用户配额",
+  "Usage": "用量",
+  "LLM service": "LLM 服务",
+  "Model": "模型",
+  "Input price / 1M": "输入价格 / 百万令牌",
+  "Output price / 1M": "输出价格 / 百万令牌",
+  "Fixed request cost": "每次请求固定费用",
+  "Currency": "货币",
+  "Status": "状态",
+  "Enabled": "已启用",
+  "Disabled": "已禁用",
+  "Actions": "操作",
+  "Edit": "编辑",
+  "Delete": "删除",
+  "Delete this price?": "删除此价格？",
+  "Add price": "添加价格",
+  "Edit price": "编辑价格",
+  "Effective from": "生效时间",
+  "Effective to": "失效时间",
+  "Notes": "备注",
+  "Saved successfully": "保存成功",
+  "Deleted successfully": "删除成功",
+  "User": "用户",
+  "Period": "周期",
+  "Request limit": "请求限制",
+  "Token limit": "令牌限制",
+  "Cost limit": "费用限制",
+  "Timezone": "时区",
+  "Unlimited": "无限制",
+  "Add quota": "添加配额",
+  "Edit quota": "编辑配额",
+  "Delete this quota?": "删除此配额？",
+  "Daily": "每日",
+  "Monthly": "每月",
+  "Reject unpriced models": "拒绝未定价模型",
+  "Missing usage behavior": "缺少用量时的行为",
+  "Use reserved estimate": "使用预留估算",
+  "Allow without token charge": "允许且不计令牌",
+  "Started at": "开始时间",
+  "Requested model": "请求模型",
+  "Resolved service": "解析后的服务",
+  "Resolved model": "解析后的模型",
+  "Input tokens": "输入令牌",
+  "Output tokens": "输出令牌",
+  "Total tokens": "总令牌",
+  "Cost": "费用",
+  "Cost status": "费用状态",
+  "Request ID": "请求 ID",
+  "Failed to load models": "加载模型失败",
+  "Select a model": "选择模型",
+  "Select an AI Employee": "选择 AI 员工",
+  "Usage guide": "使用指南",
+  "OpenAI-compatible endpoint": "OpenAI 兼容端点",
+  "Base URL": "基础 URL",
+  "Use a NocoBase API key as the Bearer token.": "使用 NocoBase API 密钥作为 Bearer Token。",
+  "List available models": "列出可用模型",
+  "Send a chat completion": "发送聊天补全请求",
+  "Usage filters": "用量筛选",
+  "Time range": "时间范围",
+  "User ID": "用户 ID",
+  "Succeeded": "成功",
+  "Failed": "失败",
+  "Started": "已开始",
+  "Apply filters": "应用筛选",
+  "Reset": "重置",
+  "Requests": "请求数",
+  "Total cost": "总费用",
+  "Usage records": "用量记录"
+}