Add assistant-ui data agent frontend

2026-05-06 16:18:32 +08:00
parent d6a2359dc1
commit 7d4ae3e7ba
119 changed files with 14330 additions and 14420 deletions
@@ -0,0 +1,127 @@
+"use client";
+
+import type { ExportedMessageRepository } from "@assistant-ui/core";
+import { AssistantRuntimeProvider } from "@assistant-ui/react";
+import {
+	AssistantChatTransport,
+	useChatRuntime,
+} from "@assistant-ui/react-ai-sdk";
+import type { UIMessage } from "ai";
+import { useCallback, useMemo } from "react";
+import {
+	ActivityPanel,
+	ActivityProvider,
+} from "@/components/assistant-ui/activity-panel";
+import { Thread } from "@/components/assistant-ui/thread";
+import { ThreadListSidebar } from "@/components/assistant-ui/threadlist-sidebar";
+import { Separator } from "@/components/ui/separator";
+import {
+	SidebarInset,
+	SidebarProvider,
+	SidebarTrigger,
+} from "@/components/ui/sidebar";
+import { ClawSessionReplayProvider } from "@/lib/claw-session-replay";
+import { ClawAccountGate, useClawAccount } from "./claw-account-gate";
+import { ClawLlmSettings } from "./claw-llm-settings";
+
+export const Assistant = () => {
+	const { account, isLoading, setAccount } = useClawAccount();
+	const transport = useMemo(
+		() =>
+			new AssistantChatTransport({
+				api: "/api/chat",
+				prepareSendMessagesRequest: async (options) => {
+					const body = options.body as Record<string, unknown>;
+					const messages = options.messages as UIMessage[];
+					const selectedSessionId =
+						typeof window !== "undefined"
+							? window.localStorage.getItem("claw.activeSessionId")
+							: null;
+					const lastSessionId = getLastSessionId(messages);
+					const selectedResumeSessionId =
+						messages.length > 1 ? selectedSessionId : null;
+
+					return {
+						body: {
+							...body,
+							id: options.id,
+							messages,
+							resumeSessionId:
+								lastSessionId ?? selectedResumeSessionId ?? undefined,
+						},
+					};
+				},
+			}),
+		[],
+	);
+	const runtime = useChatRuntime({
+		transport,
+	});
+	const replaySession = useCallback(
+		(sessionId: string, repository: ExportedMessageRepository) => {
+			window.localStorage.setItem("claw.activeSessionId", sessionId);
+			runtime.thread.import(repository);
+		},
+		[runtime],
+	);
+
+	if (isLoading) {
+		return (
+			<div className="flex min-h-dvh items-center justify-center bg-background text-muted-foreground text-sm">
+				正在加载账号...
+			</div>
+		);
+	}
+
+	if (!account) {
+		return <ClawAccountGate onAccount={setAccount} />;
+	}
+
+	return (
+		<AssistantRuntimeProvider runtime={runtime}>
+			<ClawSessionReplayProvider value={{ replaySession }}>
+				<ActivityProvider>
+					<SidebarProvider>
+						<div className="flex h-dvh w-full pr-0.5">
+							<ThreadListSidebar
+								account={account}
+								onLogout={() => setAccount(null)}
+							/>
+							<SidebarInset>
+								<header className="flex h-16 shrink-0 items-center gap-2 border-b px-4">
+									<SidebarTrigger />
+									<Separator orientation="vertical" className="mr-2 h-4" />
+									<div className="min-w-0 flex-1">
+										<div className="truncate font-medium text-sm">新会话</div>
+										<div className="text-muted-foreground text-xs">
+											Claw Data Agent
+										</div>
+									</div>
+									<ClawLlmSettings />
+								</header>
+								<div className="flex min-h-0 flex-1 overflow-hidden">
+									<div className="min-w-0 flex-1 overflow-hidden">
+										<Thread />
+									</div>
+									<ActivityPanel />
+								</div>
+							</SidebarInset>
+						</div>
+					</SidebarProvider>
+				</ActivityProvider>
+			</ClawSessionReplayProvider>
+		</AssistantRuntimeProvider>
+	);
+};
+
+function getLastSessionId(messages: UIMessage[]) {
+	for (const message of [...messages].reverse()) {
+		if (message.role !== "assistant") continue;
+		const metadata = message.metadata;
+		if (metadata && typeof metadata === "object" && "sessionId" in metadata) {
+			const sessionId = (metadata as { sessionId?: unknown }).sessionId;
+			if (typeof sessionId === "string" && sessionId) return sessionId;
+		}
+	}
+	return undefined;
+}