@lobehub/chat 1.37.1 → 1.38.0

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+## [Version 1.38.0](https://github.com/lobehub/lobe-chat/compare/v1.37.2...v1.38.0)
+
+<sup>Released on **2024-12-23**</sup>
+
+#### ✨ Features
+
+- **misc**: Support thread in client pglite.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's improved
+
+- **misc**: Support thread in client pglite, closes [#5150](https://github.com/lobehub/lobe-chat/issues/5150) ([848b29f](https://github.com/lobehub/lobe-chat/commit/848b29f))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 1.37.2](https://github.com/lobehub/lobe-chat/compare/v1.37.1...v1.37.2)
+
+<sup>Released on **2024-12-22**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Move pglite to client service.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Move pglite to client service, closes [#5133](https://github.com/lobehub/lobe-chat/issues/5133) ([c2ded24](https://github.com/lobehub/lobe-chat/commit/c2ded24))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.37.1](https://github.com/lobehub/lobe-chat/compare/v1.37.0...v1.37.1)
 
 <sup>Released on **2024-12-22**</sup>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "features": [
+        "Support thread in client pglite."
+      ]
+    },
+    "date": "2024-12-23",
+    "version": "1.38.0"
+  },
+  {
+    "children": {
+      "improvements": [
+        "Move pglite to client service."
+      ]
+    },
+    "date": "2024-12-22",
+    "version": "1.37.2"
+  },
   {
     "children": {
       "improvements": [

package/locales/en-US/common.json

@@ -59,14 +59,14 @@
       "features": {
         "knowledgeBase": {
           "desc": "Build your personal knowledge base and easily start conversations with your assistant (coming soon)",
-          "title": "Support for knowledge base conversations, unlock your second brain"
+          "title": "Support for knowledge base conversations"
         },
         "localFirst": {
           "desc": "Chat data is stored entirely in the browser, keeping your data always under your control.",
           "title": "Local first, privacy first"
         },
         "pglite": {
-          "desc": "Built on PGlite, natively supports AI Native advanced features (vector retrieval)",
+          "desc": "Built on PGlite, natively supports AI Native advanced features (vector search)",
           "title": "Next-generation client storage architecture"
         }
       },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.37.1",
+  "version": "1.38.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",

package/src/const/version.ts

@@ -8,6 +8,8 @@ export const CURRENT_VERSION = pkg.version;
 export const isServerMode = getServerDBConfig().NEXT_PUBLIC_ENABLED_SERVER_SERVICE;
 export const isUsePgliteDB = process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite';
 
+export const isDeprecatedEdition = !isServerMode && !isUsePgliteDB;
+
 // @ts-ignore
 export const isCustomBranding = BRANDING_NAME !== 'LobeChat';
 // @ts-ignore

package/src/features/Conversation/hooks/useChatListActionsBar.tsx

@@ -3,7 +3,7 @@ import { Copy, Edit, ListRestart, RotateCcw, Split, Trash } from 'lucide-react';
 import { useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
 
-import { isServerMode } from '@/const/version';
+import { isDeprecatedEdition } from '@/const/version';
 
 interface ChatListActionsBar {
   branching: ActionIconGroupItems;
@@ -23,10 +23,10 @@ export const useChatListActionsBar = ({
   return useMemo(
     () => ({
       branching: {
-        disable: !isServerMode,
+        disable: isDeprecatedEdition,
         icon: Split,
         key: 'branching',
-        label: isServerMode
+        label: !isDeprecatedEdition
           ? t('branching', { defaultValue: 'Create Sub Topic' })
           : t('branchingDisable'),
       },

package/src/locales/default/common.ts

@@ -33,7 +33,7 @@ export default {
   blog: '产品博客',
   branching: '创建子话题',
   branchingDisable:
-    '「子话题」功能仅服务端版本可用，如需该功能，请切换到服务端部署模式或使用 LobeChat Cloud',
+    '「子话题」功能在当前模式下不可用，如需该功能，请切换到 Postgres/Pglite DB 模式或使用 LobeChat Cloud',
   cancel: '取消',
   changelog: '更新日志',
   clientDB: {

package/src/services/file/{pglite.test.ts → client.test.ts}

@@ -6,7 +6,7 @@ import { files, globalFiles, users } from '@/database/schemas';
 import { clientS3Storage } from '@/services/file/ClientS3';
 import { UploadFileParams } from '@/types/files';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 
 const userId = 'file-user';
 

package/src/services/file/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/services/import/{pglite.test.ts → client.test.ts}

@@ -15,7 +15,7 @@ import {
 import { CURRENT_CONFIG_VERSION } from '@/migrations';
 import { ImportResults, ImporterEntryData } from '@/types/importer';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 
 const userId = 'test-user-id';
 const service = new ClientService(userId);

package/src/services/import/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/services/message/{pglite.test.ts → client.test.ts}

@@ -23,7 +23,7 @@ import {
   CreateMessageParams,
 } from '@/types/message';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 
 const userId = 'message-db';
 const sessionId = '1';

package/src/services/message/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/services/plugin/{pglite.test.ts → client.test.ts}

@@ -7,7 +7,7 @@ import { installedPlugins, users } from '@/database/schemas';
 import { LobeTool } from '@/types/tool';
 import { LobeToolCustomPlugin } from '@/types/tool/plugin';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { InstallPluginParams } from './type';
 
 // Mocking modules and functions

package/src/services/plugin/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/services/session/{pglite.test.ts → client.test.ts}

@@ -15,7 +15,7 @@ import {
 import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
 import { LobeAgentSession, LobeSessionType, SessionGroups } from '@/types/session';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 
 const userId = 'message-db';
 const sessionService = new ClientService(userId);

package/src/services/session/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/services/thread/client.ts

@@ -0,0 +1,57 @@
+import { INBOX_SESSION_ID } from '@/const/session';
+import { clientDB } from '@/database/client/db';
+import { MessageModel } from '@/database/server/models/message';
+import { ThreadModel } from '@/database/server/models/thread';
+import { BaseClientService } from '@/services/baseClientService';
+import { CreateMessageParams } from '@/types/message';
+import { CreateThreadParams, ThreadItem } from '@/types/topic';
+
+import { IThreadService } from './type';
+
+interface CreateThreadWithMessageParams extends CreateThreadParams {
+  message: CreateMessageParams;
+}
+export class ClientService extends BaseClientService implements IThreadService {
+  private get threadModel(): ThreadModel {
+    return new ThreadModel(clientDB as any, this.userId);
+  }
+  private get messageModel(): MessageModel {
+    return new MessageModel(clientDB as any, this.userId);
+  }
+
+  getThreads = async (topicId: string): Promise<ThreadItem[]> => {
+    return this.threadModel.queryByTopicId(topicId);
+  };
+
+  createThreadWithMessage = async (
+    input: CreateThreadWithMessageParams,
+  ): Promise<{ messageId: string; threadId: string }> => {
+    const thread = await this.threadModel.create({
+      parentThreadId: input.parentThreadId,
+      sourceMessageId: input.sourceMessageId,
+      title: input.message.content.slice(0, 20),
+      topicId: input.topicId,
+      type: input.type,
+    });
+
+    const message = await this.messageModel.create({
+      ...input.message,
+      sessionId: this.toDbSessionId(input.message.sessionId) as string,
+      threadId: thread?.id,
+    });
+
+    return { messageId: message?.id, threadId: thread?.id };
+  };
+
+  updateThread(id: string, data: Partial<ThreadItem>): Promise<any> {
+    return this.threadModel.update(id, data);
+  }
+
+  removeThread(id: string): Promise<any> {
+    return this.threadModel.delete(id);
+  }
+
+  private toDbSessionId(sessionId: string | undefined) {
+    return sessionId === INBOX_SESSION_ID ? null : sessionId;
+  }
+}

package/src/services/thread/index.ts

@@ -0,0 +1,5 @@
+import { ClientService } from './client';
+import { ServerService } from './server';
+
+export const threadService =
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : new ClientService();

package/src/services/{thread.ts → thread/server.ts}

@@ -3,10 +3,12 @@ import { lambdaClient } from '@/libs/trpc/client';
 import { CreateMessageParams } from '@/types/message';
 import { CreateThreadParams, ThreadItem } from '@/types/topic';
 
+import { IThreadService } from './type';
+
 interface CreateThreadWithMessageParams extends CreateThreadParams {
   message: CreateMessageParams;
 }
-export class ThreadService {
+export class ServerService implements IThreadService {
   getThreads = (topicId: string): Promise<ThreadItem[]> => {
     return lambdaClient.thread.getThreads.query({ topicId });
   };
@@ -21,34 +23,15 @@ export class ThreadService {
     });
   }
 
-  // createThread(params: CreateThreadParams): Promise<string> {
-  //   return lambdaClient.thread.createThread.mutate(params);
-  // }
-
   updateThread(id: string, data: Partial<ThreadItem>): Promise<any> {
     return lambdaClient.thread.updateThread.mutate({ id, value: data });
   }
 
-  //
   removeThread(id: string): Promise<any> {
     return lambdaClient.thread.removeThread.mutate({ id });
   }
-  //
-  // removeThreads(sessionId: string): Promise<any> {
-  //   return lambdaClient.thread.batchDeleteBySessionId.mutate({ id: this.toDbSessionId(sessionId) });
-  // }
-  //
-  // batchRemoveThreads(topics: string[]): Promise<any> {
-  //   return lambdaClient.thread.batchDelete.mutate({ ids: topics });
-  // }
-  //
-  // removeAllThread(): Promise<any> {
-  //   return lambdaClient.thread.removeAllThreads.mutate();
-  // }
 
   private toDbSessionId(sessionId: string | undefined) {
     return sessionId === INBOX_SESSION_ID ? null : sessionId;
   }
 }
-
-export const threadService = new ThreadService();

package/src/services/thread/type.ts

@@ -0,0 +1,20 @@
+/* eslint-disable typescript-sort-keys/interface */
+import { CreateMessageParams } from '@/types/message';
+import { CreateThreadParams, ThreadItem } from '@/types/topic';
+
+interface CreateThreadWithMessageParams extends CreateThreadParams {
+  message: CreateMessageParams;
+}
+
+export interface IThreadService {
+  getThreads(topicId: string): Promise<ThreadItem[]>;
+
+  createThreadWithMessage({
+    message,
+    ...params
+  }: CreateThreadWithMessageParams): Promise<{ messageId: string; threadId: string }>;
+
+  updateThread(id: string, data: Partial<ThreadItem>): Promise<any>;
+  //
+  removeThread(id: string): Promise<any>;
+}

package/src/services/topic/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/services/topic/pglite.test.ts

@@ -5,7 +5,7 @@ import { clientDB, initializeDB } from '@/database/client/db';
 import { sessions, topics, users } from '@/database/schemas';
 import { ChatTopic } from '@/types/topic';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 
 // Mock data
 const userId = 'topic-user-test';

package/src/services/user/{pglite.test.ts → client.test.ts}

@@ -7,7 +7,7 @@ import { userSettings, users } from '@/database/schemas';
 import { UserPreference } from '@/types/user';
 import { UserSettings } from '@/types/user/settings';
 
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 
 const mockUser = {
   avatar: 'avatar.png',

package/src/services/user/index.ts

@@ -1,5 +1,5 @@
 import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './pglite';
+import { ClientService } from './client';
 import { ServerService } from './server';
 
 const clientService =

package/src/store/chat/slices/thread/action.ts

@@ -6,7 +6,7 @@ import { StateCreator } from 'zustand/vanilla';
 
 import { chainSummaryTitle } from '@/chains/summaryTitle';
 import { LOADING_FLAT, THREAD_DRAFT_ID } from '@/const/message';
-import { isServerMode } from '@/const/version';
+import { isDeprecatedEdition } from '@/const/version';
 import { useClientDataSWR } from '@/libs/swr';
 import { chatService } from '@/services/chat';
 import { threadService } from '@/services/thread';
@@ -211,7 +211,7 @@ export const chatThreadMessage: StateCreator<
 
   useFetchThreads: (enable, topicId) =>
     useClientDataSWR<ThreadItem[]>(
-      enable && !!topicId && isServerMode ? [SWR_USE_FETCH_THREADS, topicId] : null,
+      enable && !!topicId && !isDeprecatedEdition ? [SWR_USE_FETCH_THREADS, topicId] : null,
       async ([, topicId]: [string, string]) => threadService.getThreads(topicId),
       {
         suspense: true,