@lobehub/chat 1.46.7 → 1.47.0

package/CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+## [Version 1.47.0](https://github.com/lobehub/lobe-chat/compare/v1.46.7...v1.47.0)
+
+<sup>Released on **2025-01-17**</sup>
+
+#### ✨ Features
+
+- **misc**: Support new ai provider in client pglite.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's improved
+
+- **misc**: Support new ai provider in client pglite, closes [#5488](https://github.com/lobehub/lobe-chat/issues/5488) ([08f505f](https://github.com/lobehub/lobe-chat/commit/08f505f))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.46.7](https://github.com/lobehub/lobe-chat/compare/v1.46.6...v1.46.7)
 
 <sup>Released on **2025-01-17**</sup>

package/changelog/v1.json

@@ -1,4 +1,13 @@
 [
+  {
+    "children": {
+      "features": [
+        "Support new ai provider in client pglite."
+      ]
+    },
+    "date": "2025-01-17",
+    "version": "1.47.0"
+  },
   {
     "children": {
       "fixes": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.46.7",
+  "version": "1.47.0",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",
@@ -206,6 +206,7 @@
     "react-layout-kit": "^1.9.1",
     "react-lazy-load": "^4.0.1",
     "react-pdf": "^9.2.1",
+    "react-rnd": "^10.4.14",
     "react-scan": "^0.0.54",
     "react-virtuoso": "^4.12.3",
     "react-wrap-balancer": "^1.1.1",

package/src/app/(main)/discover/(detail)/provider/[slug]/features/ProviderConfig.tsx

@@ -10,7 +10,7 @@ import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { FlexboxProps } from 'react-layout-kit';
 
-import { isServerMode } from '@/const/version';
+import { isDeprecatedEdition } from '@/const/version';
 import { DiscoverProviderItem } from '@/types/discover';
 
 const useStyles = createStyles(({ css }) => ({
@@ -32,7 +32,7 @@ const ProviderConfig = memo<ProviderConfigProps>(({ data, identifier }) => {
 
   const router = useRouter();
   const openSettings = () => {
-    router.push(!isServerMode ? '/settings/llm' : `/settings/provider/${identifier}`);
+    router.push(isDeprecatedEdition ? '/settings/llm' : `/settings/provider/${identifier}`);
   };
 
   const icon = <Icon icon={SquareArrowOutUpRight} size={{ fontSize: 16 }} />;

package/src/app/(main)/settings/hooks/useCategory.tsx

@@ -7,7 +7,7 @@ import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import type { MenuProps } from '@/components/Menu';
-import { isServerMode } from '@/const/version';
+import { isDeprecatedEdition } from '@/const/version';
 import { SettingsTabs } from '@/store/global/initialState';
 import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
 
@@ -53,7 +53,7 @@ export const useCategory = () => {
         },
         showLLM &&
         // TODO: Remove /llm when v2.0
-        !isServerMode
+        (isDeprecatedEdition
           ? {
               icon: <Icon icon={Brain} />,
               key: SettingsTabs.LLM,
@@ -71,7 +71,7 @@ export const useCategory = () => {
                   {t('tab.provider')}
                 </Link>
               ),
-            },
+            }),
 
         enableSTT && {
           icon: <Icon icon={Mic2} />,

package/src/app/(main)/settings/provider/(detail)/[id]/ClientMode.tsx

@@ -0,0 +1,25 @@
+'use client';
+
+import { memo } from 'react';
+import { Flexbox } from 'react-layout-kit';
+import useSWR from 'swr';
+
+import { aiProviderService } from '@/services/aiProvider';
+
+import ModelList from '../../features/ModelList';
+import ProviderConfig from '../../features/ProviderConfig';
+
+const ClientMode = memo<{ id: string }>(({ id }) => {
+  const { data, isLoading } = useSWR('get-client-provider', () =>
+    aiProviderService.getAiProviderById(id),
+  );
+
+  return (
+    <Flexbox gap={24} paddingBlock={8}>
+      {!isLoading && data && <ProviderConfig {...data} />}
+      <ModelList id={id} />
+    </Flexbox>
+  );
+});
+
+export default ClientMode;

package/src/app/(main)/settings/provider/(detail)/[id]/page.tsx

@@ -8,6 +8,7 @@ import { KeyVaultsGateKeeper } from '@/server/modules/KeyVaultsEncrypt';
 import { PagePropsWithId } from '@/types/next';
 import { getUserAuth } from '@/utils/server/auth';
 
+import ClientMode from './ClientMode';
 import ProviderDetail from './index';
 
 const Page = async (props: PagePropsWithId) => {
@@ -33,7 +34,7 @@ const Page = async (props: PagePropsWithId) => {
     return <ProviderDetail {...userCard} />;
   }
 
-  return <div>not found</div>;
+  return <ClientMode id={params.id} />;
 };
 
 export default Page;

package/src/app/(main)/settings/provider/ProviderMenu/SortProviderModal/index.tsx

@@ -72,7 +72,6 @@ const ConfigGroupModal = memo<ConfigGroupModalProps>(({ open, onCancel, defaultI
               id: item.id,
               sort: index,
             }));
-            console.log(sortMap);
             setLoading(true);
             await updateAiProviderSort(sortMap);
             setLoading(false);

package/src/database/client/migrations.json

@@ -282,5 +282,16 @@
     "bps": true,
     "folderMillis": 1731858381716,
     "hash": "d8263bfefe296ed366379c7b7fc65195d12e6a1c0a9f1c96097ea28f2123fe50"
+  },
+  {
+    "sql": [
+      "CREATE TABLE \"ai_models\" (\n\t\"id\" varchar(150) NOT NULL,\n\t\"display_name\" varchar(200),\n\t\"description\" text,\n\t\"organization\" varchar(100),\n\t\"enabled\" boolean,\n\t\"provider_id\" varchar(64) NOT NULL,\n\t\"type\" varchar(20) DEFAULT 'chat' NOT NULL,\n\t\"sort\" integer,\n\t\"user_id\" text NOT NULL,\n\t\"pricing\" jsonb,\n\t\"parameters\" jsonb DEFAULT '{}'::jsonb,\n\t\"config\" jsonb,\n\t\"abilities\" jsonb DEFAULT '{}'::jsonb,\n\t\"context_window_tokens\" integer,\n\t\"source\" varchar(20),\n\t\"released_at\" varchar(10),\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\tCONSTRAINT \"ai_models_id_provider_id_user_id_pk\" PRIMARY KEY(\"id\",\"provider_id\",\"user_id\")\n);\n",
+      "\nCREATE TABLE \"ai_providers\" (\n\t\"id\" varchar(64) NOT NULL,\n\t\"name\" text,\n\t\"user_id\" text NOT NULL,\n\t\"sort\" integer,\n\t\"enabled\" boolean,\n\t\"fetch_on_client\" boolean,\n\t\"check_model\" text,\n\t\"logo\" text,\n\t\"description\" text,\n\t\"key_vaults\" text,\n\t\"source\" varchar(20),\n\t\"settings\" jsonb,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\tCONSTRAINT \"ai_providers_id_user_id_pk\" PRIMARY KEY(\"id\",\"user_id\")\n);\n",
+      "\nALTER TABLE \"ai_models\" ADD CONSTRAINT \"ai_models_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"ai_providers\" ADD CONSTRAINT \"ai_providers_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;"
+    ],
+    "bps": true,
+    "folderMillis": 1735834653361,
+    "hash": "845a692ceabbfc3caf252a97d3e19a213bc0c433df2689900135f9cfded2cf49"
   }
 ]

package/src/database/repositories/tableViewer/index.test.ts

@@ -0,0 +1,256 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { clientDB, initializeDB } from '@/database/client/db';
+
+import { TableViewerRepo } from './index';
+
+const userId = 'user-table-viewer';
+const repo = new TableViewerRepo(clientDB as any, userId);
+
+// Mock database execution
+const mockExecute = vi.fn();
+const mockDB = {
+  execute: mockExecute,
+};
+
+beforeEach(async () => {
+  await initializeDB();
+  vi.clearAllMocks();
+});
+
+describe('TableViewerRepo', () => {
+  describe('getAllTables', () => {
+    it('should return all tables with counts', async () => {
+      const result = await repo.getAllTables();
+
+      expect(result.length).toEqual(39);
+      expect(result[0]).toEqual({ name: 'agents', count: 0, type: 'BASE TABLE' });
+    });
+
+    it('should handle custom schema', async () => {
+      const result = await repo.getAllTables('custom_schema');
+      expect(result).toBeDefined();
+    });
+  });
+
+  describe('getTableDetails', () => {
+    it('should return table column details', async () => {
+      const tableName = 'test_table';
+      const mockColumns = {
+        rows: [
+          {
+            column_name: 'id',
+            data_type: 'uuid',
+            is_nullable: 'NO',
+            column_default: null,
+            is_primary_key: true,
+            foreign_key: null,
+          },
+        ],
+      };
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce(mockColumns);
+
+      const result = await testRepo.getTableDetails(tableName);
+
+      expect(result).toEqual([
+        {
+          name: 'id',
+          type: 'uuid',
+          nullable: false,
+          defaultValue: null,
+          isPrimaryKey: true,
+          foreignKey: null,
+        },
+      ]);
+    });
+  });
+
+  describe('getTableData', () => {
+    it('should return paginated data with filters', async () => {
+      const tableName = 'test_table';
+      const pagination = {
+        page: 1,
+        pageSize: 10,
+        sortBy: 'id',
+        sortOrder: 'desc' as const,
+      };
+      const filters = [
+        {
+          column: 'name',
+          operator: 'contains' as const,
+          value: 'test',
+        },
+      ];
+
+      const mockData = {
+        rows: [{ id: 1, name: 'test' }],
+      };
+      const mockCount = {
+        rows: [{ total: 1 }],
+      };
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce(mockData).mockResolvedValueOnce(mockCount);
+
+      const result = await testRepo.getTableData(tableName, pagination, filters);
+
+      expect(result).toEqual({
+        data: mockData.rows,
+        pagination: {
+          page: 1,
+          pageSize: 10,
+          total: 1,
+        },
+      });
+    });
+  });
+
+  describe('updateRow', () => {
+    it('should update and return row data', async () => {
+      const tableName = 'test_table';
+      const id = '123';
+      const primaryKeyColumn = 'id';
+      const data = { name: 'updated' };
+
+      const mockResult = {
+        rows: [{ id: '123', name: 'updated' }],
+      };
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce(mockResult);
+
+      const result = await testRepo.updateRow(tableName, id, primaryKeyColumn, data);
+
+      expect(result).toEqual(mockResult.rows[0]);
+      expect(mockExecute).toHaveBeenCalledTimes(1);
+    });
+  });
+
+  describe('deleteRow', () => {
+    it('should delete a row', async () => {
+      const tableName = 'test_table';
+      const id = '123';
+      const primaryKeyColumn = 'id';
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce({ rows: [] });
+
+      await testRepo.deleteRow(tableName, id, primaryKeyColumn);
+
+      expect(mockExecute).toHaveBeenCalledTimes(1);
+    });
+  });
+
+  describe('insertRow', () => {
+    it('should insert and return new row data', async () => {
+      const tableName = 'test_table';
+      const data = { name: 'new row' };
+
+      const mockResult = {
+        rows: [{ id: '123', name: 'new row' }],
+      };
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce(mockResult);
+
+      const result = await testRepo.insertRow(tableName, data);
+
+      expect(result).toEqual(mockResult.rows[0]);
+      expect(mockExecute).toHaveBeenCalledTimes(1);
+    });
+  });
+
+  describe('getTableCount', () => {
+    it('should return table count', async () => {
+      const tableName = 'test_table';
+      const mockResult = {
+        rows: [{ total: 42 }],
+      };
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce(mockResult);
+
+      const result = await testRepo.getTableCount(tableName);
+
+      expect(result).toBe(42);
+      expect(mockExecute).toHaveBeenCalledTimes(1);
+    });
+  });
+
+  describe('batchDelete', () => {
+    it('should delete multiple rows', async () => {
+      const tableName = 'test_table';
+      const ids = ['1', '2', '3'];
+      const primaryKeyColumn = 'id';
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce({ rows: [] });
+
+      await testRepo.batchDelete(tableName, ids, primaryKeyColumn);
+
+      expect(mockExecute).toHaveBeenCalledTimes(1);
+    });
+  });
+
+  describe('exportTableData', () => {
+    it('should export table data with default pagination', async () => {
+      const tableName = 'test_table';
+      const mockData = {
+        rows: [{ id: 1, name: 'test' }],
+      };
+      const mockCount = {
+        rows: [{ total: 1 }],
+      };
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+      mockExecute.mockResolvedValueOnce(mockData).mockResolvedValueOnce(mockCount);
+
+      const result = await testRepo.exportTableData(tableName);
+
+      expect(result).toEqual({
+        data: mockData.rows,
+        pagination: {
+          page: 1,
+          pageSize: 1000,
+          total: 1,
+        },
+      });
+    });
+
+    it('should export table data with custom pagination and filters', async () => {
+      const tableName = 'test_table';
+      const pagination = { page: 2, pageSize: 50 };
+      const filters = [
+        {
+          column: 'status',
+          operator: 'equals' as const,
+          value: 'active',
+        },
+      ];
+
+      const mockData = {
+        rows: [{ id: 1, status: 'active' }],
+      };
+      const mockCount = {
+        rows: [{ total: 1 }],
+      };
+
+      mockExecute.mockResolvedValueOnce(mockData).mockResolvedValueOnce(mockCount);
+
+      const testRepo = new TableViewerRepo(mockDB as any, userId);
+
+      const result = await testRepo.exportTableData(tableName, pagination, filters);
+
+      expect(result).toEqual({
+        data: mockData.rows,
+        pagination: {
+          page: 2,
+          pageSize: 50,
+          total: 1,
+        },
+      });
+    });
+  });
+});

package/src/database/repositories/tableViewer/index.ts

@@ -0,0 +1,251 @@
+import { sql } from 'drizzle-orm';
+import pMap from 'p-map';
+
+import { LobeChatDatabase } from '@/database/type';
+import {
+  FilterCondition,
+  PaginationParams,
+  TableBasicInfo,
+  TableColumnInfo,
+} from '@/types/tableViewer';
+
+export class TableViewerRepo {
+  private userId: string;
+  private db: LobeChatDatabase;
+
+  constructor(db: LobeChatDatabase, userId: string) {
+    this.userId = userId;
+    this.db = db;
+  }
+
+  /**
+   * 获取数据库中所有的表
+   */
+  async getAllTables(schema = 'public'): Promise<TableBasicInfo[]> {
+    const query = sql`
+      SELECT
+        table_name as name,
+        table_type as type
+      FROM information_schema.tables
+      WHERE table_schema = ${schema}
+      ORDER BY table_name;
+    `;
+
+    const tables = await this.db.execute(query);
+
+    const tableNames = tables.rows.map((row) => row.name) as string[];
+
+    const counts = await pMap(tableNames, async (name) => this.getTableCount(name), {
+      concurrency: 10,
+    });
+
+    return tables.rows.map((row, index) => ({
+      count: counts[index],
+      name: row.name,
+      type: row.type,
+    })) as TableBasicInfo[];
+  }
+
+  /**
+   * 获取指定表的详细结构信息
+   */
+  async getTableDetails(tableName: string): Promise<TableColumnInfo[]> {
+    const query = sql`
+      SELECT
+        c.column_name,
+        c.data_type,
+        c.is_nullable,
+        c.column_default,
+        -- 主键信息
+        (
+          SELECT true
+          FROM information_schema.table_constraints tc
+          JOIN information_schema.key_column_usage kcu
+            ON tc.constraint_name = kcu.constraint_name
+          WHERE tc.table_name = c.table_name
+            AND kcu.column_name = c.column_name
+            AND tc.constraint_type = 'PRIMARY KEY'
+        ) is_primary_key,
+        -- 外键信息
+        (
+          SELECT json_build_object(
+            'table', ccu.table_name,
+            'column', ccu.column_name
+          )
+          FROM information_schema.table_constraints tc
+          JOIN information_schema.key_column_usage kcu
+            ON tc.constraint_name = kcu.constraint_name
+          JOIN information_schema.constraint_column_usage ccu
+            ON ccu.constraint_name = tc.constraint_name
+          WHERE tc.table_name = c.table_name
+            AND kcu.column_name = c.column_name
+            AND tc.constraint_type = 'FOREIGN KEY'
+        ) foreign_key
+      FROM information_schema.columns c
+      WHERE c.table_name = ${tableName}
+      AND c.table_schema = 'public'
+      ORDER BY c.ordinal_position;
+    `;
+
+    const columns = await this.db.execute(query);
+
+    return columns.rows.map((col: any) => ({
+      defaultValue: col.column_default,
+      foreignKey: col.foreign_key,
+      isPrimaryKey: !!col.is_primary_key,
+      name: col.column_name,
+      nullable: col.is_nullable === 'YES',
+      type: col.data_type,
+    }));
+  }
+
+  /**
+   * 获取表数据，支持分页、排序和筛选
+   */
+  async getTableData(tableName: string, pagination: PaginationParams, filters?: FilterCondition[]) {
+    const offset = (pagination.page - 1) * pagination.pageSize;
+
+    // 构建基础查询
+    let baseQuery = sql`SELECT * FROM ${sql.identifier(tableName)}`;
+
+    // 添加筛选条件
+    if (filters && filters.length > 0) {
+      const whereConditions = filters.map((filter) => {
+        const column = sql.identifier(filter.column);
+
+        switch (filter.operator) {
+          case 'equals': {
+            return sql`${column} = ${filter.value}`;
+          }
+          case 'contains': {
+            return sql`${column} ILIKE ${`%${filter.value}%`}`;
+          }
+          case 'startsWith': {
+            return sql`${column} ILIKE ${`${filter.value}%`}`;
+          }
+          case 'endsWith': {
+            return sql`${column} ILIKE ${`%${filter.value}`}`;
+          }
+          default: {
+            return sql`1=1`;
+          }
+        }
+      });
+
+      baseQuery = sql`${baseQuery} WHERE ${sql.join(whereConditions, sql` AND `)}`;
+    }
+
+    // 添加排序
+    if (pagination.sortBy) {
+      const direction = pagination.sortOrder === 'desc' ? sql`DESC` : sql`ASC`;
+      baseQuery = sql`${baseQuery} ORDER BY ${sql.identifier(pagination.sortBy)} ${direction}`;
+    }
+
+    // 添加分页
+    const query = sql`${baseQuery} LIMIT ${pagination.pageSize} OFFSET ${offset}`;
+
+    // 获取总数
+    const countQuery = sql`SELECT COUNT(*) as total FROM ${sql.identifier(tableName)}`;
+
+    // 并行执行查询
+    const [data, count] = await Promise.all([this.db.execute(query), this.db.execute(countQuery)]);
+
+    return {
+      data: data.rows,
+      pagination: {
+        page: pagination.page,
+        pageSize: pagination.pageSize,
+        total: Number(count.rows[0].total),
+      },
+    };
+  }
+
+  /**
+   * 更新表中的一行数据
+   */
+  async updateRow(
+    tableName: string,
+    id: string,
+    primaryKeyColumn: string,
+    data: Record<string, any>,
+  ) {
+    const setColumns = Object.entries(data).map(([key, value]) => {
+      return sql`${sql.identifier(key)} = ${value}`;
+    });
+
+    const query = sql`
+      UPDATE ${sql.identifier(tableName)}
+      SET ${sql.join(setColumns, sql`, `)}
+      WHERE ${sql.identifier(primaryKeyColumn)} = ${id}
+      RETURNING *
+    `;
+
+    const result = await this.db.execute(query);
+    return result.rows[0];
+  }
+
+  /**
+   * 删除表中的一行数据
+   */
+  async deleteRow(tableName: string, id: string, primaryKeyColumn: string) {
+    const query = sql`
+      DELETE FROM ${sql.identifier(tableName)}
+      WHERE ${sql.identifier(primaryKeyColumn)} = ${id}
+    `;
+
+    await this.db.execute(query);
+  }
+
+  /**
+   * 插入新行数据
+   */
+  async insertRow(tableName: string, data: Record<string, any>) {
+    const columns = Object.keys(data).map((key) => sql.identifier(key));
+    const values = Object.values(data);
+
+    const query = sql`
+      INSERT INTO ${sql.identifier(tableName)}
+      (${sql.join(columns, sql`, `)})
+      VALUES (${sql.join(
+        values.map((v) => sql`${v}`),
+        sql`, `,
+      )})
+      RETURNING *
+    `;
+
+    const result = await this.db.execute(query);
+    return result.rows[0];
+  }
+
+  /**
+   * 获取表的总记录数
+   */
+  async getTableCount(tableName: string): Promise<number> {
+    const query = sql`SELECT COUNT(*) as total FROM ${sql.identifier(tableName)}`;
+    const result = await this.db.execute(query);
+    return Number(result.rows[0].total);
+  }
+
+  /**
+   * 批量删除数据
+   */
+  async batchDelete(tableName: string, ids: string[], primaryKeyColumn: string) {
+    const query = sql`
+      DELETE FROM ${sql.identifier(tableName)}
+      WHERE ${sql.identifier(primaryKeyColumn)} = ANY(${ids})
+    `;
+
+    await this.db.execute(query);
+  }
+
+  /**
+   * 导出表数据（支持分页导出）
+   */
+  async exportTableData(
+    tableName: string,
+    pagination?: PaginationParams,
+    filters?: FilterCondition[],
+  ) {
+    return this.getTableData(tableName, pagination || { page: 1, pageSize: 1000 }, filters);
+  }
+}