assistant-ui

无头 React AI 对话框架，具有完整运行时层，与 useStream 桥接

智能体-UI 是一个无头 React UI 框架，用于 AI 对话。它提供了一个完整的运行时层——线程管理、消息分支、附件处理——通过 useExternalStoreRuntime 适配器连接到 @[useStream]。

克隆并运行 完整智能体-UI 示例，查看一个类似 Claude 风格的对话界面，该界面连接到带有 useExternalStoreRuntime 的 LangChain 智能体。

工作原理

1. 使用 @[useStream] 流式传输: 连接到您的智能体，获取反应式消息、加载状态以及提交/取消回调
2. 使用 useExternalStoreRuntime 适配: 将 stream.messages 桥接为 assistant-ui 的运行时格式，通过将 BaseMessage[] 转换为 ThreadMessageLike[]
3. 提供运行时: 将您的 UI 包裹在 AssistantRuntimeProvider 中，并渲染任何 assistant-ui 线程组件

安装

bun add @assistant-ui/react @assistant-ui/react-markdown

连接 useStream

useExternalStoreRuntime 适配器将 stream.messages 桥接为 assistant-ui 运行时。将其传递给 AssistantRuntimeProvider 并渲染任何线程组件：

import { useCallback, useMemo } from "react";
import {
  AssistantRuntimeProvider,
  useExternalStoreRuntime,
  type AppendMessage,
  type ThreadMessageLike,
} from "@assistant-ui/react";
import { useStream } from "@langchain/react";
import { Thread } from "@assistant-ui/react";

export function Chat() {
  const stream = useStream({
    apiUrl: "http://localhost:2024",
    assistantId: "claude",
  });

  const onNew = useCallback(
    async (message: AppendMessage) => {
      const text = message.content
        .filter((c) => c.type === "text")
        .map((c) => c.text)
        .join("");
      await stream.submit({ messages: [{ type: "human", content: text }] });
    },
    [stream],
  );

  // Convert LangChain messages to assistant-ui's ThreadMessageLike format
  const messages = useMemo(
    () => toThreadMessages(stream.messages),
    [stream.messages],
  );

  const runtime = useExternalStoreRuntime<ThreadMessageLike>({
    messages,
    onNew,
    onCancel: () => stream.stop(),
    convertMessage: (m) => m,
  });

  return (
    <AssistantRuntimeProvider runtime={runtime}>
      <Thread />
    </AssistantRuntimeProvider>
  );
}

转换消息

toThreadMessages 将 LangChain BaseMessage[] 映射为 assistant-ui 所期望的 ThreadMessageLike[] 格式。处理每种消息类型——人类、AI 和工具——并转换内容块、工具调用和推理 Token：

import { AIMessage, HumanMessage, ToolMessage, type BaseMessage } from "langchain";
import type { ThreadMessageLike } from "@assistant-ui/react";

export function toThreadMessages(messages: BaseMessage[]): ThreadMessageLike[] {
  const result: ThreadMessageLike[] = [];

  for (const msg of messages) {
    if (HumanMessage.isInstance(msg)) {
      result.push({
        role: "user",
        content: [{ type: "text", text: msg.text }],
      });
    } else if (AIMessage.isInstance(msg)) {
      const parts: ThreadMessageLike["content"] = [];

      // Reasoning tokens
      const reasoning = msg.contentBlocks.find((block) => block.type === "reasoning")?.reasoning;
      if (reasoning) parts.push({ type: "reasoning", text: reasoning });

      // Tool calls
      for (const tc of msg.tool_calls ?? []) {
        parts.push({
          type: "tool-call",
          toolCallId: tc.id ?? "",
          toolName: tc.name,
          args: tc.args,
        });
      }

      // Text response
      const text = msg.text;
      if (text) parts.push({ type: "text", text });

      result.push({ role: "assistant", content: parts });
    } else if (ToolMessage.isInstance(msg)) {
      // Attach tool results to the preceding assistant message
      const last = result[result.length - 1];
      if (last?.role === "assistant") {
        for (const part of last.content) {
          if (
            part.type === "tool-call" &&
            part.toolCallId === msg.tool_call_id
          ) {
            (part as { result?: string }).result = msg.text;
          }
        }
      }
    }
  }

  return result;
}

自定义线程 UI

<Thread /> 提供了一个完整的默认线程 UI，包括消息列表、作曲家和滚动管理。通过覆盖组件插槽来自定义单个部分：

import { Thread, ThreadMessages, Composer } from "@assistant-ui/react";

function CustomThread() {
  return (
    <Thread.Root>
      <ThreadMessages
        components={{
          UserMessage: MyUserMessage,
          AssistantMessage: MyAssistantMessage,
          ToolFallback: MyToolCard,
        }}
      />
      <Composer />
    </Thread.Root>
  );
}

最佳实践

• 记忆化消息转换: 将 toThreadMessages(stream.messages) 包裹在 useMemo 中，以避免在每次渲染时重新运行转换
• 处理附件: 使用 CompositeAttachmentAdapter 与 SimpleImageAttachmentAdapter 进行图片上传；通过自定义适配器扩展文件
• 使用分支: assistant-ui 通过 MessageBranch 提供内置的消息分支支持；当您需要 LangGraph 检查点分支时，将编辑与 useMessageMetadata 和 forkFrom 配合使用
• 线程持久化: 使用 onThreadId 持久化 threadId，并在页面加载时将其传递回 @[useStream]，以便 assistant-ui 重新连接到同一线程

---

官方原文：src/oss/langchain/frontend/integrations/assistant-ui.mdx · 非官方翻译，译文由 AI 生成，以英文原文为准。