fix(chat): 重新生成前撤销旧消息

2026-06-08 14:38:52 +08:00
parent f7cd5ebfa7
commit 40cc355fff
5 changed files with 45 additions and 7 deletions
@@ -56,6 +56,7 @@ type UseAgentChatSessionOptions = {
 type PromptRunOptions = {
  prompt: string;
  sessionIdOverride?: string;
+  regenerateFromMessageIndex?: number;
  preparedMessages?: Message[];
  userMessage?: Message;
  assistantMessage?: Message;
@@ -609,6 +610,7 @@ export const useAgentChatSession = ({
    async ({
      prompt: rawPrompt,
      sessionIdOverride,
+      regenerateFromMessageIndex,
      preparedMessages,
      userMessage,
      assistantMessage,
@@ -644,6 +646,7 @@ export const useAgentChatSession = ({
          sessionId: sessionIdOverride ?? sessionIdRef.current,
          model: getModel?.(),
          approvalMode: getApprovalMode?.(),
+          regenerateFromMessageIndex,
          signal: controller.signal,
          onEvent: (event) =>
            applyStreamEvent(event, {
@@ -991,6 +994,7 @@ export const useAgentChatSession = ({
      setMessages(nextMessages);
      await runPrompt({
        prompt: lastUserContent,
+        regenerateFromMessageIndex: lastUserIndex,
        preparedMessages: [
          ...nextMessages,
          nextUserMessage,