feat(chat): next step

src/common/panel/chat.ts

@@ -3,7 +3,7 @@ import * as vscode from "vscode";
 import { getUri } from "../utils/getUri";
 import { getNonce } from "../utils/getNonce";
 import { chat } from "../chat";
-import { ChatMessage } from "../prompt/promptChat";
+import { Chat, ChatMessage } from "../prompt/promptChat";
 import { state } from "../utils/state";
 
 export type MessageType =
@@ -20,32 +20,32 @@ export type MessageType =
       data: any;
     };
 
-type MessageFromWebview =
+type MessageToExtention =
   | {
       type: "send-message";
-      id: string;
       data: ChatMessage[];
     }
   | {
       type: "abort-generate";
       id: string;
     }
   | {
-      type: "get-chat-history";
-      id: string;
+      type: "get-chat";
       chatId: string;
     }
   | {
-      type: "save-chat-history";
-      id: string;
+      type: "save-chat";
       chatId: string;
-      data: ChatMessage[];
+      data: Chat;
     }
   | {
       type: "get-chats";
-      id: string;
     };
 
+type MessageFromWebview = MessageToExtention & {
+  id: string;
+};
+
 export class ChatPanel implements vscode.WebviewViewProvider {
   private disposables: Disposable[] = [];
   private webview: Webview | undefined;
@@ -137,14 +137,14 @@ export class ChatPanel implements vscode.WebviewViewProvider {
               chatMessage: message.data,
             });
             break;
