yaoyao/tests/client/toolTimeline.test.ts

import { describe, expect, it } from 'vitest'
import { chatMessagesToUi } from '@/components/workspace/viewModels'
import { normalizeChatMessage } from '@/utils/normalize'

describe('ordinary chat tool timeline', () => {
  it('folds persisted function calls and their result rows into one expandable tool', () => {
    const user = normalizeChatMessage({ id: 'user-row', role: 'user', content: 'MEDIA:/tmp/keep-raw.png', timestamp: 0 }, 'session-1', 'default')
    const call = normalizeChatMessage({
      id: 'assistant-call', role: 'assistant', content: '', tool_calls: JSON.stringify([{
        id: 'call-1', type: 'function', function: { name: 'terminal', arguments: '{"command":"pwd"}' },
      }]), timestamp: 1,
    }, 'session-1', 'default')
    const result = normalizeChatMessage({
      id: 'tool-result', role: 'tool', tool_call_id: 'call-1', tool_name: 'terminal',
      content: '{"output":"/tmp/work","exit_code":0,"error":null}', timestamp: 2,
    }, 'session-1', 'default')
    const system = normalizeChatMessage({ id: 'system-row', role: 'system', content: 'transport detail', timestamp: 3 }, 'session-1', 'default')

    const timeline = chatMessagesToUi([user, call, result, system])
    expect(timeline).toHaveLength(2)
    expect(timeline[0]).toMatchObject({ role: 'user', content: 'MEDIA:/tmp/keep-raw.png' })
    expect(timeline[1].role).toBe('assistant')
    expect(timeline[1].tools).toEqual([expect.objectContaining({
      id: 'call-1', name: 'terminal', status: 'success', input: '{"command":"pwd"}', output: { output: '/tmp/work', exit_code: 0, error: null },
    })])
  })
})