feat(chat): 添加继续生成的功能支持

- 在 ChatProvider 中新增 continueGeneration 方法以处理续写请求
- 修改状态管理，移除对继续提示的已关闭本地存储逻辑
- 优化会话状态显示，只在中断时展示继续提示
- 为用户消息生成唯一 ID，保证消息唯一性
- 在 App 组件中将 ContinuePrompt 的 onContinue 回调绑定到 continueGeneration
- 在路由逻辑中排除 "/continue" 指令的用户消息展示，避免重复发送请求

Author: hqwlkj

packages/vscode-ide-companion/src/router.ts

@@ -143,13 +143,15 @@ export const appRouter = router({
     }
 
     const displayPrompt = prompt || (normalizedImages.length > 0 ? "粘贴的图像" : "");
+    const promptTrimmed = prompt.trim();
     const isPermissionContinue =
-      prompt === "/continue" &&
+      promptTrimmed === "/continue" &&
       normalizedImages.length === 0 &&
       ((permissions?.length ?? 0) > 0 || (alwaysAllows?.length ?? 0) > 0);
+    const isPlainContinue = promptTrimmed === "/continue" && normalizedImages.length === 0;
 
-    // Show user message in webview
-    if (displayPrompt && !isPermissionContinue) {
+    // Show user message in webview (skip for /continue commands)
+    if (displayPrompt && !isPermissionContinue && !isPlainContinue) {
       ctx.postMessage({ type: "userMessage", content: displayPrompt });
     }
 

packages/vscode-ide-companion/src/webview/App.tsx

@@ -49,7 +49,7 @@ export default function App() {
         onInterrupt={actions.interrupt}
       />
       {state.showContinuePrompt && (
-        <ContinuePrompt onContinue={actions.dismissContinuePrompt} onDismiss={actions.dismissContinuePrompt} />
+        <ContinuePrompt onContinue={actions.continueGeneration} onDismiss={actions.dismissContinuePrompt} />
       )}
       {state.loading && (
         <ThinkingLiveBubble

packages/vscode-ide-companion/src/webview/context/ChatProvider.tsx

@@ -41,6 +41,7 @@ interface ChatContextValue {
     renameSession: (sessionId: string, summary: string) => Promise<void>;
     deleteSession: (sessionId: string) => Promise<void>;
     dismissContinuePrompt: () => void;
+    continueGeneration: () => Promise<void>;
   };
 }
 
@@ -93,20 +94,12 @@ function appReducer(state: AppState, action: AppAction): AppState {
         permissionPromptState: null,
         pendingPermissionReply: null,
         loading: action.status === "processing" || action.status === "pending",
-        showContinuePrompt:
-          action.status === "interrupted"
-            ? localStorage.getItem(`deepcode:continuePromptDismissed:${action.sessionId}`) !== "1"
-            : false,
+        showContinuePrompt: action.status === "interrupted",
       };
     case "SET_SESSIONS":
       return { ...state, sessions: action.sessions };
     case "SESSION_STATUS": {
       const isInterrupted = action.status === "interrupted";
-      const sessionId = action.sessionId || state.activeSessionId;
-      const dismissed =
-        isInterrupted && sessionId
-          ? localStorage.getItem(`deepcode:continuePromptDismissed:${sessionId}`) === "1"
-          : true;
 
       return {
         ...state,
@@ -115,7 +108,7 @@ function appReducer(state: AppState, action: AppAction): AppState {
         processes: action.processes ?? state.processes,
         tokenTelemetry: action.tokenTelemetry ?? state.tokenTelemetry,
         loading: action.status === "processing" || action.status === "pending",
-        showContinuePrompt: isInterrupted && !dismissed,
+        showContinuePrompt: isInterrupted,
       };
     }
     case "LLM_STREAM_PROGRESS":
@@ -124,7 +117,12 @@ function appReducer(state: AppState, action: AppAction): AppState {
       }
       return { ...state, llmStreamProgress: action.progress };
     case "USER_MESSAGE": {
-      const newMsg: SessionMessage = { role: "user", content: action.content, meta: action.meta };
+      const newMsg: SessionMessage = {
+        id: crypto.randomUUID(),
+        role: "user",
+        content: action.content,
+        meta: action.meta,
+      };
       return {
         ...state,
         messages: [...state.messages, newMsg],
@@ -399,11 +397,17 @@ export function ChatProvider({ children }: ChatProviderProps) {
   );
 
   const dismissContinuePrompt = useCallback(() => {
-    if (state.activeSessionId) {
-      localStorage.setItem(`deepcode:continuePromptDismissed:${state.activeSessionId}`, "1");
-    }
     dispatch({ type: "DISMISS_CONTINUE_PROMPT" });
-  }, [state.activeSessionId]);
+  }, []);
+
+  const continueGeneration = useCallback(async () => {
+    dispatch({ type: "DISMISS_CONTINUE_PROMPT" });
+    try {
+      await chatService.sendPrompt({ prompt: "/continue" });
+    } catch (err) {
+      console.error("[ChatProvider] continueGeneration failed:", err);
+    }
+  }, []);
 
   const contextValue: ChatContextValue = {
     state,
@@ -419,6 +423,7 @@ export function ChatProvider({ children }: ChatProviderProps) {
       renameSession,
       deleteSession,
       dismissContinuePrompt,
+      continueGeneration,
     },
   };