style: optimize code

package.json

@@ -238,8 +238,8 @@
     "@commitlint/config-conventional": "^19.2.2",
     "@commitlint/cz-commitlint": "^19.2.0",
     "@ianvs/prettier-plugin-sort-imports": "^4.3.1",
-    "@langchain/core": "0.2.16",
-    "@langchain/openai": "^0.2.3",
+    "@langchain/core": "0.2.17",
+    "@langchain/openai": "^0.2.4",
     "@types/fs-extra": "^11.0.4",
     "@types/global-agent": "^2.1.3",
     "@types/node": "^20.14.11",
@@ -286,7 +286,7 @@
   },
   "pnpm": {
     "overrides": {
-      "@langchain/core": "0.2.16"
+      "@langchain/core": "0.2.17"
     }
   },
   "commitlint": {

src/ai/model-providers/base.ts

@@ -72,25 +72,31 @@ export abstract class BaseModelProvider<Model extends BaseChatModel> {
     return prompt
   }
 
+  async getHistory(
+    sessionId: string,
+    appendHistoryMessages?: BaseMessage[]
+  ): Promise<InMemoryChatMessageHistory> {
+    if (BaseModelProvider.sessionIdHistoriesMap[sessionId] === undefined) {
+      const messageHistory = new InMemoryChatMessageHistory()
+
+      if (appendHistoryMessages && appendHistoryMessages.length > 0) {
+        await messageHistory.addMessages(appendHistoryMessages)
+      }
+
+      BaseModelProvider.sessionIdHistoriesMap[sessionId] = messageHistory
+    }
+
+    return BaseModelProvider.sessionIdHistoriesMap[sessionId]!
+  }
+
   createRunnableWithMessageHistory<Chunk extends AIMessageChunk>(
     chain: Runnable<any, Chunk, RunnableConfig>,
     historyMessages: BaseMessage[]
   ) {
     return new RunnableWithMessageHistory({
       runnable: chain,
-      getMessageHistory: async sessionId => {
-        if (BaseModelProvider.sessionIdHistoriesMap[sessionId] === undefined) {
-          const messageHistory = new InMemoryChatMessageHistory()
-
-          if (historyMessages.length > 0) {
-            await messageHistory.addMessages(historyMessages)
-          }
-
-          BaseModelProvider.sessionIdHistoriesMap[sessionId] = messageHistory
-        }
-
-        return BaseModelProvider.sessionIdHistoriesMap[sessionId]!
-      },
+      getMessageHistory: async sessionId =>
+        await this.getHistory(sessionId, historyMessages),
       inputMessagesKey: 'input',
       historyMessagesKey: 'history'
     })

src/commands/smart-paste.ts

@@ -6,7 +6,6 @@ import {
 import { getFileOrFoldersPromptInfo } from '@/file-utils/get-fs-prompt-info'
 import { insertTextAtSelection } from '@/file-utils/insert-text-at-selection'
 import { streamingCompletionWriter } from '@/file-utils/stream-completion-writer'
-import { logger } from '@/logger'
 import { getCurrentWorkspaceFolderEditor } from '@/utils'
 import type { BaseLanguageModelInput } from '@langchain/core/language_models/base'
 import type { RunnableConfig } from '@langchain/core/runnables'
@@ -43,11 +42,11 @@ const buildConvertPrompt = async ({
   const { promptFullContent: referenceFileContent } =
     await getFileOrFoldersPromptInfo(referencePaths, workspaceFolder.uri.fsPath)
 
-  logger.log('smart-paste', {
-    referencePaths,
-    currentFileRelativePath,
-    clipboardContent
-  })
+  // console.log('smart-paste', {
+  //   referencePaths,
+  //   currentFileRelativePath,
+  //   clipboardContent
+  // })
 
   const prompt = `
   I will provide the content of the current file, as well as the content of several most useful files related to the currently edited file, to you.

src/enable-global-proxy.ts

@@ -38,7 +38,7 @@ export const enableGlobalProxy = () => {
       setGlobalDispatcher(dispatcher)
     }
   } catch (err) {
-    logger.log('Failed to enable global proxy', err)
+    logger.warn('Failed to enable global proxy', err)
   }
 }