1026 lines
28 KiB
JavaScript
1026 lines
28 KiB
JavaScript
import React, { PureComponent, createRef } from 'react';
|
|
import Box from '@mui/material/Box';
|
|
import Typography from '@mui/material/Typography';
|
|
import Tooltip from '@mui/material/Tooltip';
|
|
import ClickAwayListener from '@mui/material/ClickAwayListener';
|
|
import HandshakeRoundedIcon from '@mui/icons-material/HandshakeRounded';
|
|
import PsychologyRoundedIcon from '@mui/icons-material/PsychologyRounded';
|
|
import BuildRoundedIcon from '@mui/icons-material/BuildRounded';
|
|
import SearchRoundedIcon from '@mui/icons-material/SearchRounded';
|
|
import ComputerRoundedIcon from '@mui/icons-material/ComputerRounded';
|
|
import ImageSearchRoundedIcon from '@mui/icons-material/ImageSearchRounded';
|
|
import CompareRoundedIcon from '@mui/icons-material/CompareRounded';
|
|
import SupportIcon from '@mui/icons-material/Support';
|
|
import RecordVoiceOverIcon from '@mui/icons-material/RecordVoiceOver';
|
|
import PhonelinkIcon from '@mui/icons-material/Phonelink';
|
|
import SaveAsIcon from '@mui/icons-material/SaveAs';
|
|
import DescriptionIcon from '@mui/icons-material/Description';
|
|
import DeleteForeverIcon from '@mui/icons-material/DeleteForever';
|
|
import DeleteIcon from '@mui/icons-material/Delete';
|
|
import GoogleIcon from '@mui/icons-material/Google';
|
|
import WebIcon from '@mui/icons-material/Web';
|
|
import EscalatorWarningIcon from '@mui/icons-material/EscalatorWarning';
|
|
import PictureAsPdfIcon from '@mui/icons-material/PictureAsPdf';
|
|
import ManageSearchIcon from '@mui/icons-material/ManageSearch';
|
|
import HealingMarkdown from './HealingMarkdown';
|
|
import {
|
|
CHROMA,
|
|
SHADOW_BUBBLE_ASSISTANT,
|
|
SHADOW_BUBBLE_USER,
|
|
SURFACE_RAISED,
|
|
} from '../../colors';
|
|
|
|
class GoogleFaviconIcon extends PureComponent {
|
|
state = { hasError: false };
|
|
|
|
handleError = () => {
|
|
this.setState({ hasError: true });
|
|
};
|
|
|
|
render() {
|
|
const { sx } = this.props;
|
|
if (this.state.hasError) {
|
|
return <GoogleIcon sx={sx} />;
|
|
}
|
|
return (
|
|
<Box
|
|
component="img"
|
|
src="https://www.google.com/favicon.ico"
|
|
alt="Google"
|
|
onError={this.handleError}
|
|
sx={{
|
|
width: 18,
|
|
height: 18,
|
|
borderRadius: '2px',
|
|
objectFit: 'contain',
|
|
verticalAlign: 'middle',
|
|
display: 'inline-block',
|
|
...sx,
|
|
}}
|
|
/>
|
|
);
|
|
}
|
|
}
|
|
|
|
const TOOL_ICON_MAP = {
|
|
find_files: SearchRoundedIcon,
|
|
javascript_execute: ComputerRoundedIcon,
|
|
search_images: ImageSearchRoundedIcon,
|
|
analyze_image: CompareRoundedIcon,
|
|
apisuggestionsorissues: SupportIcon,
|
|
ask_user: RecordVoiceOverIcon,
|
|
dashback_request: PhonelinkIcon,
|
|
patch: SaveAsIcon,
|
|
read_file: DescriptionIcon,
|
|
reset_vfs: DeleteForeverIcon,
|
|
rm: DeleteIcon,
|
|
web_search: GoogleFaviconIcon,
|
|
url_get: WebIcon,
|
|
sub_agent: EscalatorWarningIcon,
|
|
pdf_read: PictureAsPdfIcon,
|
|
search_articles: ManageSearchIcon,
|
|
};
|
|
|
|
function formatMessageTime(isoString) {
|
|
if (!isoString) return '';
|
|
try {
|
|
const d = new Date(isoString);
|
|
if (Number.isNaN(d.getTime())) return '';
|
|
return d.toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' });
|
|
} catch {
|
|
return '';
|
|
}
|
|
}
|
|
|
|
function parseMessagePayload(raw) {
|
|
if (typeof raw !== 'string') return { content: raw };
|
|
try {
|
|
const parsed = JSON.parse(raw);
|
|
if (parsed && typeof parsed === 'object') return parsed;
|
|
} catch {}
|
|
return { content: raw };
|
|
}
|
|
|
|
function formatUsageText(usage) {
|
|
if (!usage) return '';
|
|
const input = Number(usage.input ?? usage.prompt_tokens ?? 0);
|
|
const output = Number(usage.output ?? usage.completion_tokens ?? 0);
|
|
const cached = Number(usage.cached ?? 0);
|
|
const cost = typeof usage.cost === 'number' ? ` · $${usage.cost.toFixed(6)}` : '';
|
|
const fmt = (n) => n.toLocaleString('de-DE');
|
|
const cachedPart = cached > 0 ? ` / ${fmt(cached)} Cache` : '';
|
|
return `${fmt(input)} Prompt${cachedPart} · ${fmt(output)} Antwort${cost}`;
|
|
}
|
|
|
|
function formatToolTooltip(content, result, parameters, prefix) {
|
|
const parts = [];
|
|
|
|
if (prefix) {
|
|
parts.push(prefix);
|
|
}
|
|
|
|
if (parameters !== undefined && parameters !== null && parameters !== '') {
|
|
try {
|
|
const parsed = typeof parameters === 'string' ? JSON.parse(parameters) : parameters;
|
|
parts.push(typeof parsed === 'object' ? JSON.stringify(parsed, null, 2) : String(parameters));
|
|
} catch {
|
|
parts.push(String(parameters));
|
|
}
|
|
}
|
|
|
|
if (content !== undefined && content !== null && content !== '') {
|
|
try {
|
|
const parsed = typeof content === 'string' ? JSON.parse(content) : content;
|
|
const str = typeof parsed === 'object' ? JSON.stringify(parsed, null, 2) : String(content);
|
|
if (!parts.includes(str)) {
|
|
parts.push(str);
|
|
}
|
|
} catch {
|
|
const str = String(content);
|
|
if (!parts.includes(str)) {
|
|
parts.push(str);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (result !== undefined && result !== null) {
|
|
try {
|
|
const resStr = typeof result === 'object' ? JSON.stringify(result, null, 2) : String(result);
|
|
if (!parts.some((p) => p.includes(resStr))) {
|
|
parts.push(resStr);
|
|
}
|
|
} catch {}
|
|
}
|
|
|
|
return parts.join('\n\n') || '';
|
|
}
|
|
|
|
class ClickTooltip extends PureComponent {
|
|
state = { open: false };
|
|
|
|
handleClose = () => {
|
|
if (this.state.open) this.setState({ open: false });
|
|
};
|
|
|
|
handleToggle = (e) => {
|
|
e?.stopPropagation();
|
|
this.setState((prev) => ({ open: !prev.open }));
|
|
};
|
|
|
|
render() {
|
|
const { title, children } = this.props;
|
|
return (
|
|
<ClickAwayListener onClickAway={this.handleClose}>
|
|
<Box sx={{ display: 'inline-flex', alignItems: 'center', height: 24 }}>
|
|
<Tooltip
|
|
open={this.state.open}
|
|
onClose={this.handleClose}
|
|
disableFocusListener
|
|
disableHoverListener
|
|
disableTouchListener
|
|
enterDelay={0}
|
|
leaveDelay={0}
|
|
TransitionProps={{ timeout: 0 }}
|
|
slotProps={{ transition: { timeout: 0 } }}
|
|
title={title}
|
|
arrow
|
|
placement="top"
|
|
>
|
|
<Box
|
|
onClick={this.handleToggle}
|
|
sx={{
|
|
display: 'inline-flex',
|
|
alignItems: 'center',
|
|
height: 24,
|
|
cursor: 'pointer',
|
|
}}
|
|
>
|
|
{children}
|
|
</Box>
|
|
</Tooltip>
|
|
</Box>
|
|
</ClickAwayListener>
|
|
);
|
|
}
|
|
}
|
|
|
|
function HandshakeIconView({ state }) {
|
|
const isGreen = state === 'green';
|
|
const isBlack = state === 'black';
|
|
|
|
return (
|
|
<Box
|
|
sx={{
|
|
display: 'inline-flex',
|
|
alignItems: 'center',
|
|
justifyContent: 'center',
|
|
width: 24,
|
|
height: 24,
|
|
color: isGreen
|
|
? CHROMA.GREEN.dark
|
|
: isBlack
|
|
? '#000000'
|
|
: '#9e9e9e',
|
|
transition: 'color 180ms ease-in-out',
|
|
}}
|
|
>
|
|
<HandshakeRoundedIcon sx={{ fontSize: 20, color: 'inherit' }} />
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
function extractReasoningText(msg, payload) {
|
|
if (payload && typeof payload === 'object') {
|
|
if (payload.plain !== undefined) {
|
|
if (typeof payload.plain === 'string' && payload.plain.trim()) {
|
|
return payload.plain.trim();
|
|
}
|
|
return '';
|
|
}
|
|
if (payload.encrypted) {
|
|
return '';
|
|
}
|
|
const candidate = payload.text || payload.content || payload.thinking || payload.reasoning;
|
|
if (candidate && typeof candidate === 'string') {
|
|
const trimmed = candidate.trim();
|
|
if (trimmed.startsWith('{') && (trimmed.includes('"encrypted"') || trimmed.includes('"plain"'))) {
|
|
try {
|
|
const inner = JSON.parse(trimmed);
|
|
if (inner.plain && typeof inner.plain === 'string' && inner.plain.trim()) {
|
|
return inner.plain.trim();
|
|
}
|
|
} catch {}
|
|
return '';
|
|
}
|
|
return trimmed;
|
|
}
|
|
}
|
|
|
|
if (typeof msg?.message === 'string') {
|
|
const raw = msg.message.trim();
|
|
if (raw.startsWith('{') && (raw.includes('"encrypted"') || raw.includes('"plain"'))) {
|
|
try {
|
|
const parsed = JSON.parse(raw);
|
|
if (parsed.plain && typeof parsed.plain === 'string' && parsed.plain.trim()) {
|
|
return parsed.plain.trim();
|
|
}
|
|
} catch {}
|
|
return '';
|
|
}
|
|
if (!raw.startsWith('{')) {
|
|
return raw;
|
|
}
|
|
}
|
|
|
|
return '';
|
|
}
|
|
|
|
class ReasoningTooltipContent extends PureComponent {
|
|
boxRef = createRef();
|
|
|
|
scrollToBottom = () => {
|
|
const el = this.boxRef.current;
|
|
if (el) {
|
|
el.scrollTop = el.scrollHeight;
|
|
requestAnimationFrame(() => {
|
|
if (this.boxRef.current) {
|
|
this.boxRef.current.scrollTop = this.boxRef.current.scrollHeight;
|
|
}
|
|
});
|
|
}
|
|
};
|
|
|
|
componentDidMount() {
|
|
this.scrollToBottom();
|
|
}
|
|
|
|
componentDidUpdate(prevProps) {
|
|
if (this.props.text !== prevProps.text) {
|
|
this.scrollToBottom();
|
|
}
|
|
}
|
|
|
|
render() {
|
|
const { text } = this.props;
|
|
return (
|
|
<Box
|
|
ref={this.boxRef}
|
|
sx={{
|
|
p: 0.5,
|
|
maxWidth: 440,
|
|
maxHeight: 280,
|
|
overflowY: 'auto',
|
|
whiteSpace: 'pre-wrap',
|
|
fontSize: '0.75rem',
|
|
lineHeight: 1.45,
|
|
}}
|
|
>
|
|
{text}
|
|
</Box>
|
|
);
|
|
}
|
|
}
|
|
|
|
function BrainIconView({ reasoning }) {
|
|
const tooltipContent = reasoning?.text?.trim() || '';
|
|
if (!tooltipContent) return null;
|
|
if (tooltipContent.startsWith('{') && (tooltipContent.includes('"encrypted"') || tooltipContent.includes('"plain"'))) {
|
|
return null;
|
|
}
|
|
|
|
return (
|
|
<ClickTooltip
|
|
title={<ReasoningTooltipContent text={tooltipContent} />}
|
|
>
|
|
<Box
|
|
sx={{
|
|
display: 'inline-flex',
|
|
alignItems: 'center',
|
|
justifyContent: 'center',
|
|
width: 24,
|
|
height: 24,
|
|
color: CHROMA.PURPLE.dark,
|
|
borderRadius: 1,
|
|
'&:hover': {
|
|
bgcolor: 'action.hover',
|
|
},
|
|
}}
|
|
>
|
|
<PsychologyRoundedIcon sx={{ fontSize: 20, color: 'inherit' }} />
|
|
</Box>
|
|
</ClickTooltip>
|
|
);
|
|
}
|
|
|
|
function extractUrlFromTool(tool) {
|
|
if (!tool) return null;
|
|
const candidates = [tool.parameters, tool.arguments, tool.content, tool.result];
|
|
for (const item of candidates) {
|
|
if (!item) continue;
|
|
if (typeof item === 'object') {
|
|
const u = item.url || item.Url || item.uri || item.URI || item.link;
|
|
if (typeof u === 'string' && u.trim()) return u.trim();
|
|
}
|
|
const str = typeof item === 'string' ? item : JSON.stringify(item);
|
|
const match = str.match(/https?:\/\/[^\s"'`<>{}|\\]+/);
|
|
if (match) {
|
|
return match[0].replace(/[.,;:)]+$/, '');
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
class UrlFaviconIcon extends PureComponent {
|
|
state = {
|
|
srcIndex: 0,
|
|
hasError: false,
|
|
};
|
|
|
|
sources = [];
|
|
|
|
constructor(props) {
|
|
super(props);
|
|
this.sources = this.buildSources(props.url);
|
|
}
|
|
|
|
componentDidUpdate(prevProps) {
|
|
if (this.props.url !== prevProps.url) {
|
|
this.sources = this.buildSources(this.props.url);
|
|
this.setState({ srcIndex: 0, hasError: false });
|
|
}
|
|
}
|
|
|
|
buildSources(urlStr) {
|
|
if (!urlStr || typeof urlStr !== 'string') return [];
|
|
try {
|
|
const parsed = new URL(
|
|
urlStr.startsWith('http://') || urlStr.startsWith('https://')
|
|
? urlStr
|
|
: `https://${urlStr}`
|
|
);
|
|
const hostname = parsed.hostname;
|
|
const origin = parsed.origin;
|
|
if (!hostname) return [];
|
|
|
|
const list = [];
|
|
if (hostname === 'localhost' || hostname === '127.0.0.1' || hostname.endsWith('.local')) {
|
|
list.push(`${origin}/favicon.ico`);
|
|
} else {
|
|
list.push(`https://icons.duckduckgo.com/ip3/${hostname}.ico`);
|
|
list.push(`${origin}/favicon.ico`);
|
|
}
|
|
return list;
|
|
} catch {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
handleError = () => {
|
|
if (this.state.srcIndex + 1 < this.sources.length) {
|
|
this.setState((prev) => ({ srcIndex: prev.srcIndex + 1 }));
|
|
} else {
|
|
this.setState({ hasError: true });
|
|
}
|
|
};
|
|
|
|
render() {
|
|
const { sx } = this.props;
|
|
const { srcIndex, hasError } = this.state;
|
|
const currentSrc = this.sources[srcIndex];
|
|
|
|
if (!currentSrc || hasError) {
|
|
return <WebIcon sx={sx} />;
|
|
}
|
|
|
|
return (
|
|
<Box
|
|
component="img"
|
|
src={currentSrc}
|
|
alt=""
|
|
onError={this.handleError}
|
|
sx={{
|
|
width: 18,
|
|
height: 18,
|
|
borderRadius: '2px',
|
|
objectFit: 'contain',
|
|
verticalAlign: 'middle',
|
|
display: 'inline-block',
|
|
...sx,
|
|
}}
|
|
/>
|
|
);
|
|
}
|
|
}
|
|
|
|
function WorkIconView({ tool }) {
|
|
if (!tool || !tool.name) return null;
|
|
const isGreen = tool?.state === 'green';
|
|
const isUrlGet = tool?.name === 'url_get';
|
|
const ToolNameIcon = tool?.name ? TOOL_ICON_MAP[tool.name] : null;
|
|
const prefix = ToolNameIcon ? tool?.name : null;
|
|
const tooltipText = formatToolTooltip(tool?.content, tool?.result, tool?.parameters || tool?.arguments, prefix);
|
|
const targetUrl = isUrlGet ? extractUrlFromTool(tool) : null;
|
|
|
|
return (
|
|
<ClickTooltip
|
|
title={
|
|
<Box
|
|
sx={{
|
|
p: 0.5,
|
|
maxWidth: 480,
|
|
maxHeight: 300,
|
|
overflowY: 'auto',
|
|
whiteSpace: 'pre-wrap',
|
|
fontFamily: 'monospace',
|
|
fontSize: '0.72rem',
|
|
lineHeight: 1.4,
|
|
}}
|
|
>
|
|
{tooltipText}
|
|
</Box>
|
|
}
|
|
>
|
|
<Box
|
|
sx={{
|
|
display: 'inline-flex',
|
|
alignItems: 'center',
|
|
height: 24,
|
|
gap: 0.5,
|
|
color: isGreen ? CHROMA.GREEN.dark : '#000000',
|
|
px: 0.25,
|
|
borderRadius: 1,
|
|
'&:hover': {
|
|
bgcolor: 'action.hover',
|
|
},
|
|
}}
|
|
>
|
|
{isUrlGet ? (
|
|
<UrlFaviconIcon url={targetUrl} sx={{ fontSize: 18, color: 'inherit' }} />
|
|
) : ToolNameIcon ? (
|
|
<ToolNameIcon sx={{ fontSize: 18, color: 'inherit' }} />
|
|
) : (
|
|
<>
|
|
<BuildRoundedIcon sx={{ fontSize: 18, color: 'inherit' }} />
|
|
{tool?.name ? (
|
|
<Typography
|
|
component="span"
|
|
sx={{
|
|
fontFamily: 'monospace',
|
|
fontSize: '0.75rem',
|
|
fontWeight: 600,
|
|
color: 'text.secondary',
|
|
lineHeight: 1,
|
|
}}
|
|
>
|
|
{tool.name}
|
|
</Typography>
|
|
) : null}
|
|
</>
|
|
)}
|
|
</Box>
|
|
</ClickTooltip>
|
|
);
|
|
}
|
|
|
|
function UsageChipView({ usage }) {
|
|
const text = formatUsageText(usage);
|
|
if (!text) return null;
|
|
|
|
return (
|
|
<Box
|
|
component="span"
|
|
sx={{
|
|
display: 'inline-flex',
|
|
alignItems: 'center',
|
|
fontSize: '0.68rem',
|
|
fontFamily: 'monospace',
|
|
color: 'text.secondary',
|
|
bgcolor: 'action.hover',
|
|
px: 0.8,
|
|
py: 0.25,
|
|
borderRadius: 1.5,
|
|
whiteSpace: 'nowrap',
|
|
}}
|
|
>
|
|
{text}
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
export function TurnRow({ turn, isLive = false }) {
|
|
if (!turn) return null;
|
|
|
|
const hasReasonings = Boolean(
|
|
turn.reasonings &&
|
|
turn.reasonings.some((r) => Boolean(r.text && r.text.trim()))
|
|
);
|
|
const hasTools = Boolean(turn.tools && turn.tools.length > 0);
|
|
const hasContent = Boolean(turn.content);
|
|
const showHandshake = !hasReasonings && !hasTools && !hasContent;
|
|
const showStatusLine = showHandshake || hasReasonings || hasTools;
|
|
|
|
return (
|
|
<Box
|
|
sx={{
|
|
display: 'flex',
|
|
flexDirection: 'column',
|
|
alignItems: 'flex-start',
|
|
px: 2,
|
|
py: 0.4,
|
|
gap: 0.75,
|
|
maxWidth: '100%',
|
|
}}
|
|
>
|
|
{/* Turn status line: left of the chat */}
|
|
{showStatusLine ? (
|
|
<Box
|
|
sx={{
|
|
display: 'flex',
|
|
alignItems: 'center',
|
|
gap: 1,
|
|
flexWrap: 'wrap',
|
|
minHeight: 24,
|
|
lineHeight: 1,
|
|
}}
|
|
>
|
|
{showHandshake ? <HandshakeIconView state={turn.handshake} /> : null}
|
|
|
|
{turn.reasonings?.map((r, i) => (
|
|
<BrainIconView key={r.id || i} reasoning={r} />
|
|
))}
|
|
|
|
{turn.tools?.map((t, i) => (
|
|
<WorkIconView key={t.id || i} tool={t} />
|
|
))}
|
|
</Box>
|
|
) : null}
|
|
|
|
{/* Assistant message content for this turn (if any) */}
|
|
{hasContent ? (
|
|
<Box
|
|
sx={(theme) => ({
|
|
maxWidth: '85%',
|
|
px: 2,
|
|
py: 1.25,
|
|
borderRadius: '16px 16px 16px 4px',
|
|
bgcolor:
|
|
theme.palette.mode === 'dark'
|
|
? 'background.paper'
|
|
: SURFACE_RAISED,
|
|
color: theme.palette.text.primary,
|
|
boxShadow: SHADOW_BUBBLE_ASSISTANT,
|
|
border: `1px solid ${theme.palette.divider}`,
|
|
wordBreak: 'break-word',
|
|
fontSize: '0.875rem',
|
|
lineHeight: 1.5,
|
|
})}
|
|
>
|
|
<HealingMarkdown text={turn.content} streaming={isLive && !turn.sealed} />
|
|
</Box>
|
|
) : null}
|
|
|
|
{/* Pricing / usage is always on a new line */}
|
|
{turn.usage ? (
|
|
<Box sx={{ display: 'flex', alignItems: 'center' }}>
|
|
<UsageChipView usage={turn.usage} />
|
|
</Box>
|
|
) : null}
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
export function normalizeChatMessages(rawMessages = []) {
|
|
if (!Array.isArray(rawMessages) || rawMessages.length === 0) return [];
|
|
|
|
const result = [];
|
|
let currentAssistantTurns = [];
|
|
let currentTurn = null;
|
|
|
|
const flushAssistant = () => {
|
|
if (currentTurn) {
|
|
if (
|
|
(currentTurn.reasonings && currentTurn.reasonings.length > 0) ||
|
|
(currentTurn.tools && currentTurn.tools.length > 0) ||
|
|
currentTurn.content ||
|
|
currentTurn.usage
|
|
) {
|
|
currentAssistantTurns.push(currentTurn);
|
|
}
|
|
currentTurn = null;
|
|
}
|
|
if (currentAssistantTurns.length > 0) {
|
|
const fullText = currentAssistantTurns
|
|
.map((t) => t.content)
|
|
.filter(Boolean)
|
|
.join('\n\n');
|
|
|
|
result.push({
|
|
id: currentAssistantTurns[0]?.id ? `asst-${currentAssistantTurns[0].id}` : `asst-${result.length}`,
|
|
role: 'assistant',
|
|
message: fullText,
|
|
turns: currentAssistantTurns,
|
|
created_at: currentAssistantTurns[0]?.created_at,
|
|
});
|
|
currentAssistantTurns = [];
|
|
}
|
|
};
|
|
|
|
let turnSeq = 0;
|
|
const findExistingTool = (toolCallId) => {
|
|
if (!toolCallId) return null;
|
|
if (currentTurn?.tools) {
|
|
const t = currentTurn.tools.find((tool) => tool.id === toolCallId);
|
|
if (t) return t;
|
|
}
|
|
for (let i = currentAssistantTurns.length - 1; i >= 0; i--) {
|
|
const t = currentAssistantTurns[i].tools?.find((tool) => tool.id === toolCallId);
|
|
if (t) return t;
|
|
}
|
|
return null;
|
|
};
|
|
|
|
const getOrCreateTurn = (msg) => {
|
|
const shouldStartNew =
|
|
!currentTurn ||
|
|
currentTurn.sealed ||
|
|
(msg?.role === 'reasoning' && (currentTurn.tools.length > 0 || Boolean(currentTurn.content))) ||
|
|
(msg?.role === 'tool' && Boolean(currentTurn.content));
|
|
|
|
if (!currentTurn) {
|
|
turnSeq++;
|
|
currentTurn = {
|
|
id: msg?.id ? `turn-msg-${msg.id}` : `turn-${result.length}-${turnSeq}`,
|
|
handshake: 'green',
|
|
reasonings: [],
|
|
tools: [],
|
|
content: '',
|
|
usage: null,
|
|
sealed: false,
|
|
created_at: msg?.created_at,
|
|
};
|
|
} else if (shouldStartNew) {
|
|
currentAssistantTurns.push(currentTurn);
|
|
turnSeq++;
|
|
currentTurn = {
|
|
id: msg?.id ? `turn-msg-${msg.id}` : `turn-${result.length}-${turnSeq}`,
|
|
handshake: 'green',
|
|
reasonings: [],
|
|
tools: [],
|
|
content: '',
|
|
usage: null,
|
|
sealed: false,
|
|
created_at: msg?.created_at,
|
|
};
|
|
}
|
|
return currentTurn;
|
|
};
|
|
|
|
for (const msg of rawMessages) {
|
|
if (!msg) continue;
|
|
|
|
if (msg.turns && Array.isArray(msg.turns) && msg.turns.length > 0) {
|
|
flushAssistant();
|
|
result.push(msg);
|
|
continue;
|
|
}
|
|
|
|
const payload = parseMessagePayload(msg.message);
|
|
|
|
if (msg.role === 'user') {
|
|
flushAssistant();
|
|
result.push({
|
|
...msg,
|
|
message: payload.content !== undefined ? payload.content : msg.message,
|
|
});
|
|
continue;
|
|
}
|
|
|
|
if (msg.role === 'reasoning') {
|
|
const text = extractReasoningText(msg, payload);
|
|
if (text) {
|
|
const turn = getOrCreateTurn(msg);
|
|
turn.reasonings.push({
|
|
id: msg.id || `r-${turn.reasonings.length}`,
|
|
text,
|
|
sealed: true,
|
|
});
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (msg.role === 'tool') {
|
|
const existingTool = payload.tool_call_id ? findExistingTool(payload.tool_call_id) : null;
|
|
const isAskUser = payload.name === 'ask_user' || existingTool?.name === 'ask_user';
|
|
|
|
if (!isAskUser && existingTool) {
|
|
existingTool.result = payload.result !== undefined ? payload.result : (payload.content !== undefined ? payload.content : '');
|
|
if (payload.parameters) {
|
|
existingTool.parameters = payload.parameters;
|
|
}
|
|
if (payload.name && !existingTool.name) {
|
|
existingTool.name = payload.name;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
const turn = getOrCreateTurn(msg);
|
|
turn.tools.push({
|
|
id: payload.tool_call_id || msg.id,
|
|
state: 'green',
|
|
name: isAskUser ? 'ask_user' : (payload.name || ''),
|
|
content: payload.content !== undefined ? payload.content : (payload.answer || payload.parameters || ''),
|
|
parameters: payload.parameters,
|
|
result: payload.result,
|
|
});
|
|
continue;
|
|
}
|
|
|
|
if (msg.role === 'usage') {
|
|
const turn = getOrCreateTurn(msg);
|
|
turn.usage = payload;
|
|
turn.sealed = true;
|
|
continue;
|
|
}
|
|
|
|
if (msg.role === 'assistant' || msg.role === 'model') {
|
|
const toolCalls = Array.isArray(payload.tool_calls) && payload.tool_calls.length > 0 ? payload.tool_calls : null;
|
|
if (toolCalls) {
|
|
const turn = getOrCreateTurn(msg);
|
|
for (const tc of toolCalls) {
|
|
let args = {};
|
|
try {
|
|
args = typeof tc.function?.arguments === 'string' ? JSON.parse(tc.function.arguments) : (tc.function?.arguments || {});
|
|
} catch {}
|
|
|
|
const isAskUser = tc.function?.name === 'ask_user';
|
|
const existing = !isAskUser && tc.id ? findExistingTool(tc.id) : null;
|
|
if (existing) {
|
|
existing.parameters = tc.function?.arguments || args;
|
|
if (tc.function?.name && !existing.name) existing.name = tc.function.name;
|
|
} else {
|
|
turn.tools.push({
|
|
id: tc.id || `tc-${Date.now()}`,
|
|
state: 'green',
|
|
name: tc.function?.name || 'tool',
|
|
parameters: tc.function?.arguments || args,
|
|
content: null,
|
|
result: null,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
const content = payload.content !== undefined ? payload.content : msg.message;
|
|
if (content && typeof content === 'string') {
|
|
const turn = getOrCreateTurn(msg);
|
|
turn.content = content;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
flushAssistant();
|
|
result.push(msg);
|
|
}
|
|
|
|
flushAssistant();
|
|
return result;
|
|
}
|
|
|
|
function MessageRow({ msg, showTime = true }) {
|
|
// If the message has structured turns, render each turn
|
|
if (msg.turns && Array.isArray(msg.turns) && msg.turns.length > 0) {
|
|
return (
|
|
<Box sx={{ display: 'flex', flexDirection: 'column', gap: 0.25 }}>
|
|
{msg.turns.map((t, idx) => (
|
|
<TurnRow key={t.id || idx} turn={t} isLive={false} />
|
|
))}
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
const payload = parseMessagePayload(msg.message);
|
|
const textContent = payload.content !== undefined ? payload.content : msg.message;
|
|
|
|
if (msg.role === 'system') {
|
|
return (
|
|
<Box sx={{ display: 'flex', justifyContent: 'center', py: 0.75, px: 2 }}>
|
|
<Box
|
|
sx={{
|
|
bgcolor: 'action.hover',
|
|
color: 'text.secondary',
|
|
px: 2,
|
|
py: 0.5,
|
|
borderRadius: 4,
|
|
fontSize: '0.75rem',
|
|
fontWeight: 500,
|
|
maxWidth: '85%',
|
|
textAlign: 'center',
|
|
}}
|
|
>
|
|
{textContent}
|
|
</Box>
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
const isUser = msg.role === 'user';
|
|
const timeStr = showTime ? formatMessageTime(msg.created_at) : '';
|
|
|
|
return (
|
|
<Box
|
|
sx={{
|
|
display: 'flex',
|
|
justifyContent: isUser ? 'flex-end' : 'flex-start',
|
|
px: 2,
|
|
py: 0.75,
|
|
}}
|
|
>
|
|
<Box
|
|
sx={{
|
|
maxWidth: '85%',
|
|
display: 'flex',
|
|
flexDirection: 'column',
|
|
alignItems: isUser ? 'flex-end' : 'flex-start',
|
|
}}
|
|
>
|
|
{timeStr ? (
|
|
<Box
|
|
sx={{
|
|
display: 'flex',
|
|
alignItems: 'center',
|
|
mb: 0.3,
|
|
px: 0.5,
|
|
}}
|
|
>
|
|
<Typography variant="caption" sx={{ color: 'text.disabled', fontSize: '0.68rem' }}>
|
|
{timeStr}
|
|
</Typography>
|
|
</Box>
|
|
) : null}
|
|
|
|
<Box
|
|
sx={(theme) => ({
|
|
px: 2,
|
|
py: 1.25,
|
|
borderRadius: isUser ? '16px 16px 4px 16px' : '16px 16px 16px 4px',
|
|
bgcolor: isUser
|
|
? theme.palette.primary.main
|
|
: theme.palette.mode === 'dark'
|
|
? 'background.paper'
|
|
: SURFACE_RAISED,
|
|
color: isUser ? theme.palette.primary.contrastText : theme.palette.text.primary,
|
|
boxShadow: isUser ? SHADOW_BUBBLE_USER : SHADOW_BUBBLE_ASSISTANT,
|
|
border: isUser ? 'none' : `1px solid ${theme.palette.divider}`,
|
|
wordBreak: 'break-word',
|
|
whiteSpace: isUser ? 'pre-wrap' : 'normal',
|
|
fontSize: '0.875rem',
|
|
lineHeight: 1.5,
|
|
})}
|
|
>
|
|
{isUser ? (
|
|
textContent
|
|
) : (
|
|
<HealingMarkdown text={textContent} streaming={msg.streaming} />
|
|
)}
|
|
</Box>
|
|
</Box>
|
|
</Box>
|
|
);
|
|
}
|
|
|
|
export default class ChatFlowList extends PureComponent {
|
|
containerRef = createRef();
|
|
bottomRef = createRef();
|
|
scrollTimeout = null;
|
|
|
|
componentDidMount() {
|
|
this.scrollToBottom('auto');
|
|
}
|
|
|
|
componentDidUpdate(prevProps) {
|
|
if (
|
|
this.props.messages !== prevProps.messages ||
|
|
this.props.messages?.length !== prevProps.messages?.length ||
|
|
this.props.activeTurns !== prevProps.activeTurns
|
|
) {
|
|
this.scrollToBottom('smooth');
|
|
}
|
|
}
|
|
|
|
componentWillUnmount() {
|
|
if (this.scrollTimeout) {
|
|
clearTimeout(this.scrollTimeout);
|
|
this.scrollTimeout = null;
|
|
}
|
|
}
|
|
|
|
scrollToBottom = (behavior = 'smooth') => {
|
|
const doScroll = () => {
|
|
if (this.bottomRef.current?.scrollIntoView) {
|
|
this.bottomRef.current.scrollIntoView({ behavior, block: 'end' });
|
|
}
|
|
if (this.containerRef.current) {
|
|
this.containerRef.current.scrollTop = this.containerRef.current.scrollHeight;
|
|
}
|
|
};
|
|
|
|
// Immediate frame
|
|
requestAnimationFrame(doScroll);
|
|
|
|
// Delayed fallback for font/layout reflows
|
|
if (this.scrollTimeout) clearTimeout(this.scrollTimeout);
|
|
this.scrollTimeout = setTimeout(doScroll, 60);
|
|
};
|
|
|
|
render() {
|
|
const { messages = [], activeTurns = [] } = this.props;
|
|
const normalizedMessages = normalizeChatMessages(messages);
|
|
|
|
if (normalizedMessages.length === 0 && activeTurns.length === 0) {
|
|
return null;
|
|
}
|
|
|
|
return (
|
|
<Box
|
|
ref={this.containerRef}
|
|
sx={{
|
|
flexGrow: 1,
|
|
height: '100%',
|
|
width: '100%',
|
|
minHeight: 0,
|
|
overflowY: 'auto',
|
|
overflowX: 'hidden',
|
|
display: 'flex',
|
|
flexDirection: 'column',
|
|
py: 1,
|
|
gap: 0.25,
|
|
}}
|
|
>
|
|
{normalizedMessages.map((msg, index) => {
|
|
const prevMsg = index > 0 ? normalizedMessages[index - 1] : null;
|
|
const currTime = formatMessageTime(msg.created_at);
|
|
const prevTime = prevMsg ? formatMessageTime(prevMsg.created_at) : '';
|
|
const showTime = Boolean(currTime && currTime !== prevTime);
|
|
|
|
return (
|
|
<MessageRow
|
|
key={msg.id || index}
|
|
msg={msg}
|
|
showTime={showTime}
|
|
/>
|
|
);
|
|
})}
|
|
|
|
{/* Live streaming turns */}
|
|
{activeTurns && activeTurns.length > 0
|
|
? activeTurns.map((turn, idx) => (
|
|
<TurnRow key={turn.id || idx} turn={turn} isLive={true} />
|
|
))
|
|
: null}
|
|
|
|
<div ref={this.bottomRef} style={{ height: 1, flexShrink: 0 }} />
|
|
</Box>
|
|
);
|
|
}
|
|
}
|