feat: add Replay Mode

This commit is contained in:
Li Xin 2025-04-24 21:51:08 +08:00
parent 3735706065
commit cb4b7b7495
10 changed files with 1035 additions and 40 deletions

File diff suppressed because one or more lines are too long

View File

@ -1,8 +1,15 @@
// Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
// SPDX-License-Identifier: MIT
import { useSearchParams } from "next/navigation";
import { useCallback, useRef, useState } from "react";
import {
Card,
CardDescription,
CardHeader,
CardTitle,
} from "~/components/ui/card";
import type { Option } from "~/core/messages";
import { sendMessage, useStore } from "~/core/store";
import { cn } from "~/lib/utils";
@ -10,10 +17,12 @@ import { cn } from "~/lib/utils";
import { ConversationStarter } from "./conversation-starter";
import { InputBox } from "./input-box";
import { MessageListView } from "./message-list-view";
import { RainbowText } from "./rainbow-text";
export function MessagesBlock({ className }: { className?: string }) {
const messageCount = useStore((state) => state.messageIds.length);
const responding = useStore((state) => state.responding);
const replaying = useSearchParams().get("replay") !== null;
const abortControllerRef = useRef<AbortController | null>(null);
const [feedback, setFeedback] = useState<{ option: Option } | null>(null);
const handleSend = useCallback(
@ -56,6 +65,7 @@ export function MessagesBlock({ className }: { className?: string }) {
onFeedback={handleFeedback}
onSendMessage={handleSend}
/>
{!replaying ? (
<div className="relative flex h-42 shrink-0 pb-4">
{!responding && messageCount === 0 && (
<ConversationStarter
@ -72,6 +82,22 @@ export function MessagesBlock({ className }: { className?: string }) {
onRemoveFeedback={handleRemoveFeedback}
/>
</div>
) : (
<div className="flex h-42 w-full items-center justify-center">
<Card className="w-full">
<CardHeader>
<CardTitle>
<RainbowText animated={responding}>Replay Mode</RainbowText>
</CardTitle>
<CardDescription>
<RainbowText animated={responding}>
DeerFlow is now replaying the conversation...
</RainbowText>
</CardDescription>
</CardHeader>
</Card>
</div>
)}
</div>
);
}

View File

@ -5,10 +5,11 @@
import { GithubOutlined } from "@ant-design/icons";
import Link from "next/link";
import { useMemo } from "react";
import { useEffect, useMemo } from "react";
import { Button } from "~/components/ui/button";
import { useStore } from "~/core/store";
import { useReplay } from "~/core/replay";
import { sendMessage, useStore } from "~/core/store";
import { cn } from "~/lib/utils";
import { Logo } from "./_components/logo";
@ -19,11 +20,17 @@ import { Tooltip } from "./_components/tooltip";
import { SettingsDialog } from "./_settings/dialogs/settings-dialog";
export default function HomePage() {
const { isReplay } = useReplay();
const openResearchId = useStore((state) => state.openResearchId);
const doubleColumnMode = useMemo(
() => openResearchId !== null,
[openResearchId],
);
useEffect(() => {
if (isReplay) {
void sendMessage();
}
}, [isReplay]);
return (
<div className="flex h-full w-full justify-center">
<header className="fixed top-0 left-0 flex h-12 w-full w-screen items-center justify-between px-4">

View File

@ -2,6 +2,7 @@
// SPDX-License-Identifier: MIT
import type { MCPServerMetadata } from "../mcp";
import { extractReplayIdFromURL } from "../replay/get-replay-id";
import { fetchStream } from "../sse";
import { sleep } from "../utils";
@ -28,8 +29,8 @@ export function chatStream(
},
options: { abortSignal?: AbortSignal } = {},
) {
if (location.search.includes("mock")) {
return chatStreamMock(userMessage, params, options);
if (location.search.includes("mock") || location.search.includes("replay=")) {
return chatReplayStream(userMessage, params, options);
}
return fetchStream<ChatEvent>(resolveServiceURL("chat/stream"), {
body: JSON.stringify({
@ -40,7 +41,7 @@ export function chatStream(
});
}
async function* chatStreamMock(
async function* chatReplayStream(
userMessage: string,
params: {
thread_id: string;
@ -57,30 +58,51 @@ async function* chatStreamMock(
},
options: { abortSignal?: AbortSignal } = {},
): AsyncIterable<ChatEvent> {
const mockFile =
const urlParams = new URLSearchParams(window.location.search);
let replayFilePath = "";
if (urlParams.has("mock")) {
replayFilePath =
params.interrupt_feedback === "accepted"
? "/mock-before-interrupt.txt"
: "/mock-after-interrupt.txt";
const res = await fetch(mockFile, {
? "/mock/before-interrupt.txt"
: "/mock/after-interrupt.txt";
} else {
const replayId = extractReplayIdFromURL();
if (replayId) {
replayFilePath = `/replay/${replayId}.txt`;
} else {
replayFilePath = "/mock/before-interrupt.txt";
}
}
const res = await fetch(replayFilePath, {
signal: options.abortSignal,
});
await sleep(500);
const text = await res.text();
const chunks = text.split("\n\n");
for (const chunk of chunks) {
const [eventRaw, dataRaw] = chunk.split("\n") as [string, string];
const [, event] = eventRaw.split("event: ", 2) as [string, string];
const [, data] = dataRaw.split("data: ", 2) as [string, string];
if (event === "message_chunk") {
await sleep(100);
} else if (event === "tool_call_result") {
await sleep(2000);
}
try {
yield {
const chatEvent = {
type: event,
data: JSON.parse(data),
} as ChatEvent;
if (chatEvent.type === "message_chunk") {
if (!chatEvent.data.finish_reason) {
await sleep(250 + Math.random() * 250);
}
} else if (chatEvent.type === "tool_call_result") {
await sleep(1500);
}
yield chatEvent;
if (chatEvent.type === "tool_call_result") {
await sleep(800);
} else if (chatEvent.type === "message_chunk") {
if (chatEvent.data.role === "user") {
await sleep(1000);
}
}
} catch (e) {
console.error(e);
}

View File

@ -0,0 +1,13 @@
// Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
// SPDX-License-Identifier: MIT
export function extractReplayIdFromURL() {
if (typeof window === "undefined") {
return null;
}
const urlParams = new URLSearchParams(window.location.search);
if (urlParams.has("replay")) {
return urlParams.get("replay");
}
return null;
}

View File

@ -0,0 +1,13 @@
// Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
// SPDX-License-Identifier: MIT
import { useMemo } from "react";
import { extractReplayIdFromURL } from "./get-replay-id";
export function useReplay() {
const replayId = useMemo(() => {
return extractReplayIdFromURL();
}, []);
return { isReplay: replayId != null, replayId };
}

View File

@ -0,0 +1,4 @@
// Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
// SPDX-License-Identifier: MIT
export * from "./hooks";

View File

@ -39,7 +39,7 @@ export const useStore = create<{
}));
export async function sendMessage(
content: string,
content?: string,
{
interruptFeedback,
}: {
@ -47,6 +47,7 @@ export async function sendMessage(
} = {},
options: { abortSignal?: AbortSignal } = {},
) {
if (content !== undefined) {
appendMessage({
id: nanoid(),
threadId: THREAD_ID,
@ -54,6 +55,7 @@ export async function sendMessage(
content: content,
contentChunks: [content],
});
}
setResponding(true);
try {
@ -104,7 +106,7 @@ export async function sendMessage(
};
}
const stream = chatStream(
content,
content ?? "[REPLAY]",
{
thread_id: THREAD_ID,
auto_accepted_plan: generalSettings.autoAcceptedPlan,