@lobehub/lobehub 2.0.0-next.173 → 2.0.0-next.174

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

@@ -441,6 +441,13 @@
       "when": 1765728439737,
       "tag": "0062_add_more_index",
       "breakpoints": true
+    },
+    {
+      "idx": 63,
+      "version": "7",
+      "when": 1766157362540,
+      "tag": "0063_add_columns_for_several_tables",
+      "breakpoints": true
     }
   ],
   "version": "6"

package/packages/database/src/core/migrations.json

@@ -994,5 +994,34 @@
     "bps": true,
     "folderMillis": 1765728439737,
     "hash": "ba6a6beff2ad39419ec4aa7887539c9db0472e13f9242cb8780ff96eec450268"
+  },
+  {
+    "sql": [
+      "DROP INDEX IF EXISTS \"user_memories_contexts_title_vector_index\";",
+      "\nALTER TABLE \"agents\" ALTER COLUMN \"plugins\" DROP DEFAULT;",
+      "\nALTER TABLE \"agents\" ADD COLUMN IF NOT EXISTS \"pinned\" boolean;",
+      "\nALTER TABLE \"message_groups\" ADD COLUMN IF NOT EXISTS \"type\" text;",
+      "\nALTER TABLE \"message_groups\" ADD COLUMN IF NOT EXISTS \"content\" text;",
+      "\nALTER TABLE \"message_groups\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;",
+      "\nALTER TABLE \"messages\" ADD COLUMN IF NOT EXISTS \"summary\" text;",
+      "\nALTER TABLE \"threads\" ADD COLUMN IF NOT EXISTS \"agent_id\" text;",
+      "\nALTER TABLE \"threads\" ADD COLUMN IF NOT EXISTS \"group_id\" text;",
+      "\nALTER TABLE \"threads\" ADD COLUMN IF NOT EXISTS \"metadata\" jsonb;",
+      "\nALTER TABLE \"user_settings\" ADD COLUMN IF NOT EXISTS \"memory\" jsonb;",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"interests\" varchar(64)[];",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"onboarding\" jsonb;",
+      "\nDO $$ BEGIN\n  IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'threads_agent_id_agents_id_fk') THEN\n    ALTER TABLE \"threads\" ADD CONSTRAINT \"threads_agent_id_agents_id_fk\" FOREIGN KEY (\"agent_id\") REFERENCES \"public\".\"agents\"(\"id\") ON DELETE cascade ON UPDATE no action;\n  END IF;\nEND $$;",
+      "\nDO $$ BEGIN\n  IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'threads_group_id_chat_groups_id_fk') THEN\n    ALTER TABLE \"threads\" ADD CONSTRAINT \"threads_group_id_chat_groups_id_fk\" FOREIGN KEY (\"group_id\") REFERENCES \"public\".\"chat_groups\"(\"id\") ON DELETE cascade ON UPDATE no action;\n  END IF;\nEND $$;",
+      "\nCREATE INDEX IF NOT EXISTS \"ai_models_user_id_idx\" ON \"ai_models\" USING btree (\"user_id\");",
+      "\nCREATE INDEX IF NOT EXISTS \"ai_providers_user_id_idx\" ON \"ai_providers\" USING btree (\"user_id\");",
+      "\nCREATE INDEX IF NOT EXISTS \"message_groups_type_idx\" ON \"message_groups\" USING btree (\"type\");",
+      "\nCREATE INDEX IF NOT EXISTS \"message_plugins_tool_call_id_idx\" ON \"message_plugins\" USING btree (\"tool_call_id\");",
+      "\nCREATE INDEX IF NOT EXISTS \"threads_agent_id_idx\" ON \"threads\" USING btree (\"agent_id\");",
+      "\nCREATE INDEX IF NOT EXISTS \"threads_group_id_idx\" ON \"threads\" USING btree (\"group_id\");",
+      "\nALTER TABLE \"user_memories_contexts\" DROP COLUMN IF EXISTS \"title_vector\";\n"
+    ],
+    "bps": true,
+    "folderMillis": 1766157362540,
+    "hash": "7a8ee107778222390e676951173baa81bfa09dd47216a8467575fca54915172c"
   }
 ]

package/packages/database/src/schemas/_helpers.ts

@@ -10,7 +10,11 @@ export const updatedAt = () =>
     .notNull()
     .defaultNow()
     .$onUpdate(() => new Date());
