fix: fix entity bug

src/extension/webview-api/chat-context-processor/utils/convert-langchain-message-to-conversation.ts

@@ -1,17 +1,18 @@
 import type { MessageType } from '@langchain/core/messages'
+import { ConversationEntity } from '@shared/entities/conversation-entity'
 import type { Conversation } from '@shared/types/chat-context'
 import type {
   LangchainMessage,
   LangchainMessageContents
 } from '@shared/types/chat-context/langchain-message'
 import { convertToLangchainMessageContents } from '@shared/utils/convert-to-langchain-message-contents'
-import { getDefaultConversation } from '@shared/utils/get-default-conversation'
 
 export const convertLangchainMessageToConversation = (
   message: LangchainMessage
 ): Conversation => {
   const messageType = message.toDict().type as MessageType
-  const defaultConversation = getDefaultConversation(messageType)
+  const defaultConversation = new ConversationEntity({ role: messageType })
+    .entity
   const contents: LangchainMessageContents = convertToLangchainMessageContents(
     message.content
   )

src/extension/webview-api/controllers/chat-controller.ts

@@ -20,17 +20,4 @@ export class ChatController extends Controller {
       yield conversations
     }
   }
-
-  // sendMessage(req: { message: string }): Promise<string> {
-  //   return Promise.resolve(`Hi, bro, I'm response to: ${req.message}`)
-  // }
-
-  // async *streamChat(req: {
-  //   prompt: string
-  // }): AsyncGenerator<string, void, unknown> {
-  //   for (let i = 0; i < 5; i++) {
-  //     yield `Chunk ${i + 1} for prompt: ${req.prompt}`
-  //     await new Promise(resolve => setTimeout(resolve, 1000))
-  //   }
-  // }
 }

src/extension/webview-api/controllers/chat-session-controller.ts

@@ -1,9 +1,8 @@
 import { aidePaths } from '@extension/file-utils/paths'
 import { VsCodeFS } from '@extension/file-utils/vscode-fs'
 import { logger } from '@extension/logger'
-import type { ChatSession } from '@shared/entities'
+import { ChatContextEntity, type ChatSession } from '@shared/entities'
 import type { ChatContext, Conversation } from '@shared/types/chat-context'
-import { convertChatContextToChatSession } from '@shared/utils/convert-chat-context-to-chat-session'
 
 import { chatSessionsDB } from '../lowdb/chat-sessions-db'
 import { Controller } from '../types'
