Add execution graph to chat history (#776)
This commit is contained in:
Lingxuan Zuo
committed by
GitHub
Unverified
GitHub
parent
91c735c9f4
commit
c866205eac
221
tests/e2e/chat-task-visualizer.spec.ts
Normal file
221
tests/e2e/chat-task-visualizer.spec.ts
Normal file
@@ -0,0 +1,221 @@
|
||||
import { closeElectronApp, expect, getStableWindow, installIpcMocks, test } from './fixtures/electron';
|
||||
|
||||
const PROJECT_MANAGER_SESSION_KEY = 'agent:main:main';
|
||||
const CODER_SESSION_KEY = 'agent:coder:subagent:child-123';
|
||||
const CODER_SESSION_ID = 'child-session-id';
|
||||
|
||||
function stableStringify(value: unknown): string {
|
||||
if (value == null || typeof value !== 'object') return JSON.stringify(value);
|
||||
if (Array.isArray(value)) return `[${value.map((item) => stableStringify(item)).join(',')}]`;
|
||||
const entries = Object.entries(value as Record<string, unknown>)
|
||||
.sort(([left], [right]) => left.localeCompare(right))
|
||||
.map(([key, entryValue]) => `${JSON.stringify(key)}:${stableStringify(entryValue)}`);
|
||||
return `{${entries.join(',')}}`;
|
||||
}
|
||||
|
||||
const seededHistory = [
|
||||
{
|
||||
role: 'user',
|
||||
content: [{ type: 'text', text: '[Mon 2026-04-06 15:18 GMT+8] 分析 Velaria 当前未提交改动' }],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'assistant',
|
||||
content: [{
|
||||
type: 'toolCall',
|
||||
id: 'spawn-call',
|
||||
name: 'sessions_spawn',
|
||||
arguments: { agentId: 'coder', task: 'analyze core blocks' },
|
||||
}],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'toolResult',
|
||||
toolCallId: 'spawn-call',
|
||||
toolName: 'sessions_spawn',
|
||||
content: [{
|
||||
type: 'text',
|
||||
text: JSON.stringify({
|
||||
status: 'accepted',
|
||||
childSessionKey: CODER_SESSION_KEY,
|
||||
runId: 'child-run-id',
|
||||
mode: 'run',
|
||||
}, null, 2),
|
||||
}],
|
||||
details: {
|
||||
status: 'accepted',
|
||||
childSessionKey: CODER_SESSION_KEY,
|
||||
runId: 'child-run-id',
|
||||
mode: 'run',
|
||||
},
|
||||
isError: false,
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'assistant',
|
||||
content: [{
|
||||
type: 'toolCall',
|
||||
id: 'yield-call',
|
||||
name: 'sessions_yield',
|
||||
arguments: { message: '我让 coder 去拆 ~/Velaria 当前未提交改动的核心块了，等它回来我直接给你结论。' },
|
||||
}],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'toolResult',
|
||||
toolCallId: 'yield-call',
|
||||
toolName: 'sessions_yield',
|
||||
content: [{
|
||||
type: 'text',
|
||||
text: JSON.stringify({
|
||||
status: 'yielded',
|
||||
message: '我让 coder 去拆 ~/Velaria 当前未提交改动的核心块了，等它回来我直接给你结论。',
|
||||
}, null, 2),
|
||||
}],
|
||||
details: {
|
||||
status: 'yielded',
|
||||
message: '我让 coder 去拆 ~/Velaria 当前未提交改动的核心块了，等它回来我直接给你结论。',
|
||||
},
|
||||
isError: false,
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'user',
|
||||
content: [{
|
||||
type: 'text',
|
||||
text: `[Internal task completion event]
|
||||
source: subagent
|
||||
session_key: ${CODER_SESSION_KEY}
|
||||
session_id: ${CODER_SESSION_ID}
|
||||
type: subagent task
|
||||
status: completed successfully`,
|
||||
}],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'assistant',
|
||||
content: [{ type: 'text', text: '我让 coder 分析完了，下面是结论。' }],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
];
|
||||
|
||||
const childTranscriptMessages = [
|
||||
{
|
||||
role: 'user',
|
||||
content: [{ type: 'text', text: '分析 ~/Velaria 当前未提交改动的核心内容' }],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'assistant',
|
||||
content: [{
|
||||
type: 'toolCall',
|
||||
id: 'coder-exec-call',
|
||||
name: 'exec',
|
||||
arguments: {
|
||||
command: "cd ~/Velaria && git status --short && sed -n '1,200p' src/dataflow/core/logical/planner/plan.h",
|
||||
workdir: '/Users/bytedance/.openclaw/workspace-coder',
|
||||
},
|
||||
}],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'toolResult',
|
||||
toolCallId: 'coder-exec-call',
|
||||
toolName: 'exec',
|
||||
content: [{ type: 'text', text: 'M src/dataflow/core/logical/planner/plan.h' }],
|
||||
details: {
|
||||
status: 'completed',
|
||||
aggregated: "M src/dataflow/core/logical/planner/plan.h\nM src/dataflow/core/execution/runtime/execution_optimizer.cc",
|
||||
cwd: '/Users/bytedance/.openclaw/workspace-coder',
|
||||
},
|
||||
isError: false,
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
{
|
||||
role: 'assistant',
|
||||
content: [{ type: 'text', text: '已完成分析，最关键的有 4 块。' }],
|
||||
timestamp: Date.now(),
|
||||
},
|
||||
];
|
||||
|
||||
test.describe('ClawX chat execution graph', () => {
|
||||
test('renders internal yield status and linked subagent branch from mocked IPC', async ({ launchElectronApp }) => {
|
||||
const app = await launchElectronApp({ skipSetup: true });
|
||||
|
||||
try {
|
||||
await installIpcMocks(app, {
|
||||
gatewayStatus: { state: 'running', port: 18789, pid: 12345 },
|
||||
gatewayRpc: {
|
||||
[stableStringify(['sessions.list', {}])]: {
|
||||
success: true,
|
||||
result: {
|
||||
sessions: [{ key: PROJECT_MANAGER_SESSION_KEY, displayName: 'main' }],
|
||||
},
|
||||
},
|
||||
[stableStringify(['chat.history', { sessionKey: PROJECT_MANAGER_SESSION_KEY, limit: 200 }])]: {
|
||||
success: true,
|
||||
result: {
|
||||
messages: seededHistory,
|
||||
},
|
||||
},
|
||||
[stableStringify(['chat.history', { sessionKey: PROJECT_MANAGER_SESSION_KEY, limit: 1000 }])]: {
|
||||
success: true,
|
||||
result: {
|
||||
messages: seededHistory,
|
||||
},
|
||||
},
|
||||
},
|
||||
hostApi: {
|
||||
[stableStringify(['/api/gateway/status', 'GET'])]: {
|
||||
ok: true,
|
||||
data: {
|
||||
status: 200,
|
||||
ok: true,
|
||||
json: { state: 'running', port: 18789, pid: 12345 },
|
||||
},
|
||||
},
|
||||
[stableStringify(['/api/agents', 'GET'])]: {
|
||||
ok: true,
|
||||
data: {
|
||||
status: 200,
|
||||
ok: true,
|
||||
json: {
|
||||
success: true,
|
||||
agents: [
|
||||
{ id: 'main', name: 'main' },
|
||||
{ id: 'coder', name: 'coder' },
|
||||
],
|
||||
},
|
||||
},
|
||||
},
|
||||
[stableStringify([`/api/sessions/transcript?agentId=coder&sessionId=${CODER_SESSION_ID}`, 'GET'])]: {
|
||||
ok: true,
|
||||
data: {
|
||||
status: 200,
|
||||
ok: true,
|
||||
json: {
|
||||
success: true,
|
||||
messages: childTranscriptMessages,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
const page = await getStableWindow(app);
|
||||
await page.reload();
|
||||
await expect(page.getByTestId('main-layout')).toBeVisible();
|
||||
await expect(page.getByTestId('chat-execution-graph')).toBeVisible({ timeout: 30_000 });
|
||||
await expect(
|
||||
page.locator('[data-testid="chat-execution-graph"] [data-testid="chat-execution-step"]').getByText('sessions_yield', { exact: true }),
|
||||
).toBeVisible();
|
||||
await expect(page.getByText('coder subagent')).toBeVisible();
|
||||
await expect(
|
||||
page.locator('[data-testid="chat-execution-graph"] [data-testid="chat-execution-step"]').getByText('exec', { exact: true }),
|
||||
).toBeVisible();
|
||||
await expect(page.locator('[data-testid="chat-execution-graph"]').getByText('我让 coder 去拆 ~/Velaria 当前未提交改动的核心块了，等它回来我直接给你结论。')).toBeVisible();
|
||||
} finally {
|
||||
await closeElectronApp(app);
|
||||
}
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user