67 lines
2.0 KiB
TypeScript
67 lines
2.0 KiB
TypeScript
|
import { describe, expect, it } from 'vitest';
|
||
|
|
import { assistantRoleLabel } from '../../src/components/AssistantMessage';
|
||
|
|
import type { ChatMessage } from '../../src/types';
|
||
|
|
|
||
|
|
const t = () => 'Assistant';
|
||
|
|
|
||
|
|
describe('assistantRoleLabel', () => {
|
||
|
|
it('prefers the persisted assistant display name over the protocol id', () => {
|
||
|
|
const message: ChatMessage = {
|
||
|
|
id: 'message-1',
|
||
|
|
role: 'assistant',
|
||
|
|
content: '',
|
||
|
|
agentId: 'openai-api',
|
||
|
|
agentName: 'OpenAI API · google/gemma-4-e4b',
|
||
|
|
};
|
||
|
|
|
||
|
|
expect(assistantRoleLabel(message, t)).toBe('OpenAI API · google/gemma-4-e4b');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('maps API protocol ids to readable labels when no display name is saved', () => {
|
||
|
|
const message: ChatMessage = {
|
||
|
|
id: 'message-2',
|
||
|
|
role: 'assistant',
|
||
|
|
content: '',
|
||
|
|
agentId: 'openai-api',
|
||
|
|
};
|
||
|
|
|
||
|
|
expect(assistantRoleLabel(message, t)).toBe('OpenAI API');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('normalizes saved API protocol ids used as display names', () => {
|
||
|
|
const message: ChatMessage = {
|
||
|
|
id: 'message-3',
|
||
|
|
role: 'assistant',
|
||
|
|
content: '',
|
||
|
|
agentName: 'openai-api',
|
||
|
|
};
|
||
|
|
|
||
|
|
expect(assistantRoleLabel(message, t)).toBe('OpenAI API');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('preserves an explicit local agent model in the display name', () => {
|
||
|
|
const message: ChatMessage = {
|
||
|
|
id: 'message-4',
|
||
|
|
role: 'assistant',
|
||
|
|
content: '',
|
||
|
|
agentId: 'claude',
|
||
|
|
agentName: 'Claude · claude-sonnet-4-6',
|
||
|
|
};
|
||
|
|
|
||
|
|
expect(assistantRoleLabel(message, t)).toBe('Claude · claude-sonnet-4-6');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('adds the model reported by a local CLI initializing event', () => {
|
||
|
|
const message: ChatMessage = {
|
||
|
|
id: 'message-5',
|
||
|
|
role: 'assistant',
|
||
|
|
content: '',
|
||
|
|
agentId: 'claude',
|
||
|
|
agentName: 'Claude',
|
||
|
|
events: [{ kind: 'status', label: 'initializing', detail: 'claude-sonnet-4-6' }],
|
||
|
|
};
|
||
|
|
|
||
|
|
expect(assistantRoleLabel(message, t)).toBe('Claude · claude-sonnet-4-6');
|
||
|
|
});
|
||
|
|
});
|