AI前端性能优化完全指南:流式渲染、Token节约与缓存策略
现在Agent能查资料、能回答、能引用来源了。但上线后你会发现一个扎心的现实:用户觉得太慢了。
点击发送后等5秒才开始出字,一个复杂问题烧掉几万token,月底账单看得心惊肉跳。更要命的是,同样的问题用户问了10遍,后端就老老实实调了10次LLM——每次都花钱、每次都慢。
传统前端性能优化讲的是首屏渲染、代码分割、CDN缓存。AI应用的性能优化有自己的一套逻辑——快不快看流式、省不省看Token、值不值看缓存。
一、流式渲染:让用户“感觉”快
AI应用最影响体验的不是总耗时,而是首字时间(Time to First Token)——用户点击发送后多久看到第一个字。
非流式请求:用户等5~10秒,然后“啪”一下全出来
流式请求:200ms就开始出字,一个一个蹦出来,总时间可能一样长,但用户感知完全不同
这跟前端的SSR流式渲染是一个道理——不等数据全齐了再渲染,边到边渲染。
基础:用AI SDK实现流式
Vercel AI SDK把流式封装得非常简洁——后端用streamText,前端用useChat,开箱即用。
后端API Route:
// app/api/chat/route.ts
import { streamText } from "ai";
import { openai } from "@ai-sdk/openai";
export async function POST(req: Request) {
const { messages } = await req.json();
const result = streamText({
model: openai("gpt-4o"),
messages,
});
return result.toDataStreamResponse();
}前端组件:
"use client";
import { useChat } from "ai/react";
export default function Chat() {
const { messages, input, handleInputChange, handleSubmit, isLoading } = useChat();
return (
<div>
{messages.map((msg) => (
<div key={msg.id}>{msg.content}</div>
))}
<form onSubmit={handleSubmit}>
<input value={input} onChange={handleInputChange} />
<button type="submit" disabled={isLoading}>发送</button>
</form>
</div>
);
}useChat内部做了这些:
发起POST请求到/api/chat
拿到ReadableStream响应
用TextDecoder逐chunk解析SSE事件
实时更新messages状态,触发React重渲染
底层:手动处理SSE流
如果不用AI SDK,自己处理流式也不复杂。核心是ReadableStream + TextDecoder:
async function streamChat(messages: Message[]) {
const response = await fetch("/api/chat", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({ messages }),
});
const reader = response.body!.getReader();
const decoder = new TextDecoder();
let buffer = "";
while (true) {
const { done, value } = await reader.read();
if (done) break;
buffer += decoder.decode(value, { stream: true });
const lines = buffer.split("\n");
buffer = lines.pop() || "";
for (const line of lines) {
if (!line.startsWith("data: ")) continue;
const data = line.slice(6);
if (data === "[DONE]") return;
const parsed = JSON.parse(data);
const token = parsed.choices?.[0]?.delta?.content;
if (token) {
onToken(token); // 逐token回调,更新UI
}
}
}
}流式渲染的3个优化技巧
技巧1:防抖更新,减少重渲染
每个token都触发setState的话,一秒可能更新30~50次。用requestAnimationFrame合并更新:
let pendingText = "";
let rafId: number | null = null;
function onToken(token: string) {
pendingText += token;
if (!rafId) {
rafId = requestAnimationFrame(() => {
setText((prev) => prev + pendingText);
pendingText = "";
rafId = null;
});
}
}技巧2:Markdown增量解析
AI返回的内容通常是Markdown格式。如果每次都重新解析整个字符串,性能很差。推荐用增量解析:
import { marked } from "marked";
const renderedRef = useRef("");
const rawRef = useRef("");
function onToken(token: string) {
rawRef.current += token;
// 只在段落结束时重新解析,避免频繁DOM更新
if (token.includes("\n") || token.includes("```")) {
renderedRef.current = marked.parse(rawRef.current);
forceUpdate();
}
}技巧3:代码块延迟高亮
代码块在流式过程中是不完整的——等到```闭合后再做语法高亮,否则Prism.js/highlight.js会反复解析半成品代码:
function renderContent(content: string) {
const codeBlockRegex = /```(\w+)?\n([\s\S]*?)```/g;
return content.replace(codeBlockRegex, (match, lang, code) => {
// 只有完整的代码块才高亮
return highlightCode(code, lang);
});
// 未闭合的代码块保持原始文本
}中断生成:AbortController
用户不想等了,点“停止生成”——前端要能取消正在进行的流式请求:
const abortControllerRef = useRef<AbortController | null>(null);
async function handleSubmit() {
abortControllerRef.current = new AbortController();
const response = await fetch("/api/chat", {
method: "POST",
body: JSON.stringify({ messages }),
signal: abortControllerRef.current.signal,
});
// 处理流式响应...
}
function handleStop() {
abortControllerRef.current?.abort();
}AI SDK的useChat内置了stop()方法,更简洁:
const { messages, stop, isLoading } = useChat();
// 直接调用
<button onClick={stop} disabled={!isLoading}>停止生成</button>二、Token节约:花更少的钱办更多的事
LLM的计费方式是按token收费——输入token + 输出token。一个中型AI应用,日均几万次请求,Token成本很容易失控。
策略1:Prompt压缩
System Prompt是每次请求都要发的——它越长,每次请求的基础成本越高。
// ❌ 冗余的System Prompt(约500 token)
const systemPrompt = `
你是一个非常专业的前端技术助手。你的目标是帮助用户解决各种前端开发中遇到的问题。
你应该用通俗易懂的语言来解释技术概念,同时提供可运行的代码示例。
...(省略300字)
`;
// ✅ 精简的System Prompt(约80 token)
const systemPrompt = `前端技术助手。通俗解释,附可运行代码。不确定时如实说明。`;效果一样,但每次请求省了400+ token。假设日均10万次请求,一天就省了4000万token。
策略2:对话历史滑动窗口
多轮对话中,完整历史越来越长。只保留最近N轮 + 早期对话的摘要:
function trimMessages(messages: Message[], maxTokens: number = 4000) {
let totalTokens = 0;
const trimmed: Message[] = [];
// 从最新消息往前保留
for (let i = messages.length - 1; i >= 0; i--) {
const msgTokens = estimateTokens(messages[i].content);
if (totalTokens + msgTokens > maxTokens) break;
trimmed.unshift(messages[i]);
totalTokens += msgTokens;
}
// 如果有被截断的历史,加一条摘要
if (trimmed.length < messages.length) {
const droppedMessages = messages.slice(0, messages.length - trimmed.length);
const summary = await summarizeMessages(droppedMessages);
trimmed.unshift({
role: "system",
content: `之前的对话摘要:${summary}`,
});
}
return trimmed;
}
function estimateTokens(text: string): number {
// 粗略估算:中文1字≈2 token,英文1词≈1.3 token
return Math.ceil(text.length * 1.5);
}策略3:模型路由——简单问题用便宜模型
不是所有问题都需要GPT-4o。“今天星期几”用GPT-4o-mini就够了。
import { openai } from "@ai-sdk/openai";
import { generateText } from "ai";
async function routeToModel(query: string) {
// 先用便宜模型判断复杂度
const { text: complexity } = await generateText({
model: openai("gpt-4o-mini"),
prompt: `判断以下问题的复杂度,回答simple或complex:\n${query}`,
maxTokens: 10,
});
const model = complexity.trim() === "simple"
? openai("gpt-4o-mini") // $0.15 / 1M input tokens
: openai("gpt-4o"); // $2.50 / 1M input tokens
return model;
}实际效果:70%的请求走mini模型,成本直降80%。
策略4:控制输出长度
用户问“React是什么”,不需要输出2000字的长文。
const result = streamText({
model: openai("gpt-4o"),
messages,
maxTokens: 500, // 限制输出长度
});结合结构化输出,让LLM按固定格式回答,避免“废话”:
import { z } from "zod";
import { generateObject } from "ai";
const { object } = await generateObject({
model: openai("gpt-4o"),
schema: z.object({
answer: z.string().describe("简洁的回答,不超过200字"),
codeExample: z.string().optional().describe("代码示例(如有必要)"),
references: z.array(z.string()).optional().describe("参考资料链接"),
}),
prompt: userQuestion,
});成本对照表
| 模型 | 输入价格(每百万token) | 输出价格(每百万token) | 适合场景 |
|---|---|---|---|
| GPT-4o | $2.50 | $10.00 | 复杂推理、代码生成 |
| GPT-4o-mini | $0.15 | $0.60 | 简单问答、分类、路由 |
| Claude 3.5 Sonnet | $3.00 | $15.00 | 长文本、复杂分析 |
| Claude 3.5 Haiku | $0.80 | $4.00 | 快速响应、轻量任务 |
经验法则:先在mini上跑,效果不够再切大模型。别一上来就用最贵的。
三、缓存策略:同样的问题不要重复花钱
AI应用的一个特点:很多用户会问类似的问题。“React hooks怎么用”和“如何使用React hooks”本质是同一个问题——但没有缓存的话,每次都要调LLM。
第一层:精确匹配缓存
最简单——完全相同的输入直接返回缓存结果:
import { Redis } from "ioredis";
import { createHash } from "crypto";
const redis = new Redis();
function getCacheKey(messages: Message[]): string {
const content = JSON.stringify(messages);
return `ai:chat:${createHash("md5").update(content).digest("hex")}`;
}
async function cachedChat(messages: Message[]) {
const key = getCacheKey(messages);
// 查缓存
const cached = await redis.get(key);
if (cached) {
console.log("Cache hit!");
return JSON.parse(cached);
}
// 没命中,调LLM
const result = await generateText({
model: openai("gpt-4o"),
messages,
});
// 写缓存,TTL 1小时
await redis.setex(key, 3600, JSON.stringify(result.text));
return result.text;
}局限:必须完全一样才命中。“React hooks怎么用”和“react hooks怎么用”(大小写不同)就是两个key。
第二层:语义缓存(Semantic Cache)
用Embedding相似度匹配——“React hooks怎么用”和“如何使用React hooks”能命中同一条缓存。
import { OpenAIEmbeddings } from "@langchain/openai";
import { MemoryVectorStore } from "langchain/vectorstores/memory";
const embeddings = new OpenAIEmbeddings({
modelName: "text-embedding-3-small",
});
interface CacheEntry {
question: string;
answer: string;
embedding: number[];
}
const cacheStore = await MemoryVectorStore.fromTexts([], [], embeddings);
const cacheMap = new Map<string, string>();
async function semanticCachedChat(question: string) {
// 1. 语义检索:找相似问题
const results = await cacheStore.similaritySearchWithScore(question, 1);
if (results.length > 0) {
const [doc, score] = results[0];
// 相似度 > 0.92 认为是同一个问题
if (score > 0.92) {
console.log(`Semantic cache hit! (similarity: ${score.toFixed(3)})`);
return cacheMap.get(doc.pageContent);
}
}
// 2. 没命中,调LLM
const { text: answer } = await generateText({
model: openai("gpt-4o"),
prompt: question,
});
// 3. 写入缓存
await cacheStore.addDocuments([
{ pageContent: question, metadata: {} },
]);
cacheMap.set(question, answer);
return answer;
}语义缓存的命中率远高于精确缓存——实际测试中,语义缓存能把命中率从15%提升到60%以上。
第三层:Embedding缓存
RAG场景中,每次用户提问都要把问题转成Embedding向量。这个Embedding调用虽然便宜,但积少成多也是成本。
const embeddingCache = new Map<string, number[]>();
async function cachedEmbed(text: string): Promise<number[]> {
const key = text.trim().toLowerCase();
if (embeddingCache.has(key)) {
return embeddingCache.get(key)!;
}
const result = await embeddings.embedQuery(text);
embeddingCache.set(key, result);
return result;
}缓存失效策略
| 策略 | 做法 | 适合场景 |
|---|---|---|
| TTL | 固定时间过期(如1小时) | 信息时效性不强 |
| LRU | 淘汰最久未使用的 | 缓存空间有限 |
| 知识库更新时清除 | 文档变更后清缓存 | RAG场景 |
| 版本标记 | 模型/Prompt变了就失效 | 迭代频繁 |
四、UX优化:让AI应用“像人一样”
技术优化之外,UX设计对AI应用的体验影响巨大。
骨架屏 + 打字机效果
用户点击发送后,立即展示占位UI,避免“空白等待”:
function AIMessage({ content, isStreaming }: { content: string; isStreaming: boolean }) {
if (!content && isStreaming) {
return (
<div className="flex items-center gap-2 text-gray-400">
<span className="animate-pulse">●</span>
<span className="animate-pulse delay-100">●</span>
<span className="animate-pulse delay-200">●</span>
<span className="ml-2">思考中...</span>
</div>
);
}
return (
<div className="prose">
<ReactMarkdown>{content}</ReactMarkdown>
{isStreaming && <span className="animate-blink">▎</span>}
</div>
);
}重试与反馈
AI的回答不一定对——给用户“重新生成”和“反馈”的能力:
function MessageActions({ messageId, onRegenerate }: Props) {
const [feedback, setFeedback] = useState<"up" | "down" | null>(null);
return (
<div className="flex gap-2 mt-2 text-gray-400">
<button onClick={onRegenerate} className="hover:text-blue-500" title="重新生成">
🔄
</button>
<button
onClick={() => {
setFeedback("up");
trackFeedback(messageId, "positive");
}}
className={feedback === "up" ? "text-green-500" : "hover:text-green-500"}
>
👍
</button>
<button
onClick={() => {
setFeedback("down");
trackFeedback(messageId, "negative");
}}
className={feedback === "down" ? "text-red-500" : "hover:text-red-500"}
>
👎
</button>
</div>
);
}Token用量可视化
让用户知道“这次对话花了多少钱”,有助于控制使用:
function TokenUsage({ usage }: { usage: { input: number; output: number } }) {
const cost = (usage.input * 2.5 + usage.output * 10) / 1_000_000;
return (
<div className="text-xs text-gray-400 flex gap-4">
<span>输入: {usage.input.toLocaleString()} tokens</span>
<span>输出: {usage.output.toLocaleString()} tokens</span>
<span>≈ ${cost.toFixed(4)}</span>
</div>
);
}AI SDK返回的结果中包含usage字段:
const result = await generateText({
model: openai("gpt-4o"),
messages,
});
console.log(result.usage);
// { promptTokens: 150, completionTokens: 280, totalTokens: 430 }五、性能监控:建立AI应用的指标体系
传统前端用LCP、FID、CLS衡量性能。AI应用需要一套新指标。
核心指标
| 指标 | 含义 | 达标线 | 怎么测 |
|---|---|---|---|
| TTFT | Time to First Token,首字时间 | < 1s | 从请求发出到第一个token到达 |
| TPS | Tokens Per Second,生成速度 | > 30 | 输出token数 ÷ 生成时间 |
| 总延迟 | 从发送到完成的总时间 | < 10s | 端到端计时 |
| 缓存命中率 | 命中缓存的请求比例 | > 50% | cache hit / total requests |
| 单次成本 | 每次请求的token花费 | < $0.01 | (input + output) × unit price |
| 错误率 | 请求失败或超时的比例 | < 1% | error count / total |
前端埋点方案
function trackAIMetrics(metrics: {
ttft: number;
totalLatency: number;
inputTokens: number;
outputTokens: number;
model: string;
cached: boolean;
}) {
analytics.track("ai_request", {
...metrics,
tps: metrics.outputTokens / (metrics.totalLatency / 1000),
cost:
(metrics.inputTokens * getInputPrice(metrics.model) +
metrics.outputTokens * getOutputPrice(metrics.model)) /
1_000_000,
});
}
// 在useChat的回调中埋点
const { messages } = useChat({
onResponse(response) {
startTime.current = Date.now();
},
onFinish(message) {
trackAIMetrics({
ttft: firstTokenTime.current - startTime.current,
totalLatency: Date.now() - startTime.current,
inputTokens: message.usage?.promptTokens ?? 0,
outputTokens: message.usage?.completionTokens ?? 0,
model: "gpt-4o",
cached: response.headers.get("x-cache") === "HIT",
});
},
});六、实战清单:AI应用上线前的性能检查
| 检查项 | 具体做法 | 优先级 |
|---|---|---|
| ✅ 流式渲染 | 用streamText + useChat,确保首字<1s | P0 |
| ✅ 中断能力 | 实现“停止生成”按钮,用AbortController | P0 |
| ✅ 模型路由 | 简单问题走mini,复杂问题走大模型 | P1 |
| ✅ Prompt精简 | System Prompt < 200 token | P1 |
| ✅ 对话历史裁剪 | 滑动窗口 + 摘要,控制在4k token以内 | P1 |
| ✅ 精确缓存 | Redis缓存完全相同的请求 | P1 |
| ✅ 语义缓存 | Embedding相似度匹配相似问题 | P2 |
| ✅ 输出长度控制 | maxTokens + 结构化输出 | P2 |
| ✅ 错误重试 | 加重试逻辑 + 降级策略 | P1 |
| ✅ 性能监控 | TTFT、TPS、成本、缓存命中率全量埋点 | P1 |
| ✅ UX反馈 | 骨架屏、打字机效果、重新生成、反馈按钮 | P1 |
七、避坑指南
坑1:流式渲染导致Markdown闪烁
每个token都重新解析Markdown,导致页面闪烁——尤其是代码块和表格。
// ❌ 每个token都重新渲染
useEffect(() => {
setHtml(marked.parse(content));
}, [content]);
// ✅ 用useDeferredValue或节流
const deferredContent = useDeferredValue(content);
const html = useMemo(() => marked.parse(deferredContent), [deferredContent]);坑2:缓存了错误的回答
LLM有时会给出不正确的答案——如果缓存了,所有后续用户都会看到错误答案。
解法:
缓存前做质量检查(用LLM-as-Judge打分,低于阈值不缓存)
TTL不要太长(1~4小时)
提供“反馈”按钮,用户标记错误后自动清除缓存
坑3:模型路由判断本身就花Token
用GPT-4o-mini判断复杂度,虽然便宜,但也是一次额外调用。
解法:用规则优先,模型兜底
function quickRouting(query: string): "simple" | "complex" | "unknown" {
if (query.length < 20) return "simple";
if (/写.*代码|实现.*功能|重构|优化/.test(query)) return "complex";
return "unknown";
}
async function getModel(query: string) {
const quick = quickRouting(query);
if (quick !== "unknown") {
return quick === "simple" ? openai("gpt-4o-mini") : openai("gpt-4o");
}
return await routeWithLLM(query);
}坑4:流式请求没做错误处理
SSE连接断了、服务端500了——前端一片空白,用户不知道发生了什么。
const { messages, error, reload } = useChat({
onError(err) {
toast.error("AI服务暂时不可用,请稍后重试");
},
});
// 显示错误状态 + 重试按钮
{error && (
<div className="text-red-500 flex items-center gap-2">
<span>出错了:{error.message}</span>
<button onClick={reload} className="underline">重试</button>
</div>
)}坑5:忽略了并发控制
用户疯狂点击发送——前端同时发了10个请求,后端token烧到冒烟。
const isLoadingRef = useRef(false);
async function handleSubmit() {
if (isLoadingRef.current) return;
isLoadingRef.current = true;
try {
await sendMessage();
} finally {
isLoadingRef.current = false;
}
}AI SDK的useChat已经内置了这个——isLoading为true时不会重复发送。
总结
AI应用的前端性能优化,核心就是三板斧:
流式渲染——让用户感觉快
streamText + useChat实现开箱即用的流式
requestAnimationFrame合并更新减少重渲染
Markdown增量解析 + 代码块延迟高亮
AbortController实现中断生成
Token节约——让老板省心
Prompt压缩:System Prompt精简到<200 token
滑动窗口:只保留最近N轮 + 历史摘要
模型路由:简单问题走mini模型,成本降80%
输出控制:maxTokens + 结构化输出
缓存策略——让重复问题零成本
精确缓存:Redis存完全相同的请求
语义缓存:Embedding相似度匹配,命中率60%+
Embedding缓存:避免重复向量化
合理的TTL + 质量检查兜底
这三板斧下来,一个AI应用的体验和成本都能优化一个量级。
本文内容仅供个人学习、研究或参考使用,不构成任何形式的决策建议、专业指导或法律依据。未经授权,禁止任何单位或个人以商业售卖、虚假宣传、侵权传播等非学习研究目的使用本文内容。如需分享或转载,请保留原文来源信息,不得篡改、删减内容或侵犯相关权益。感谢您的理解与支持!