@lobehub/chat 1.92.0 → 1.92.2

package/src/database/migrations/meta/_journal.json

@@ -168,6 +168,13 @@
       "when": 1748925630721,
       "tag": "0023_remove_param_and_doubao",
       "breakpoints": true
+    },
+    {
+      "idx": 24,
+      "version": "7",
+      "when": 1749301573666,
+      "tag": "0024_add_rbac_tables",
+      "breakpoints": true
     }
   ],
   "version": "6"

package/src/database/schemas/index.ts

@@ -8,6 +8,7 @@ export * from './nextauth';
 export * from './oidc';
 export * from './rag';
 export * from './ragEvals';
+export * from './rbac';
 export * from './relations';
 export * from './session';
 export * from './topic';

package/src/database/schemas/rbac.ts

@@ -0,0 +1,82 @@
+/* eslint-disable sort-keys-fix/sort-keys-fix  */
+import { boolean, index, integer, pgTable, primaryKey, text, timestamp } from 'drizzle-orm/pg-core';
+
+import { timestamps } from './_helpers';
+import { users } from './user';
+
+// Roles table
+export const roles = pgTable('rbac_roles', {
+  id: integer('id').primaryKey().generatedByDefaultAsIdentity(),
+  name: text('name').notNull().unique(), // Role name, e.g.: admin, user, guest
+  displayName: text('display_name').notNull(), // Display name
+  description: text('description'), // Role description
+  isSystem: boolean('is_system').default(false).notNull(), // Whether it's a system role
+  isActive: boolean('is_active').default(true).notNull(), // Whether it's active
+
+  ...timestamps,
+});
+
+export type NewRole = typeof roles.$inferInsert;
+export type RoleItem = typeof roles.$inferSelect;
+
+// Permissions table
+export const permissions = pgTable('rbac_permissions', {
+  id: integer('id').primaryKey().generatedByDefaultAsIdentity(),
+  code: text('code').notNull().unique(), // Permission code, e.g.: chat:create, file:upload
+  name: text('name').notNull(), // Permission name
+  description: text('description'), // Permission description
+  category: text('category').notNull(), // Category it belongs to, e.g.: message, knowledge_base, agent
+  isActive: boolean('is_active').default(true).notNull(), // Whether it's active
+
+  ...timestamps,
+});
+
+export type NewPermission = typeof permissions.$inferInsert;
+export type PermissionItem = typeof permissions.$inferSelect;
+
+// Role-permission association table
+export const rolePermissions = pgTable(
+  'rbac_role_permissions',
+  {
+    roleId: integer('role_id')
+      .references(() => roles.id, { onDelete: 'cascade' })
+      .notNull(),
+    permissionId: integer('permission_id')
+      .references(() => permissions.id, { onDelete: 'cascade' })
+      .notNull(),
+
+    createdAt: timestamp('created_at', { withTimezone: true }).notNull().defaultNow(),
+  },
+  (self) => [
+    primaryKey({ columns: [self.roleId, self.permissionId] }),
+    index('rbac_role_permissions_role_id_idx').on(self.roleId),
+    index('rbac_role_permissions_permission_id_idx').on(self.permissionId),
+  ],
+);
+
+export type NewRolePermission = typeof rolePermissions.$inferInsert;
+export type RolePermissionItem = typeof rolePermissions.$inferSelect;
+
+// User-role association table
+export const userRoles = pgTable(
+  'rbac_user_roles',
+  {
+    userId: text('user_id')
+      .references(() => users.id, { onDelete: 'cascade' })
+      .notNull(),
+    roleId: integer('role_id')
+      .references(() => roles.id, { onDelete: 'cascade' })
+      .notNull(),
+
+    createdAt: timestamp('created_at', { withTimezone: true }).notNull().defaultNow(),
+    expiresAt: timestamp('expires_at', { withTimezone: true }), // Support for temporary roles
+  },
+  (self) => [
+    primaryKey({ columns: [self.userId, self.roleId] }),
+    index('rbac_user_roles_user_id_idx').on(self.userId),
+    index('rbac_user_roles_role_id_idx').on(self.roleId),
+  ],
+);
+
+export type NewUserRole = typeof userRoles.$inferInsert;
+export type UserRoleItem = typeof userRoles.$inferSelect;

