ai问片移除think,标题可点击

This commit is contained in:
mtvpls
2026-03-07 12:32:26 +08:00
parent 6481772777
commit bc25791fd9
2 changed files with 99 additions and 13 deletions

View File

@@ -74,6 +74,9 @@ function transformToSSE(
return new ReadableStream({
async start(controller) {
let buffer = ''; // 缓冲区,用于保存不完整的行
let contentBuffer = ''; // 累积的内容用于处理跨chunk的thinking标签
let inThinkingBlock = false; // 是否在thinking块内
try {
while (true) {
const { done, value } = await reader.read();
@@ -122,9 +125,32 @@ function transformToSSE(
}
if (text) {
controller.enqueue(
new TextEncoder().encode(`data: ${JSON.stringify({ text })}\n\n`)
);
// 累积内容并处理thinking标签
contentBuffer += text;
// 检查是否进入thinking块
if (contentBuffer.includes('<think>')) {
inThinkingBlock = true;
}
// 检查是否退出thinking块
if (inThinkingBlock && contentBuffer.includes('</think>')) {
// 移除thinking块内容
contentBuffer = contentBuffer.replace(/<think>[\s\S]*?<\/think>/g, '');
inThinkingBlock = false;
}
// 只有在不在thinking块内时才输出内容
if (!inThinkingBlock) {
// 输出非thinking部分的内容
const outputText = contentBuffer;
if (outputText) {
controller.enqueue(
new TextEncoder().encode(`data: ${JSON.stringify({ text: outputText })}\n\n`)
);
contentBuffer = ''; // 清空已输出的内容
}
}
}
} catch (e) {
// 只在非空数据解析失败时打印错误
@@ -153,9 +179,14 @@ function transformToSSE(
text = json.choices?.[0]?.delta?.content || '';
}
if (text) {
controller.enqueue(
new TextEncoder().encode(`data: ${JSON.stringify({ text })}\n\n`)
);
contentBuffer += text;
// 最后清理一次thinking标签
contentBuffer = contentBuffer.replace(/<think>[\s\S]*?<\/think>/g, '');
if (contentBuffer) {
controller.enqueue(
new TextEncoder().encode(`data: ${JSON.stringify({ text: contentBuffer })}\n\n`)
);
}
}
} catch (e) {
console.error('Parse final buffer error:', e);
@@ -298,7 +329,10 @@ export async function POST(request: NextRequest) {
// 非流式响应等待完整响应后返回JSON
const response = result as Response;
const data = await response.json();
const content = data.choices?.[0]?.message?.content || '';
let content = data.choices?.[0]?.message?.content || '';
// 移除thinking标签内容
content = content.replace(/<think>[\s\S]*?<\/think>/g, '');
return NextResponse.json({ content });
}

View File

@@ -2,6 +2,8 @@
'use client';
import { Bot, Loader2, Send, Sparkles, Trash2,X } from 'lucide-react';
import Link from 'next/link';
import { usePathname } from 'next/navigation';
import React, { useEffect, useMemo, useRef, useState } from 'react';
import { createPortal } from 'react-dom';
import ReactMarkdown from 'react-markdown';
@@ -33,6 +35,8 @@ export default function AIChatPanel({
useDrawer = false,
drawerWidth = 'w-full md:w-[25%]',
}: AIChatPanelProps) {
const pathname = usePathname();
// 使用 useMemo 稳定 storage key只在实际内容变化时才改变
const storageKey = useMemo(() => {
if (context?.title) {
@@ -53,6 +57,14 @@ export default function AIChatPanel({
const abortControllerRef = useRef<AbortController | null>(null);
const hasLoadedRef = useRef(false);
// 将《》包裹的影视名称转换为链接
const convertTitleToLink = (content: string): string => {
return content.replace(/《([^》]+)》/g, (match, title) => {
const encodedTitle = encodeURIComponent(title);
return `[《${title}》](/play?title=${encodedTitle})`;
});
};
// 自动滚动到底部
const scrollToBottom = () => {
messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
@@ -424,9 +436,29 @@ export default function AIChatPanel({
{message.content}
</p>
) : (
<div className='prose prose-sm max-w-none dark:prose-invert prose-p:my-2 prose-p:leading-relaxed prose-pre:bg-gray-800 prose-pre:text-gray-100 dark:prose-pre:bg-gray-900 prose-code:text-purple-600 dark:prose-code:text-purple-400 prose-code:bg-purple-50 dark:prose-code:bg-purple-900/20 prose-code:px-1 prose-code:py-0.5 prose-code:rounded prose-code:before:content-none prose-code:after:content-none prose-a:text-blue-600 dark:prose-a:text-blue-400 prose-strong:text-gray-900 dark:prose-strong:text-white prose-ul:my-2 prose-ol:my-2 prose-li:my-1'>
<ReactMarkdown remarkPlugins={[remarkGfm as any]}>
{message.content}
<div className='prose prose-sm max-w-none dark:prose-invert prose-p:my-2 prose-p:leading-relaxed prose-pre:bg-gray-800 prose-pre:text-gray-100 dark:prose-pre:bg-gray-900 prose-code:text-purple-600 dark:prose-code:text-purple-400 prose-code:bg-purple-50 dark:prose-code:bg-purple-900/20 prose-code:px-1 prose-code:py-0.5 prose-code:rounded prose-code:before:content-none prose-code:after:content-none prose-a:text-inherit dark:prose-a:text-inherit prose-a:no-underline hover:prose-a:underline prose-strong:text-gray-900 dark:prose-strong:text-white prose-ul:my-2 prose-ol:my-2 prose-li:my-1'>
<ReactMarkdown
remarkPlugins={[remarkGfm as any]}
components={{
a: ({ node, href, children, ...props }) => {
// 如果是内部链接(以 / 开头),使用 Next.js Link
if (href?.startsWith('/')) {
// 如果当前在 /play 页面且链接也是 /play不做处理返回纯文本
if (pathname === '/play' && href.startsWith('/play')) {
return <span>{children}</span>;
}
return (
<Link href={href} {...props}>
{children}
</Link>
);
}
// 外部链接使用普通 a 标签
return <a href={href} target="_blank" rel="noopener noreferrer" {...props}>{children}</a>;
}
}}
>
{convertTitleToLink(message.content)}
</ReactMarkdown>
</div>
)}
@@ -610,9 +642,29 @@ export default function AIChatPanel({
{message.content}
</p>
) : (
<div className='prose prose-sm max-w-none dark:prose-invert prose-p:my-2 prose-p:leading-relaxed prose-pre:bg-gray-800 prose-pre:text-gray-100 dark:prose-pre:bg-gray-900 prose-code:text-purple-600 dark:prose-code:text-purple-400 prose-code:bg-purple-50 dark:prose-code:bg-purple-900/20 prose-code:px-1 prose-code:py-0.5 prose-code:rounded prose-code:before:content-none prose-code:after:content-none prose-a:text-blue-600 dark:prose-a:text-blue-400 prose-strong:text-gray-900 dark:prose-strong:text-white prose-ul:my-2 prose-ol:my-2 prose-li:my-1'>
<ReactMarkdown remarkPlugins={[remarkGfm as any]}>
{message.content}
<div className='prose prose-sm max-w-none dark:prose-invert prose-p:my-2 prose-p:leading-relaxed prose-pre:bg-gray-800 prose-pre:text-gray-100 dark:prose-pre:bg-gray-900 prose-code:text-purple-600 dark:prose-code:text-purple-400 prose-code:bg-purple-50 dark:prose-code:bg-purple-900/20 prose-code:px-1 prose-code:py-0.5 prose-code:rounded prose-code:before:content-none prose-code:after:content-none prose-a:text-inherit dark:prose-a:text-inherit prose-a:no-underline hover:prose-a:underline prose-strong:text-gray-900 dark:prose-strong:text-white prose-ul:my-2 prose-ol:my-2 prose-li:my-1'>
<ReactMarkdown
remarkPlugins={[remarkGfm as any]}
components={{
a: ({ node, href, children, ...props }) => {
// 如果是内部链接(以 / 开头),使用 Next.js Link
if (href?.startsWith('/')) {
// 如果当前在 /play 页面且链接也是 /play不做处理返回纯文本
if (pathname === '/play' && href.startsWith('/play')) {
return <span>{children}</span>;
}
return (
<Link href={href} {...props}>
{children}
</Link>
);
}
// 外部链接使用普通 a 标签
return <a href={href} target="_blank" rel="noopener noreferrer" {...props}>{children}</a>;
}
}}
>
{convertTitleToLink(message.content)}
</ReactMarkdown>
</div>
)}