File size: 2,568 Bytes
4e23b01 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | import type { Message } from '#/message';
import { convertToolMessageContent } from '#/providers/openai-common';
import { describe, expect, test } from 'vitest';
describe('convertToolMessageContent', () => {
test('extract_text merges multiple text parts into a single string', () => {
const msg: Message = {
role: 'tool',
content: [
{ type: 'text', text: 'Result: ' },
{ type: 'text', text: '42' },
],
toolCalls: [],
toolCallId: 'tc_001',
};
const result = convertToolMessageContent(msg, 'extract_text');
expect(typeof result).toBe('string');
expect(result).toBe('Result: 42');
});
test('extract_text returns empty string for empty content', () => {
const msg: Message = {
role: 'tool',
content: [],
toolCalls: [],
toolCallId: 'tc_002',
};
const result = convertToolMessageContent(msg, 'extract_text');
expect(result).toBe('');
});
test('extract_text skips non-text content parts', () => {
const msg: Message = {
role: 'tool',
content: [
{ type: 'text', text: 'Before' },
{ type: 'image_url', imageUrl: { url: 'data:image/png;base64,xxx' } },
{ type: 'text', text: 'After' },
],
toolCalls: [],
toolCallId: 'tc_003',
};
const result = convertToolMessageContent(msg, 'extract_text');
expect(result).toBe('BeforeAfter');
});
test('null conversion returns an array of OpenAI content parts', () => {
const msg: Message = {
role: 'tool',
content: [
{ type: 'text', text: 'hello' },
{ type: 'text', text: 'world' },
],
toolCalls: [],
toolCallId: 'tc_004',
};
const result = convertToolMessageContent(msg, null);
expect(Array.isArray(result)).toBe(true);
expect(result).toHaveLength(2);
if (Array.isArray(result)) {
expect(result[0]).toMatchObject({ type: 'text', text: 'hello' });
expect(result[1]).toMatchObject({ type: 'text', text: 'world' });
}
});
test('null conversion filters out ThinkPart (returns null from convertContentPart)', () => {
const msg: Message = {
role: 'tool',
content: [
{ type: 'text', text: 'visible' },
{ type: 'think', think: 'hidden reasoning' },
],
toolCalls: [],
toolCallId: 'tc_005',
};
const result = convertToolMessageContent(msg, null);
if (Array.isArray(result)) {
expect(result).toHaveLength(1);
expect(result[0]).toMatchObject({ type: 'text', text: 'visible' });
}
});
});
|