package/src/libs/model-runtime/azureOpenai/index.ts

@@ -5,7 +5,14 @@ import { systemToUserModels } from '@/const/models';
 
 import { LobeRuntimeAI } from '../BaseAI';
 import { AgentRuntimeErrorType } from '../error';
-import { ChatMethodOptions, ChatStreamPayload, ModelProvider } from '../types';
+import {
+  ChatMethodOptions,
+  ChatStreamPayload,
+  Embeddings,
+  EmbeddingsOptions,
+  EmbeddingsPayload,
+  ModelProvider,
+} from '../types';
 import { AgentRuntimeError } from '../utils/createError';
 import { debugStream } from '../utils/debugStream';
 import { transformResponseToStream } from '../utils/openaiCompatibleFactory';
@@ -75,33 +82,50 @@ export class LobeAzureOpenAI implements LobeRuntimeAI {
         });
       }
     } catch (e) {
-      let error = e as { [key: string]: any; code: string; message: string };
+      return this.handleError(e, model);
+    }
+  }
+
+  async embeddings(payload: EmbeddingsPayload, options?: EmbeddingsOptions): Promise<Embeddings[]> {
+    try {
+      const res = await this.client.embeddings.create(
+        { ...payload, encoding_format: 'float', user: options?.user },
+        { headers: options?.headers, signal: options?.signal },
+      );
+
+      return res.data.map((item) => item.embedding);
+    } catch (error) {
+      return this.handleError(error, payload.model);
+    }
+  }
+
+  protected handleError(e: any, model?: string): never {
+    let error = e as { [key: string]: any; code: string; message: string };
 
-      if (error.code) {
-        switch (error.code) {
-          case 'DeploymentNotFound': {
-            error = { ...error, deployId: model };
-          }
+    if (error.code) {
+      switch (error.code) {
+        case 'DeploymentNotFound': {
+          error = { ...error, deployId: model };
         }
-      } else {
-        error = {
-          cause: error.cause,
-          message: error.message,
-          name: error.name,
-        } as any;
       }
+    } else {
+      error = {
+        cause: error.cause,
+        message: error.message,
+        name: error.name,
+      } as any;
+    }
 
-      const errorType = error.code
-        ? AgentRuntimeErrorType.ProviderBizError
-        : AgentRuntimeErrorType.AgentRuntimeError;
+    const errorType = error.code
+      ? AgentRuntimeErrorType.ProviderBizError
+      : AgentRuntimeErrorType.AgentRuntimeError;
 
-      throw AgentRuntimeError.chat({
-        endpoint: this.maskSensitiveUrl(this.baseURL),
-        error,
-        errorType,
-        provider: ModelProvider.Azure,
-      });
-    }
+    throw AgentRuntimeError.chat({
+      endpoint: this.maskSensitiveUrl(this.baseURL),
+      error,
+      errorType,
+      provider: ModelProvider.Azure,
+    });
   }
 
   // Convert object keys to camel case, copy from `@azure/openai` in `node_modules/@azure/openai/dist/index.cjs`

package/src/store/user/slices/auth/selectors.test.ts

@@ -33,6 +33,82 @@ afterEach(() => {
 });
 
 describe('userProfileSelectors', () => {
+  describe('displayUserName', () => {
+    it('should return default username when auth is disabled and not desktop', () => {
+      enableAuth = false;
+      isDesktop = false;
+
+      const store: UserStore = {
+        isSignedIn: false,
+        user: null,
+        enableAuth: () => false,
+      } as unknown as UserStore;
+
+      expect(userProfileSelectors.displayUserName(store)).toBe('LobeChat');
+    });
+
+    it('should return user username when auth is disabled and is desktop', () => {
+      enableAuth = false;
+      isDesktop = true;
+
+      const store: UserStore = {
+        isSignedIn: false,
+        user: { username: 'johndoe' },
+        enableAuth: () => false,
+      } as unknown as UserStore;
+
+      expect(userProfileSelectors.displayUserName(store)).toBe('johndoe');
+    });
+
+    it('should return user username when signed in', () => {
+      const store: UserStore = {
+        isSignedIn: true,
+        user: { username: 'johndoe' },
+        enableAuth: () => true,
+      } as UserStore;
+
+      expect(userProfileSelectors.displayUserName(store)).toBe('johndoe');
+    });
+
+    it('should return email when signed in but username is not existed in UserStore', () => {
+      const store: UserStore = {
+        isSignedIn: true,
+        user: { email: 'demo@lobehub.com' },
+        enableAuth: () => true,
+      } as UserStore;
+
+      expect(userProfileSelectors.displayUserName(store)).toBe('demo@lobehub.com');
+    });
+
+    it('should return "anonymous" when not signed in', () => {
+      const store: UserStore = {
+        enableAuth: () => true,
+        isSignedIn: false,
+        user: null,
+      } as unknown as UserStore;
+
+      expect(userProfileSelectors.displayUserName(store)).toBe('anonymous');
+    });
+  });
+
+  describe('email', () => {
+    it('should return user email if exist', () => {
+      const store: UserStore = {
+        user: { email: 'demo@lobehub.com' },
+      } as UserStore;
+
+      expect(userProfileSelectors.email(store)).toBe('demo@lobehub.com');
+    });
+
+    it('should return empty string if not exist', () => {
+      const store: UserStore = {
+        user: { email: undefined },
+      } as UserStore;
+
+      expect(userProfileSelectors.email(store)).toBe('');
+    });
+  });
+
   describe('fullName', () => {
     it('should return user fullName if exist', () => {
       const store: UserStore = {

package/src/store/user/slices/auth/selectors.ts

@@ -36,6 +36,8 @@ const username = (s: UserStore) => {
 };
 
 export const userProfileSelectors = {
+  displayUserName: (s: UserStore): string => username(s) || s.user?.email || '',
+  email: (s: UserStore): string => s.user?.email || '',
   fullName: (s: UserStore): string => s.user?.fullName || '',
   nickName,
   userAvatar: (s: UserStore): string => s.user?.avatar || '',

package/src/utils/client/parserPlaceholder.test.ts

@@ -1,35 +1,36 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { parsePlaceholderVariablesMessages, VARIABLE_GENERATORS } from './parserPlaceholder';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { VARIABLE_GENERATORS, parsePlaceholderVariablesMessages } from './parserPlaceholder';
 
 // Mock dependencies
 vi.mock('@/utils/uuid', () => ({
-  uuid: () => 'mocked-uuid-12345'
+  uuid: () => 'mocked-uuid-12345',
 }));
 
 vi.mock('@/store/user', () => ({
   useUserStore: {
-    getState: () => ({})
-  }
+    getState: () => ({}),
+  },
 }));
 
 vi.mock('@/store/user/selectors', () => ({
   userProfileSelectors: {
+    displayUserName: () => 'testuser',
     nickName: () => 'Test User',
-    username: () => 'testuser',
-    fullName: () => 'Test Full Name'
-  }
+    fullName: () => 'Test Full Name',
+  },
 }));
 
 vi.mock('@/store/agent/store', () => ({
-  getAgentStoreState: () => ({})
+  getAgentStoreState: () => ({}),
 }));
 
 vi.mock('@/store/agent/selectors', () => ({
   agentChatConfigSelectors: {
     currentChatConfig: () => ({
-      inputTemplate: 'Hello {{username}}!'
-    })
-  }
+      inputTemplate: 'Hello {{username}}!',
+    }),
+  },
 }));
 
 describe('parsePlaceholderVariablesMessages', () => {
@@ -52,8 +53,8 @@ describe('parsePlaceholderVariablesMessages', () => {
       const messages = [
         {
           id: '1',
-          content: 'Hello {{username}}, today is {{date}}'
-        }
+          content: 'Hello {{username}}, today is {{date}}',
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -66,8 +67,8 @@ describe('parsePlaceholderVariablesMessages', () => {
       const messages = [
         {
           id: '1',
-          content: 'Time: {{time}}, Date: {{date}}, User: {{nickname}}'
-        }
+          content: 'Time: {{time}}, Date: {{date}}, User: {{nickname}}',
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -81,8 +82,8 @@ describe('parsePlaceholderVariablesMessages', () => {
         {
           id: '1',
           role: 'user',
-          content: 'Hello {{username}}'
-        }
+          content: 'Hello {{username}}',
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -90,7 +91,7 @@ describe('parsePlaceholderVariablesMessages', () => {
       expect(result[0]).toEqual({
         id: '1',
         role: 'user',
-        content: 'Hello testuser'
+        content: 'Hello testuser',
       });
     });
   });
@@ -103,14 +104,14 @@ describe('parsePlaceholderVariablesMessages', () => {
           content: [
             {
               type: 'text',
-              text: 'Hello {{username}}'
+              text: 'Hello {{username}}',
             },
             {
               type: 'image_url',
-              image_url: 'image.jpg'
-            }
-          ]
-        }
+              image_url: 'image.jpg',
+            },
+          ],
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -118,7 +119,7 @@ describe('parsePlaceholderVariablesMessages', () => {
       expect(result[0].content[0].text).toBe('Hello testuser');
       expect(result[0].content[1]).toEqual({
         type: 'image_url',
-        image_url: 'image.jpg'
+        image_url: 'image.jpg',
       });
     });
 
@@ -129,18 +130,18 @@ describe('parsePlaceholderVariablesMessages', () => {
           content: [
             {
               type: 'text',
-              text: 'Date: {{date}}'
+              text: 'Date: {{date}}',
             },
             {
               type: 'text',
-              text: 'Time: {{time}}'
+              text: 'Time: {{time}}',
             },
             {
               type: 'image_url',
-              image_url: 'test.jpg'
-            }
-          ]
-        }
+              image_url: 'test.jpg',
+            },
+          ],
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -149,7 +150,7 @@ describe('parsePlaceholderVariablesMessages', () => {
       expect(result[0].content[1].text).toContain(new Date().toLocaleTimeString());
       expect(result[0].content[2]).toEqual({
         type: 'image_url',
-        image_url: 'test.jpg'
+        image_url: 'test.jpg',
       });
     });
 
@@ -164,10 +165,10 @@ describe('parsePlaceholderVariablesMessages', () => {
             },
             {
               type: 'image_url',
-              name: 'image2.jpg'
-            }
-          ]
-        }
+              name: 'image2.jpg',
+            },
+          ],
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -175,12 +176,12 @@ describe('parsePlaceholderVariablesMessages', () => {
       expect(result[0].content).toEqual([
         {
           type: 'image_url',
-          image_url: 'image.jpg'
+          image_url: 'image.jpg',
         },
         {
           type: 'image_url',
-          name: 'image2.jpg'
-        }
+          name: 'image2.jpg',
+        },
       ]);
     });
   });
@@ -192,25 +193,19 @@ describe('parsePlaceholderVariablesMessages', () => {
     });
 
     it('should handle messages without content', () => {
-      const messages = [
-        { id: '1' },
-        { id: '2', content: null },
-        { id: '3', content: undefined }
-      ];
+      const messages = [{ id: '1' }, { id: '2', content: null }, { id: '3', content: undefined }];
 
       const result = parsePlaceholderVariablesMessages(messages);
 
       expect(result).toEqual([
         { id: '1' },
         { id: '2', content: null },
-        { id: '3', content: undefined }
+        { id: '3', content: undefined },
       ]);
     });
 
     it('should handle empty string content', () => {
-      const messages = [
-        { id: '1', content: '' }
-      ];
+      const messages = [{ id: '1', content: '' }];
 
       const result = parsePlaceholderVariablesMessages(messages);
 
@@ -220,13 +215,13 @@ describe('parsePlaceholderVariablesMessages', () => {
     it('should handle content without variables', () => {
       const messages = [
         { id: '1', content: 'Hello world!' },
-        { 
-          id: '2', 
+        {
+          id: '2',
           content: [
             { type: 'text', text: 'No variables here' },
-            { type: 'image_url', image_url: 'test.jpg' }
-          ]
-        }
+            { type: 'image_url', image_url: 'test.jpg' },
+          ],
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -236,9 +231,7 @@ describe('parsePlaceholderVariablesMessages', () => {
     });
 
     it('should handle unknown variable types', () => {
-      const messages = [
-        { id: '1', content: 'Hello {{unknown_variable}}!' }
-      ];
+      const messages = [{ id: '1', content: 'Hello {{unknown_variable}}!' }];
 
       const result = parsePlaceholderVariablesMessages(messages);
 
@@ -247,9 +240,7 @@ describe('parsePlaceholderVariablesMessages', () => {
     });
 
     it('should handle nested variables (input_template)', () => {
-      const messages = [
-        { id: '1', content: 'Template: {{input_template}}' }
-      ];
+      const messages = [{ id: '1', content: 'Template: {{input_template}}' }];
 
       const result = parsePlaceholderVariablesMessages(messages);
 
@@ -261,10 +252,10 @@ describe('parsePlaceholderVariablesMessages', () => {
   describe('specific variable types', () => {
     it('should handle time variables', () => {
       const messages = [
-        { 
-          id: '1', 
-          content: 'Year: {{year}}, Month: {{month}}, Day: {{day}}' 
-        }
+        {
+          id: '1',
+          content: 'Year: {{year}}, Month: {{month}}, Day: {{day}}',
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -276,10 +267,10 @@ describe('parsePlaceholderVariablesMessages', () => {
 
     it('should handle random variables', () => {
       const messages = [
-        { 
-          id: '1', 
-          content: 'Random: {{random}}, Bool: {{random_bool}}, UUID: {{uuid}}' 
-        }
+        {
+          id: '1',
+          content: 'Random: {{random}}, Bool: {{random_bool}}, UUID: {{uuid}}',
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -291,10 +282,10 @@ describe('parsePlaceholderVariablesMessages', () => {
 
     it('should handle user variables', () => {
       const messages = [
-        { 
-          id: '1', 
-          content: 'User: {{username}}, Nickname: {{nickname}}' 
-        }
+        {
+          id: '1',
+          content: 'User: {{username}}, Nickname: {{nickname}}',
+        },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);
@@ -307,13 +298,11 @@ describe('parsePlaceholderVariablesMessages', () => {
     it('should process multiple messages correctly', () => {
       const messages = [
         { id: '1', content: 'Hello {{username}}' },
-        { 
-          id: '2', 
-          content: [
-            { type: 'text', text: 'Today is {{date}}' }
-          ]
+        {
+          id: '2',
+          content: [{ type: 'text', text: 'Today is {{date}}' }],
         },
-        { id: '3', content: 'Time: {{time}}' }
+        { id: '3', content: 'Time: {{time}}' },
       ];
 
       const result = parsePlaceholderVariablesMessages(messages);

package/src/utils/client/parserPlaceholder.ts

@@ -53,12 +53,14 @@ export const VARIABLE_GENERATORS = {
   *
   * | Value | Example |
   * |-------|---------|
+  * | `{{email}}` | demo@lobehub.com |
   * | `{{nickname}}` | 社区版用户 |
   * | `{{username}}` | LobeChat |
   *
   */
+  email: () => userProfileSelectors.email(useUserStore.getState()) ?? '',
   nickname: () => userProfileSelectors.nickName(useUserStore.getState()) ?? '',
-  username: () => userProfileSelectors.username(useUserStore.getState()) ?? userProfileSelectors.fullName(useUserStore.getState()) ?? '',
+  username: () => userProfileSelectors.displayUserName(useUserStore.getState()) ?? userProfileSelectors.fullName(useUserStore.getState()) ?? '',
 
   /**
   * 随机值类模板变量