-          case "get-chat-history":
-            await this.handleGetChatHistory({
+          case "get-chat":
+            await this.handleGetChat({
               id: message.id,
               chatId: message.chatId,
             });
             break;
-          case "save-chat-history":
-            await this.handleSaveChatHistory({
+          case "save-chat":
+            await this.handleSaveChat({
               id: message.id,
               chatId: message.chatId,
               history: message.data,
@@ -195,17 +195,8 @@ export class ChatPanel implements vscode.WebviewViewProvider {
     sendResponse("", true);
   }
 
-  private async handleGetChatHistory({
-    chatId,
-    id,
-  }: {
-    chatId: string;
-    id: string;
-  }) {
-    const sendResponse = (
-      messageToResponse: ChatMessage[] | null,
-      done: boolean
-    ) => {
+  private async handleGetChat({ chatId, id }: { chatId: string; id: string }) {
+    const sendResponse = (messageToResponse: Chat | null, done: boolean) => {
       this.postMessage({
         type: "e2w-response",
         id: id,
@@ -222,13 +213,13 @@ export class ChatPanel implements vscode.WebviewViewProvider {
     }
   }
 
-  private async handleSaveChatHistory({
+  private async handleSaveChat({
     chatId,
     history,
     id,
   }: {
     chatId: string;
-    history: ChatMessage[];
+    history: Chat;
     id: string;
   }) {
     await state.global.update(`chat-${chatId}`, history);

src/common/prompt/promptChat.ts

@@ -1,6 +1,14 @@
 export type ChatMessage = {
   role: string;
   content: string;
+  // chatMessageId: string;
+};
+
+export type Chat = {
+  messages: ChatMessage[];
+  chatId: string;
+  date: number;
+  title: string;
 };
 
 const promptBaseDefault = `You are an AI programming assistant, utilizing the DeepSeek Coder model, developed by DeepSeek Company, and you only answer questions related to computer science. For politically sensitive questions, security and privacy issues, and other non-computer science questions, you will refuse to answer.

src/common/utils/state.ts

@@ -1,6 +1,6 @@
 import * as vscode from "vscode";
 import type { Spec } from "../download";
-import { ChatMessage } from "../prompt/promptChat";
+import { Chat } from "../prompt/promptChat";
 
 const StateValues = {
   inlineSuggestModeAuto: true,
@@ -9,7 +9,7 @@ const StateValues = {
 type StateValuesType = {
   inlineSuggestModeAuto: boolean;
   serverSpec: Spec | null;
-  [key: `chat-${string}`]: ChatMessage[] | undefined;
+  [key: `chat-${string}`]: Chat | undefined;
 };
 
 class State {
@@ -34,14 +34,14 @@ class State {
     await this.state?.update(key, value);
   }
 
-  public getChats(): ChatMessage[][] {
+  public getChats(): Chat[] {
     const allKeys = (this.state?.keys() ||
       []) as unknown as (keyof StateValuesType)[];
 
     return allKeys
       .filter((key) => key.startsWith("chat-"))
       .map((key) => {
-        return this.get(key as `chat-${string}`) as ChatMessage[];
+        return this.get(key as `chat-${string}`) as Chat;
       });
   }
 }

webviews/src/hooks/useChat.ts

@@ -9,6 +9,13 @@ export type ChatMessage = {
   chatMessageId: string;
 };
 
+export type Chat = {
+  messages: ChatMessage[];
+  chatId: string;
+  date: number;
+  title: string;
+};
+
 export const useChat = (chatId?: string) => {
   const { chatMessages, setChatMessages } = useChatMessages();
 
@@ -21,9 +28,9 @@ export const useChat = (chatId?: string) => {
   useEffect(() => {
     const getChatHistory = async () => {
       if (chatId) {
-        const history = await vscode.getChatHistory(chatId);
-        if (history) {
-          setChatMessages(history);
+        const chat = await vscode.getChat(chatId);
+        if (chat) {
+          setChatMessages(chat.messages);
         }
       }
     };
@@ -58,7 +65,13 @@ export const useChat = (chatId?: string) => {
     }
     setChatMessages((chatHistoryLocal) => {
       (async () => {
-        await vscode.saveChatHistory(chatIdLocal, chatHistoryLocal);
+        const chat = {
+          chatId: chatIdLocal,
+          date: Date.now(),
+          messages: chatHistoryLocal,
+          title: "Chat with AI",
+        } satisfies Chat;
+        await vscode.saveChatHistory(chatIdLocal, chat);
       })();
       return chatHistoryLocal;
     });

webviews/src/utilities/vscode.ts

@@ -1,6 +1,6 @@
 import type { WebviewApi } from "vscode-webview";
 import { randomId } from "./messageId";
-import { ChatMessage } from "../hooks/useChat";
+import { Chat, ChatMessage } from "../hooks/useChat";
 import { Transform } from "./transformCallback2AsyncGenerator";
 
 export type MessageType =
@@ -27,13 +27,13 @@ type MessageToExtention =
       id: string;
     }
   | {
-      type: "get-chat-history";
+      type: "get-chat";
       chatId: string;
     }
   | {
-      type: "save-chat-history";
+      type: "save-chat";
       chatId: string;
-      data: ChatMessage[];
+      data: Chat;
     }
   | {
       type: "get-chats";
@@ -119,11 +119,11 @@ class VSCodeAPIWrapper {
     return transform.stream();
   }
 
-  public getChatHistory(chatId: string) {
-    return new Promise<ChatMessage[] | null>((resolve) => {
+  public getChat(chatId: string) {
+    return new Promise<Chat | null>((resolve) => {
       this.postMessageCallback(
         {
-          type: "get-chat-history",
+          type: "get-chat",
           chatId: chatId,
         },
         (message) => {
@@ -133,11 +133,11 @@ class VSCodeAPIWrapper {
     });
   }
 
-  public saveChatHistory(chatId: string, history: ChatMessage[]) {
+  public saveChatHistory(chatId: string, history: Chat) {
     return new Promise<void>((resolve) => {
       this.postMessageCallback(
         {
-          type: "save-chat-history",
+          type: "save-chat",
           chatId: chatId,
           data: history,
         },
@@ -149,7 +149,7 @@ class VSCodeAPIWrapper {
   }
 
   public getChats() {
-    return new Promise<ChatMessage[][]>((resolve) => {
+    return new Promise<Chat[]>((resolve) => {
       this.postMessageCallback({
         type: "get-chats",
       });