5e8cb628f0
The "Thinking" (reasoning) block rendered with large vertical gaps: models emit reasoning with a blank line (\n\n) between every list item and paragraph, which `marked` turns into loose lists (each <li> wrapped in a <p>) and separate <p> paragraphs, each carrying a margin. - Add `collapseBlankLines(text)`: collapse 2+ newlines to one, EXCEPT inside fenced code blocks (``` / ~~~) where blank lines are significant. Applied in reasoning-block.tsx before renderChatMarkdown, so loose lists become tight (no <li><p>) and paragraphs join; `breaks: true` keeps single \n as <br>, preserving line breaks. Reasoning-only — the normal answer is untouched. - Drop `white-space: pre-wrap` from `.reasoningText`: on the rendered markdown <div> it turned the newlines between block tags into visible blank lines on top of the margins. The plain-text fallback <Text> that needs pre-wrap already sets it inline. Tests: collapseBlankLines unit (collapse, fence preservation incl. tilde and unclosed fences) + rendered-HTML assertions that a blank-line-separated list becomes a tight list and still parses as a list after a paragraph. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
90 lines
3.4 KiB
TypeScript
90 lines
3.4 KiB
TypeScript
import { useState } from "react";
|
|
import { Box, Collapse, Group, Text, UnstyledButton } from "@mantine/core";
|
|
import { IconChevronDown } from "@tabler/icons-react";
|
|
import { useTranslation } from "react-i18next";
|
|
import { estimateTokens } from "@/features/ai-chat/utils/count-stream-tokens.ts";
|
|
import { collapseBlankLines } from "@/features/ai-chat/utils/collapse-blank-lines.ts";
|
|
import { renderChatMarkdown } from "@/features/ai-chat/utils/markdown.ts";
|
|
import classes from "@/features/ai-chat/components/ai-chat.module.css";
|
|
|
|
interface ReasoningBlockProps {
|
|
/** The streamed/persisted reasoning (thinking) text. May be empty when the
|
|
* provider reports only a reasoning token COUNT without the text. */
|
|
text: string;
|
|
/** Authoritative reasoning token count from `usage.reasoningTokens`, when the
|
|
* step/turn has finished. When absent (or 0) the count is estimated from the
|
|
* text length so it ticks live as the reasoning streams in. */
|
|
tokens?: number;
|
|
}
|
|
|
|
/**
|
|
* Collapsible "Thinking" block for an assistant `reasoning` part. Mirrors Claude
|
|
* Code's surfacing of the model's thinking: a header that shows the thinking
|
|
* token count (authoritative when the step has reported usage, else a live
|
|
* estimate from the streamed text) and an expandable body with the reasoning
|
|
* prose. Collapsed by default so it never crowds out the answer.
|
|
*
|
|
* Providers that don't stream reasoning TEXT still render this block from the
|
|
* authoritative count alone (header only, empty body) so the cost is visible.
|
|
*/
|
|
export default function ReasoningBlock({ text, tokens }: ReasoningBlockProps) {
|
|
const { t } = useTranslation();
|
|
const [open, setOpen] = useState(false);
|
|
|
|
// Authoritative count wins; otherwise estimate live from the streamed text.
|
|
const count = tokens && tokens > 0 ? tokens : estimateTokens(text);
|
|
const trimmed = text.trim();
|
|
// Collapse the blank-line gaps the model emits between every list item /
|
|
// paragraph so the reasoning renders compactly (tight lists, joined
|
|
// paragraphs) — see collapseBlankLines. ONLY here, not in the normal answer.
|
|
const html = trimmed
|
|
? renderChatMarkdown(collapseBlankLines(trimmed), {})
|
|
: "";
|
|
|
|
return (
|
|
<Box className={classes.reasoningBlock} mb={6}>
|
|
<UnstyledButton
|
|
onClick={() => setOpen((o) => !o)}
|
|
// No body to expand when the provider reported only a token count.
|
|
disabled={!trimmed}
|
|
aria-expanded={open}
|
|
>
|
|
<Group gap={6} wrap="nowrap" align="center">
|
|
<IconChevronDown
|
|
size={12}
|
|
style={{
|
|
transform: open ? "none" : "rotate(-90deg)",
|
|
transition: "transform 150ms ease",
|
|
opacity: trimmed ? 1 : 0.4,
|
|
}}
|
|
/>
|
|
<Text size="xs" c="dimmed">
|
|
{count > 0
|
|
? t("Thinking · {{count}} tokens", { count })
|
|
: t("Thinking")}
|
|
</Text>
|
|
</Group>
|
|
</UnstyledButton>
|
|
|
|
{trimmed && (
|
|
<Collapse in={open}>
|
|
{html ? (
|
|
<div
|
|
className={classes.reasoningText}
|
|
// Sanitized by renderChatMarkdown (DOMPurify) before insertion.
|
|
dangerouslySetInnerHTML={{ __html: html }}
|
|
/>
|
|
) : (
|
|
<Text
|
|
className={classes.reasoningText}
|
|
style={{ whiteSpace: "pre-wrap" }}
|
|
>
|
|
{trimmed}
|
|
</Text>
|
|
)}
|
|
</Collapse>
|
|
)}
|
|
</Box>
|
|
);
|
|
}
|