mirror of
https://github.com/siteboon/claudecodeui.git
synced 2026-06-28 23:35:27 +08:00
Fix/websocket streaming issues (#748)
This commit is contained in:
@@ -157,9 +157,14 @@ export class ClaudeSessionSynchronizer implements IProviderSessionSynchronizer {
|
||||
const eventSessionId = typeof data.sessionId === 'string' ? data.sessionId : undefined;
|
||||
const aiTitle = typeof data.aiTitle === 'string' ? data.aiTitle : undefined;
|
||||
const lastPrompt = typeof data.lastPrompt === 'string' ? data.lastPrompt : undefined;
|
||||
const claudeRenamedTitle = typeof data.customTitle === 'string' ? data.customTitle : undefined;
|
||||
|
||||
if ((eventType === 'ai-title' && eventSessionId === sessionId && aiTitle?.trim()) || (eventType === 'last-prompt' && eventSessionId === sessionId && lastPrompt?.trim())) {
|
||||
return aiTitle || lastPrompt;
|
||||
if (
|
||||
(eventType === 'ai-title' && eventSessionId === sessionId && aiTitle?.trim()) ||
|
||||
(eventType === 'last-prompt' && eventSessionId === sessionId && lastPrompt?.trim()) ||
|
||||
(eventType === "custom-title" && eventSessionId === sessionId && claudeRenamedTitle?.trim())
|
||||
) {
|
||||
return aiTitle || lastPrompt || claudeRenamedTitle;
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
|
||||
@@ -200,17 +200,18 @@ async function getSessionMessages(
|
||||
}
|
||||
|
||||
/**
|
||||
* Claude writes internal command and system reminder entries into history.
|
||||
* Those are useful for the CLI but should not appear in the user-facing chat.
|
||||
* Claude writes a mix of truly internal transcript rows and "UI-hidden" local
|
||||
* command artifacts into the same JSONL stream.
|
||||
*
|
||||
* Important distinction:
|
||||
* - system reminders / caveats / interruption banners should stay hidden
|
||||
* - local command payloads (`<command-name>...`) and stdout wrappers
|
||||
* (`<local-command-stdout>...`) should be remapped into normal chat messages
|
||||
* instead of being discarded as internal content
|
||||
*/
|
||||
const INTERNAL_CONTENT_PREFIXES = [
|
||||
'<command-name>',
|
||||
'<command-message>',
|
||||
'<command-args>',
|
||||
'<local-command-stdout>',
|
||||
'<system-reminder>',
|
||||
'Caveat:',
|
||||
'This session is being continued from a previous',
|
||||
'[Request interrupted',
|
||||
] as const;
|
||||
|
||||
@@ -218,6 +219,73 @@ function isInternalContent(content: string): boolean {
|
||||
return INTERNAL_CONTENT_PREFIXES.some((prefix) => content.startsWith(prefix));
|
||||
}
|
||||
|
||||
/**
|
||||
* Claude wraps local slash-command metadata in lightweight XML-like tags inside
|
||||
* a plain string payload. We intentionally parse only the small tag surface we
|
||||
* care about instead of introducing a generic XML parser for untrusted history.
|
||||
*/
|
||||
function extractTaggedContent(content: string, tagName: string): string | null {
|
||||
const escapedTagName = tagName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
|
||||
const match = new RegExp(`<${escapedTagName}>([\\s\\S]*?)<\\/${escapedTagName}>`).exec(content);
|
||||
return match ? match[1] : null;
|
||||
}
|
||||
|
||||
type ClaudeLocalCommandPayload = {
|
||||
commandName: string;
|
||||
commandMessage: string;
|
||||
commandArgs: string;
|
||||
};
|
||||
|
||||
/**
|
||||
* Converts Claude's hidden local command wrapper into structured metadata.
|
||||
*
|
||||
* The three tags often coexist in one string payload. Returning `null` lets the
|
||||
* normal text path continue untouched for unrelated messages.
|
||||
*/
|
||||
function parseLocalCommandPayload(content: string): ClaudeLocalCommandPayload | null {
|
||||
const commandName = extractTaggedContent(content, 'command-name');
|
||||
const commandMessage = extractTaggedContent(content, 'command-message');
|
||||
const commandArgs = extractTaggedContent(content, 'command-args');
|
||||
|
||||
if (commandName === null && commandMessage === null && commandArgs === null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return {
|
||||
commandName: commandName ?? '',
|
||||
commandMessage: commandMessage ?? '',
|
||||
commandArgs: commandArgs ?? '',
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Produces the short user-visible command string that should appear in chat.
|
||||
*
|
||||
* We prefer the slash-prefixed command name because that most closely matches
|
||||
* what the user actually typed, and only fall back to the message body when the
|
||||
* command name is unavailable in older transcript variants.
|
||||
*/
|
||||
function buildLocalCommandDisplayText(payload: ClaudeLocalCommandPayload): string {
|
||||
const commandName = payload.commandName.trim();
|
||||
const commandMessage = payload.commandMessage.trim();
|
||||
const commandArgs = payload.commandArgs.trim();
|
||||
const baseCommand = commandName || commandMessage;
|
||||
|
||||
if (!baseCommand) {
|
||||
return '';
|
||||
}
|
||||
|
||||
return commandArgs ? `${baseCommand} ${commandArgs}` : baseCommand;
|
||||
}
|
||||
|
||||
/**
|
||||
* Claude local-command stdout may contain ANSI styling codes because it was
|
||||
* captured from the terminal. The web chat should receive readable plain text.
|
||||
*/
|
||||
function stripAnsiFormatting(text: string): string {
|
||||
return text.replace(/\u001B\[[0-9;?]*[ -/]*[@-~]/g, '');
|
||||
}
|
||||
|
||||
export class ClaudeSessionsProvider implements IProviderSessions {
|
||||
/**
|
||||
* Normalizes one Claude JSONL entry or live SDK stream event into the shared
|
||||
@@ -240,7 +308,7 @@ export class ClaudeSessionsProvider implements IProviderSessions {
|
||||
const ts = raw.timestamp || new Date().toISOString();
|
||||
const baseId = raw.uuid || generateMessageId('claude');
|
||||
|
||||
if (raw.message?.role === 'user' && raw.message?.content) {
|
||||
if (raw.message?.role === 'user' && raw.message?.content && raw.isMeta !== true) {
|
||||
if (Array.isArray(raw.message.content)) {
|
||||
for (let partIndex = 0; partIndex < raw.message.content.length; partIndex++) {
|
||||
const part = raw.message.content[partIndex];
|
||||
@@ -293,6 +361,80 @@ export class ClaudeSessionsProvider implements IProviderSessions {
|
||||
}
|
||||
} else if (typeof raw.message.content === 'string') {
|
||||
const text = raw.message.content;
|
||||
|
||||
/**
|
||||
* Claude stores compact summaries as synthetic "user" rows so the CLI
|
||||
* can resume the next session turn with the summary in-context.
|
||||
*
|
||||
* For the web UI this is much more useful as assistant-authored summary
|
||||
* text; otherwise it is both filtered by the generic internal-prefix
|
||||
* check and visually mislabeled as a user message.
|
||||
*/
|
||||
if (raw.isCompactSummary === true && text.trim()) {
|
||||
messages.push(createNormalizedMessage({
|
||||
id: baseId,
|
||||
sessionId,
|
||||
timestamp: ts,
|
||||
provider: PROVIDER,
|
||||
kind: 'text',
|
||||
role: 'assistant',
|
||||
content: text,
|
||||
isCompactSummary: true,
|
||||
}));
|
||||
return messages;
|
||||
}
|
||||
|
||||
/**
|
||||
* Local slash commands are serialized as tagged text even though they
|
||||
* are semantically a user action. Expose the parsed fields to the
|
||||
* frontend and emit a plain user-visible command string so the command
|
||||
* no longer disappears from history.
|
||||
*/
|
||||
const localCommandPayload = parseLocalCommandPayload(text);
|
||||
if (localCommandPayload) {
|
||||
const displayText = buildLocalCommandDisplayText(localCommandPayload);
|
||||
if (displayText) {
|
||||
messages.push(createNormalizedMessage({
|
||||
id: baseId,
|
||||
sessionId,
|
||||
timestamp: ts,
|
||||
provider: PROVIDER,
|
||||
kind: 'text',
|
||||
role: 'user',
|
||||
content: displayText,
|
||||
commandName: localCommandPayload.commandName,
|
||||
commandMessage: localCommandPayload.commandMessage,
|
||||
commandArgs: localCommandPayload.commandArgs,
|
||||
isLocalCommand: true,
|
||||
}));
|
||||
}
|
||||
return messages;
|
||||
}
|
||||
|
||||
/**
|
||||
* Local command stdout is also written as a "user" row in Claude's
|
||||
* transcript, but it is terminal output produced in response to the
|
||||
* command. Re-label it as assistant text so the chat transcript matches
|
||||
* the actual conversational flow seen by the user.
|
||||
*/
|
||||
const localCommandStdout = extractTaggedContent(text, 'local-command-stdout');
|
||||
if (localCommandStdout !== null) {
|
||||
const stdoutText = stripAnsiFormatting(localCommandStdout).trim();
|
||||
if (stdoutText) {
|
||||
messages.push(createNormalizedMessage({
|
||||
id: baseId,
|
||||
sessionId,
|
||||
timestamp: ts,
|
||||
provider: PROVIDER,
|
||||
kind: 'text',
|
||||
role: 'assistant',
|
||||
content: stdoutText,
|
||||
isLocalCommandStdout: true,
|
||||
}));
|
||||
}
|
||||
return messages;
|
||||
}
|
||||
|
||||
if (text && !isInternalContent(text)) {
|
||||
messages.push(createNormalizedMessage({
|
||||
id: baseId,
|
||||
@@ -414,7 +556,9 @@ export class ClaudeSessionsProvider implements IProviderSessions {
|
||||
|
||||
let result: ClaudeHistoryResult;
|
||||
try {
|
||||
result = await getSessionMessages(sessionId, limit, offset);
|
||||
// Load full history first so `total` reflects frontend-normalized messages,
|
||||
// not raw JSONL records.
|
||||
result = await getSessionMessages(sessionId, null, 0);
|
||||
} catch (error) {
|
||||
const message = error instanceof Error ? error.message : String(error);
|
||||
console.warn(`[ClaudeProvider] Failed to load session ${sessionId}:`, message);
|
||||
@@ -422,8 +566,6 @@ export class ClaudeSessionsProvider implements IProviderSessions {
|
||||
}
|
||||
|
||||
const rawMessages = Array.isArray(result) ? result : (result.messages || []);
|
||||
const total = Array.isArray(result) ? rawMessages.length : (result.total || 0);
|
||||
const hasMore = Array.isArray(result) ? false : Boolean(result.hasMore);
|
||||
|
||||
const toolResultMap = new Map<string, ClaudeToolResult>();
|
||||
for (const raw of rawMessages) {
|
||||
@@ -464,12 +606,31 @@ export class ClaudeSessionsProvider implements IProviderSessions {
|
||||
}
|
||||
}
|
||||
|
||||
const totalNormalized = normalized.length;
|
||||
let total = 0;
|
||||
for (const msg of normalized) {
|
||||
if (msg.kind !== 'tool_result') {
|
||||
total += 1;
|
||||
}
|
||||
}
|
||||
const normalizedOffset = Math.max(0, offset);
|
||||
const normalizedLimit = limit === null ? null : Math.max(0, limit);
|
||||
const messages = normalizedLimit === null
|
||||
? normalized
|
||||
: normalized.slice(
|
||||
Math.max(0, totalNormalized - normalizedOffset - normalizedLimit),
|
||||
Math.max(0, totalNormalized - normalizedOffset),
|
||||
);
|
||||
const hasMore = normalizedLimit === null
|
||||
? false
|
||||
: Math.max(0, totalNormalized - normalizedOffset - normalizedLimit) > 0;
|
||||
|
||||
return {
|
||||
messages: normalized,
|
||||
messages,
|
||||
total,
|
||||
hasMore,
|
||||
offset,
|
||||
limit,
|
||||
offset: normalizedOffset,
|
||||
limit: normalizedLimit,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user