@@ -16,7 +15,7 @@ export class ChatSessionController extends Controller {
   }
 
   async createSession(req: { chatContext: ChatContext }): Promise<ChatSession> {
-    const chatSession = convertChatContextToChatSession(req.chatContext)
+    const chatSession = new ChatContextEntity(req.chatContext).toChatSession()
     const now = new Date().getTime()
     const session = await chatSessionsDB.add({
       ...chatSession,
@@ -48,7 +47,7 @@ export class ChatSessionController extends Controller {
   async updateSession(req: { chatContext: ChatContext }): Promise<void> {
     const now = new Date().getTime()
     const session = await chatSessionsDB.update(req.chatContext.id, {
-      ...convertChatContextToChatSession(req.chatContext),
+      ...new ChatContextEntity(req.chatContext).toChatSession(),
       updatedAt: now
     })
 
@@ -65,7 +64,7 @@ export class ChatSessionController extends Controller {
   }): Promise<void> {
     const now = new Date().getTime()
     const session = await chatSessionsDB.createOrUpdate({
-      ...convertChatContextToChatSession(req.chatContext),
+      ...new ChatContextEntity(req.chatContext).toChatSession(),
       updatedAt: now
     })
 

src/extension/webview-api/lowdb/ai-model-db.ts

@@ -8,8 +8,7 @@ class AIModelDB extends BaseDB<AIModel> {
   static readonly schemaVersion = 1
 
   constructor() {
-    // Use entity's defaults
-    const defaults = new AIModelEntity().getDefaults()
+    const defaults = new AIModelEntity().entity
 
     super(
       path.join(aidePaths.getGlobalLowdbPath(), 'ai-models.json'),

src/extension/webview-api/lowdb/ai-provider-db.ts

@@ -31,16 +31,15 @@ const findNewModel = async (
         new AIModelEntity({
           name,
           providerOrBaseUrl
-        })
+        }).entity
     )
 }
 
 class AIProviderDB extends BaseDB<AIProvider> {
   static readonly schemaVersion = 1
 
   constructor() {
-    // Use entity's defaults
-    const defaults = new AIProviderEntity().getDefaults()
+    const defaults = new AIProviderEntity().entity
 
     super(
       path.join(aidePaths.getGlobalLowdbPath(), 'ai-providers.json'),

src/extension/webview-api/lowdb/chat-sessions-db.ts

@@ -11,8 +11,7 @@ class ChatSessionsDB extends BaseDB<ChatSession> {
   static readonly schemaVersion = 1
 
   constructor() {
-    // Use entity's defaults
-    const defaults = new ChatSessionEntity().getDefaults()
+    const defaults = new ChatSessionEntity().entity
 
     super(
       path.join(aidePaths.getWorkspaceLowdbPath(), 'sessions.json'),

src/extension/webview-api/lowdb/doc-sites-db.ts

@@ -8,7 +8,7 @@ class DocSitesDB extends BaseDB<DocSite> {
   static readonly schemaVersion = 1
 
   constructor() {
-    const defaults = new DocSiteEntity().getDefaults()
+    const defaults = new DocSiteEntity().entity
     super(
       path.join(aidePaths.getGlobalLowdbPath(), 'doc-sites.json'),
       defaults,
@@ -21,7 +21,7 @@ class DocSitesDB extends BaseDB<DocSite> {
       id?: string
     }
   ): Promise<DocSite> {
-    const docSite = new DocSiteEntity(item)
+    const docSite = new DocSiteEntity(item).entity
     return super.add(docSite)
   }
 
@@ -30,7 +30,7 @@ class DocSitesDB extends BaseDB<DocSite> {
       id?: string
     })[]
   ): Promise<DocSite[]> {
-    const docSites = items.map(item => new DocSiteEntity(item))
+    const docSites = items.map(item => new DocSiteEntity(item).entity)
     return super.batchAdd(docSites)
   }
 

src/extension/webview-api/lowdb/settings-db.ts

@@ -15,7 +15,7 @@ class SettingsDB extends BaseDB<Settings> {
   static readonly schemaVersion = 1
 
   constructor(filePath: string) {
-    const defaults = new SettingsEntity().getDefaults()
+    const defaults = new SettingsEntity().entity
     super(filePath, defaults, SettingsDB.schemaVersion)
   }
 
@@ -39,7 +39,8 @@ class SettingsDB extends BaseDB<Settings> {
       value,
       category: config.category,
       updatedAt: Date.now()
-    })
+    }).entity
+
     return this.add(setting)
   }
 

src/shared/entities/ai-model-entity.ts

@@ -12,20 +12,8 @@ export interface AIModel extends IBaseEntity {
   toolsCallSupport: AIModelSupport
 }
 
-export class AIModelEntity extends BaseEntity<AIModel> implements AIModel {
-  id!: string
-
-  providerOrBaseUrl!: AIProviderType | string
-
-  name!: string
-
-  imageSupport!: AIModelSupport
-
-  audioSupport!: AIModelSupport
-
-  toolsCallSupport!: AIModelSupport
-
-  getDefaults(): AIModel {
+export class AIModelEntity extends BaseEntity<AIModel> {
+  protected getDefaults(): AIModel {
     return {
       id: uuidv4(),
       providerOrBaseUrl: AIProviderType.OpenAI,

src/shared/entities/ai-provider-entity.ts

@@ -12,27 +12,8 @@ export interface AIProvider extends IBaseEntity {
   manualModels: string[]
 }
 
-export class AIProviderEntity
-  extends BaseEntity<AIProvider>
-  implements AIProvider
-{
-  id!: string
-
-  name!: string
-
-  type!: AIProviderType
-
-  order!: number
-
-  extraFields!: Record<string, string>
-
-  allowRealTimeModels!: boolean
-
-  realTimeModels!: string[]
-
-  manualModels!: string[]
-
-  getDefaults(): AIProvider {
+export class AIProviderEntity extends BaseEntity<AIProvider> {
+  protected getDefaults(): AIProvider {
     return {
       id: uuidv4(),
       name: 'unknown',

src/shared/entities/base-entity.ts

@@ -4,9 +4,11 @@ export interface IBaseEntity {
 }
 
 export abstract class BaseEntity<T extends IBaseEntity> {
+  entity: T
+
   constructor(data?: Partial<T>) {
-    Object.assign(this, this.getDefaults(), data ?? {})
+    this.entity = { ...this.getDefaults(), ...(data ?? {}) }
   }
 
-  abstract getDefaults(): Partial<T>
+  protected abstract getDefaults(): T
 }

src/shared/entities/chat-context-entity.ts

@@ -0,0 +1,65 @@
+import { ChatContextType, type Conversation } from '@shared/types/chat-context'
+import type { SettingsContext } from '@shared/types/chat-context/settings-context'
+import { v4 as uuidv4 } from 'uuid'
+
+import { BaseEntity, type IBaseEntity } from './base-entity'
+import { ChatSessionEntity, type ChatSession } from './chat-session-entity'
+
+export interface ChatContext extends IBaseEntity {
+  type: ChatContextType
+  createdAt: number
+  updatedAt: number
+  conversations: Conversation[]
+  settings: SettingsContext
+}
+
+export class ChatContextEntity extends BaseEntity<ChatContext> {
+  getDefaults(): ChatContext {
+    const now = Date.now()
+    return {
+      id: uuidv4(),
+      type: ChatContextType.Chat,
+      createdAt: now,
+      updatedAt: now,
+      conversations: [],
+      settings: {
+        allowLongFileScan: false,
+        explicitContext: '总是用中文回复',
+        fastApplyModelName: 'gpt-4o-mini',
+        modelName: 'gpt-4o',
+        useFastApply: true
+      }
+    }
+  }
+
+  toChatSession(): ChatSession {
+    const { entity } = this
+
+    return new ChatSessionEntity({
+      id: entity.id,
+      title: this.getTitleFromConversations(entity.conversations),
+      type: entity.type,
+      createdAt: entity.createdAt,
+      updatedAt: entity.updatedAt
+    }).entity
+  }
+
+  private getTitleFromConversations = (
+    conversations: Conversation[],
+    defaultTitle = 'New Chat'
+  ) => {
+    let firstHumanMessageText = ''
+
+    conversations
+      .filter(conversation => conversation.role === 'human')
+      .forEach(conversation =>
+        conversation.contents.forEach(content => {
+          if (content.type === 'text' && !firstHumanMessageText) {
+            firstHumanMessageText = content.text
+          }
+        })
+      )
+
+    return firstHumanMessageText || defaultTitle
+  }
+}

src/shared/entities/chat-session-entity.ts

@@ -10,21 +10,8 @@ export interface ChatSession extends IBaseEntity {
   title: string
 }
 
-export class ChatSessionEntity
-  extends BaseEntity<ChatSession>
-  implements ChatSession
-{
-  id!: string
-
-  type!: ChatContextType
-
-  createdAt!: number
-
-  updatedAt!: number
-
-  title!: string
-
-  getDefaults(): ChatSession {
+export class ChatSessionEntity extends BaseEntity<ChatSession> {
+  protected getDefaults(): ChatSession {
     const now = Date.now()
     return {
       id: uuidv4(),

src/shared/entities/conversation-entity.ts

@@ -0,0 +1,26 @@
+import type { MessageType } from '@langchain/core/messages'
+import type { PluginState } from '@shared/plugins/base/types'
+import type { LangchainMessageContents } from '@shared/types/chat-context/langchain-message'
+import { v4 as uuidv4 } from 'uuid'
+
+import { BaseEntity, type IBaseEntity } from './base-entity'
+
+export interface Conversation extends IBaseEntity {
+  createdAt: number
+  role: MessageType
+  contents: LangchainMessageContents
+  richText?: string
+  pluginStates: Record<string, PluginState>
+}
+
+export class ConversationEntity extends BaseEntity<Conversation> {
+  protected getDefaults(): Conversation {
+    return {
+      id: uuidv4(),
+      createdAt: Date.now(),
+      role: 'human',
+      contents: [],
+      pluginStates: {}
+    }
+  }
+}

src/shared/entities/doc-site-entity.ts

@@ -9,18 +9,8 @@ export interface DocSite extends IBaseEntity {
   isIndexed: boolean
 }
 
-export class DocSiteEntity extends BaseEntity<DocSite> implements DocSite {
-  id!: string
-
-  name!: string
-
-  url!: string
-
-  isCrawled!: boolean
-
-  isIndexed!: boolean
-
-  getDefaults(): DocSite {
+export class DocSiteEntity extends BaseEntity<DocSite> {
+  protected getDefaults(): DocSite {
     return {
       id: uuidv4(),
       name: 'unknown',

src/shared/entities/index.ts

@@ -1,6 +1,8 @@
 export * from './ai-model-entity'
 export * from './ai-provider-entity'
 export * from './base-entity'
+export * from './chat-context-entity'
 export * from './chat-session-entity'
+export * from './conversation-entity'
 export * from './doc-site-entity'
 export * from './settings-entity'

src/shared/entities/settings-entity.ts

@@ -9,18 +9,8 @@ export interface Settings extends IBaseEntity {
   updatedAt: number
 }
 
-export class SettingsEntity extends BaseEntity<Settings> implements Settings {
-  id!: string
-
-  key!: SettingKey
-
-  value!: SettingValue<SettingKey>
-
-  category!: SettingCategory
-
-  updatedAt!: number
-
-  getDefaults(): Settings {
+export class SettingsEntity extends BaseEntity<Settings> {
+  protected getDefaults(): Settings {
     return {
       id: uuidv4(),
       key: 'unknown' as SettingKey,