-export const accessedAt = () => timestamptz('accessed_at').notNull().defaultNow();
+export const accessedAt = () =>
+  timestamptz('accessed_at')
+    .notNull()
+    .defaultNow()
+    .$onUpdate(() => new Date());
 
 // columns.helpers.ts
 export const timestamps = {

package/packages/database/src/schemas/agent.ts

@@ -37,7 +37,7 @@ export const agents = pgTable(
     backgroundColor: text('background_color'),
     marketIdentifier: text('market_identifier'),
 
-    plugins: jsonb('plugins').$type<string[]>().default([]),
+    plugins: jsonb('plugins').$type<string[]>(),
 
     clientId: text('client_id'),
 
@@ -55,6 +55,7 @@ export const agents = pgTable(
     tts: jsonb('tts').$type<LobeAgentTTSConfig>(),
 
     virtual: boolean('virtual').default(false),
+    pinned: boolean('pinned'),
 
     openingMessage: text('opening_message'),
     openingQuestions: text('opening_questions').array().default([]),

package/packages/database/src/schemas/aiInfra.ts

@@ -1,6 +1,15 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
 import type { AiProviderConfig, AiProviderSettings } from '@lobechat/types';
-import { boolean, integer, jsonb, pgTable, primaryKey, text, varchar } from 'drizzle-orm/pg-core';
+import {
+  boolean,
+  index,
+  integer,
+  jsonb,
+  pgTable,
+  primaryKey,
+  text,
+  varchar,
+} from 'drizzle-orm/pg-core';
 import { AiModelSettings } from 'model-bank';
 
 import { timestamps } from './_helpers';
@@ -36,7 +45,10 @@ export const aiProviders = pgTable(
 
     ...timestamps,
   },
-  (table) => [primaryKey({ columns: [table.id, table.userId] })],
+  (table) => [
+    primaryKey({ columns: [table.id, table.userId] }),
+    index('ai_providers_user_id_idx').on(table.userId),
+  ],
 );
 
 export type NewAiProviderItem = Omit<typeof aiProviders.$inferInsert, 'userId'>;
@@ -68,7 +80,10 @@ export const aiModels = pgTable(
 
     ...timestamps,
   },
-  (table) => [primaryKey({ columns: [table.id, table.providerId, table.userId] })],
+  (table) => [
+    primaryKey({ columns: [table.id, table.providerId, table.userId] }),
+    index('ai_models_user_id_idx').on(table.userId),
+  ],
 );
 
 export type NewAiModelItem = Omit<typeof aiModels.$inferInsert, 'userId'>;

package/packages/database/src/schemas/message.ts

@@ -58,6 +58,11 @@ export const messageGroups = pgTable(
     title: varchar255('title'),
     description: text('description'),
 
+    // Compression fields
+    type: text('type', { enum: ['parallel', 'compression'] }),
+    content: text('content'), // compression summary (plain text)
+    editorData: jsonb('editor_data'), // rich text editor data (future extension)
+
     clientId: varchar255('client_id'),
 
     ...timestamps,
@@ -65,6 +70,7 @@ export const messageGroups = pgTable(
   (t) => [
     uniqueIndex('message_groups_client_id_user_id_unique').on(t.clientId, t.userId),
     index('message_groups_topic_id_idx').on(t.topicId),
+    index('message_groups_type_idx').on(t.type),
   ],
 );
 
@@ -84,6 +90,7 @@ export const messages = pgTable(
     role: varchar255('role').notNull(),
     content: text('content'),
     editorData: jsonb('editor_data'),
+    summary: text('summary'),
     reasoning: jsonb('reasoning').$type<ModelReasoning>(),
     search: jsonb('search').$type<GroundingSearch>(),
     metadata: jsonb('metadata'),
@@ -164,12 +171,10 @@ export const messagePlugins = pgTable(
       .references(() => users.id, { onDelete: 'cascade' })
       .notNull(),
   },
-  (t) => ({
-    clientIdUnique: uniqueIndex('message_plugins_client_id_user_id_unique').on(
-      t.clientId,
-      t.userId,
-    ),
-  }),
+  (t) => [
+    uniqueIndex('message_plugins_client_id_user_id_unique').on(t.clientId, t.userId),
+    index('message_plugins_tool_call_id_idx').on(t.toolCallId),
+  ],
 );
 
 export const messageTTS = pgTable(

package/packages/database/src/schemas/topic.ts

@@ -1,5 +1,5 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
-import type { ChatTopicMetadata } from '@lobechat/types';
+import type { ChatTopicMetadata, ThreadMetadata } from '@lobechat/types';
 import { sql } from 'drizzle-orm';
 import { boolean, index, jsonb, pgTable, primaryKey, text, uniqueIndex } from 'drizzle-orm/pg-core';
 import { createInsertSchema } from 'drizzle-zod';
@@ -63,7 +63,16 @@ export const threads = pgTable(
     editor_data: jsonb('editor_data'),
     type: text('type', { enum: ['continuation', 'standalone', 'isolation'] }).notNull(),
     status: text('status', {
-      enum: ['active', 'processing', 'pending', 'inReview', 'todo', 'cancel'],
+      enum: [
+        'active',
+        'processing',
+        'pending',
+        'inReview',
+        'todo',
+        'cancel',
+        'completed',
+        'failed',
+      ],
     }),
 
     topicId: text('topic_id')
@@ -74,6 +83,10 @@ export const threads = pgTable(
     parentThreadId: text('parent_thread_id').references(() => threads.id, { onDelete: 'set null' }),
     clientId: text('client_id'),
 
+    agentId: text('agent_id').references(() => agents.id, { onDelete: 'cascade' }),
+    groupId: text('group_id').references(() => chatGroups.id, { onDelete: 'cascade' }),
+    metadata: jsonb('metadata').$type<ThreadMetadata | undefined>(),
+
     userId: text('user_id')
       .references(() => users.id, { onDelete: 'cascade' })
       .notNull(),
@@ -84,6 +97,8 @@ export const threads = pgTable(
   (t) => [
     uniqueIndex('threads_client_id_user_id_unique').on(t.clientId, t.userId),
     index('threads_topic_id_idx').on(t.topicId),
+    index('threads_agent_id_idx').on(t.agentId),
+    index('threads_group_id_idx').on(t.groupId),
   ],
 );
 

package/packages/database/src/schemas/user.ts

@@ -1,9 +1,9 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
 import { DEFAULT_PREFERENCE } from '@lobechat/const';
-import type { CustomPluginParams } from '@lobechat/types';
+import type { CustomPluginParams, UserOnboarding } from '@lobechat/types';
 import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
 import { sql } from 'drizzle-orm';
-import { boolean, index, jsonb, pgTable, primaryKey, text } from 'drizzle-orm/pg-core';
+import { boolean, index, jsonb, pgTable, primaryKey, text, varchar } from 'drizzle-orm/pg-core';
 
 import { timestamps, timestamptz, varchar255 } from './_helpers';
 
@@ -20,8 +20,10 @@ export const users = pgTable(
     firstName: text('first_name'),
     lastName: text('last_name'),
     fullName: text('full_name'),
+    interests: varchar('interests', { length: 64 }).array(),
 
     isOnboarded: boolean('is_onboarded').default(false),
+    onboarding: jsonb('onboarding').$type<UserOnboarding>(),
     // Time user was created in Clerk
     clerkCreatedAt: timestamptz('clerk_created_at'),
 
@@ -77,6 +79,7 @@ export const userSettings = pgTable('user_settings', {
   systemAgent: jsonb('system_agent'),
   defaultAgent: jsonb('default_agent'),
   market: jsonb('market'),
+  memory: jsonb('memory'),
   tool: jsonb('tool'),
   image: jsonb('image'),
 });

package/packages/database/src/schemas/userMemories.ts

@@ -55,16 +55,21 @@ export const userMemoriesContexts = pgTable(
       .primaryKey(),
 
     userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
-    userMemoryIds: jsonb('user_memory_ids'),
+    userMemoryIds: jsonb('user_memory_ids').$type<string[]>(),
 
     metadata: jsonb('metadata').$type<Record<string, unknown>>(),
     tags: text('tags').array(),
 
-    associatedObjects: jsonb('associated_objects'),
-    associatedSubjects: jsonb('associated_subjects'),
+    associatedObjects:
+      jsonb('associated_objects').$type<
+        { extra?: Record<string, unknown>; name?: string; type?: string }[]
+      >(),
+    associatedSubjects:
+      jsonb('associated_subjects').$type<
+        { extra?: Record<string, unknown>; name?: string; type?: string }[]
+      >(),
 
     title: text('title'),
-    titleVector: vector('title_vector', { dimensions: 1024 }),
     description: text('description'),
     descriptionVector: vector('description_vector', { dimensions: 1024 }),
 
@@ -79,10 +84,6 @@ export const userMemoriesContexts = pgTable(
     ...timestamps,
   },
   (table) => [
-    index('user_memories_contexts_title_vector_index').using(
-      'hnsw',
-      table.titleVector.op('vector_cosine_ops'),
-    ),
     index('user_memories_contexts_description_vector_index').using(
       'hnsw',
       table.descriptionVector.op('vector_cosine_ops'),

package/packages/types/src/topic/thread.ts

@@ -25,6 +25,30 @@ export interface ThreadItem {
   userId: string;
 }
 
+/**
+ * Metadata for Thread, used for agent task execution
+ */
+export interface ThreadMetadata {
+  /** Task completion time */
+  completedAt?: string;
+  /** Execution duration in milliseconds */
+  duration?: number;
+  /** Error message when task failed */
+  error?: string;
+  /** Operation ID for tracking */
+  operationId?: string;
+  /** Task start time, used to calculate duration */
+  startedAt?: string;
+  /** Total cost in dollars */
+  totalCost?: number;
+  /** Total messages created during execution */
+  totalMessages?: number;
+  /** Total tokens consumed */
+  totalTokens?: number;
+  /** Total tool calls made */
+  totalToolCalls?: number;
+}
+
 export interface CreateThreadParams {
   parentThreadId?: string;
   sourceMessageId?: string;

package/packages/types/src/user/index.ts

@@ -1,2 +1,3 @@
+export * from './onboarding';
 export * from './preference';
 export * from './settings';

package/packages/types/src/user/onboarding.ts

@@ -0,0 +1,16 @@
+import { z } from 'zod';
+
+export interface UserOnboarding {
+  /** Current step number (1-based), for resuming onboarding */
+  currentStep?: number;
+  /** Timestamp when onboarding was completed (ISO 8601) */
+  finishedAt?: string;
+  /** Onboarding flow version for future upgrades */
+  version: number;
+}
+
+export const UserOnboardingSchema = z.object({
+  currentStep: z.number().optional(),
+  finishedAt: z.string().optional(),
+  version: z.number(),
+});

package/vitest.config.mts

@@ -47,6 +47,9 @@ export default defineConfig({
       '**/node_modules/**',
       '**/dist/**',
       '**/build/**',
+      '**/tmp/**',
+      '**/temp/**',
+      '**/.temp/**',
       '**/apps/desktop/**',
       '**/apps/mobile/**',
       '**/packages/**',

package/.cursor/rules/group-chat.mdc

@@ -1,35 +0,0 @@
----
-description: Explain how group chat works in LobeHub (Multi-agent orchestratoin)
-globs:
-alwaysApply: false
----
-
-This rule explains how group chat (multi-agent orchestration) works. Not confused with session group, which is a organization method to manage session.
-
-## Key points
-
-- A supervisor will devide who and how will speak next
-- Each agent will speak just like in single chat (if was asked to speak)
-- Not coufused with session group
-
-## Related Files
-
-- src/store/chat/slices/message/supervisor.ts
-- src/store/chat/slices/aiChat/actions/generateAIGroupChat.ts
-- src/prompts/groupChat/index.ts (All prompts here)
-
-## Snippets
-
-```tsx
-// Detect whether in group chat
-const isGroupSession = useSessionStore(sessionSelectors.isCurrentSessionGroupSession);
-
-// Member actions
-const addAgentsToGroup = useChatGroupStore((s) => s.addAgentsToGroup);
-const removeAgentFromGroup = useChatGroupStore((s) => s.removeAgentFromGroup);
-const persistReorder = useChatGroupStore((s) => s.reorderGroupMembers);
-
-// Get group info
-const groupConfig = useChatGroupStore(chatGroupSelectors.currentGroupConfig);
-const currentGroupMemebers = useSessionStore(sessionSelectors.currentGroupAgents);
-```

package/.cursor/rules/react-component.mdc

@@ -1,173 +0,0 @@
----
-description: 
-globs: *.tsx
-alwaysApply: false
----
-
-# react component 编写指南
-
-- 如果要写复杂样式的话用 antd-style ，简单的话可以用 style 属性直接写内联样式
-- 如果需要 flex 布局或者居中布局应该使用 react-layout-kit 的 Flexbox 和 Center 组件
-- 选择组件时优先顺序应该是 src/components > 安装的组件 package > lobe-ui > antd
-- 使用 selector 访问 zustand store 的数据，而不是直接从 store 获取
-
-## antd-style token system
-
-### 访问 token system 的两种方式
-
-#### 使用 antd-style 的 useTheme hook
-
-```tsx
-import { useTheme } from 'antd-style';
-
-const MyComponent = () => {
-  const theme = useTheme();
-
-  return (
-    <div
-      style={{
-        color: theme.colorPrimary,
-        backgroundColor: theme.colorBgContainer,
-        padding: theme.padding,
-        borderRadius: theme.borderRadius,
-      }}
-    >
-      使用主题 token 的组件
-    </div>
-  );
-};
-```
-
-#### 使用 antd-style 的 createStyles
-
-```tsx
-const useStyles = createStyles(({ css, token }) => {
-  return {
-    container: css`
-      background-color: ${token.colorBgContainer};
-      border-radius: ${token.borderRadius}px;
-      padding: ${token.padding}px;
-      color: ${token.colorText};
-    `,
-    title: css`
-      font-size: ${token.fontSizeLG}px;
-      font-weight: ${token.fontWeightStrong};
-      margin-bottom: ${token.marginSM}px;
-    `,
-    content: css`
-      font-size: ${token.fontSize}px;
-      line-height: ${token.lineHeight};
-    `,
-  };
-});
-
-const Card: FC<CardProps> = ({ title, content }) => {
-  const { styles } = useStyles();
-
-  return (
-    <Flexbox className={styles.container}>
-      <div className={styles.title}>{title}</div>
-      <div className={styles.content}>{content}</div>
-    </Flexbox>
-  );
-};
-```
-
-### 一些你经常会忘记使用的 token
-
-请注意使用下面的 token 而不是 css 字面值。可以访问 https://ant.design/docs/react/customize-theme-cn 了解所有 token
-
-- 动画类
-  - token.motionDurationMid
-  - token.motionEaseInOut
-- 包围盒属性
-  - token.paddingSM
-  - token.marginLG
-
-## Lobe UI 包含的组件
-
-- 不知道 `@lobehub/ui` 的组件怎么用，有哪些属性，就自己搜下这个项目其它地方怎么用的，不要瞎猜，大部分组件都是在 antd 的基础上扩展了属性
-- 具体用法不懂可以联网搜索，例如 ActionIcon 就爬取 https://ui.lobehub.com/components/action-icon
-- 可以阅读 `node_modules/@lobehub/ui/es/index.js` 了解有哪些组件，每个组件的属性是什么
-
-- General
-  - ActionIcon
-  - ActionIconGroup
-  - Block
-  - Button
-  - DownloadButton
-  - Icon
-- Data Display
-  - Avatar
-  - AvatarGroup
-  - GroupAvatar
-  - Collapse
-  - FileTypeIcon
-  - FluentEmoji
-  - GuideCard
-  - Highlighter
-  - Hotkey
-  - Image
-  - List
-  - Markdown
-  - SearchResultCards
-  - MaterialFileTypeIcon
-  - Mermaid
-  - Typography
-  - Text
-  - Segmented
-  - Snippet
-  - SortableList
-  - Tag
-  - Tooltip
-  - Video
-- Data Entry
-  - AutoComplete
-  - CodeEditor
-  - ColorSwatches
-  - CopyButton
-  - DatePicker
-  - EditableText
-  - EmojiPicker
-  - Form
-  - FormModal
-  - HotkeyInput
-  - ImageSelect
-  - Input
-  - SearchBar
-  - Select
-  - SliderWithInput
-  - ThemeSwitch
-- Feedback
-  - Alert
-  - Drawer
-  - Modal
-- Layout
-  - DraggablePanel
-  - DraggablePanelBody
-  - DraggablePanelContainer
-  - DraggablePanelFooter
-  - DraggablePanelHeader
-  - Footer
-  - Grid
-  - Header
-  - Layout
-  - LayoutFooter
-  - LayoutHeader
-  - LayoutMain
-  - LayoutSidebar
-  - LayoutSidebarInner
-  - LayoutToc
-  - MaskShadow
-  - ScrollShadow
-- Navigation
-  - Burger
-  - Dropdown
-  - Menu
-  - SideNav
-  - Tabs
-  - Toc
-- Theme
-  - ConfigProvider
-  - FontLoader
-  - ThemeProvider