File size: 4,490 Bytes
f56a29b | 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 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | import type { StatelessChatRequest } from '@/lib/types/chat';
// ==================== Message Conversion ====================
/**
* Convert UI messages to OpenAI format
* Includes tool call information so the model knows what actions were taken
*/
export function convertMessagesToOpenAI(
messages: StatelessChatRequest['messages'],
currentAgentId?: string,
): Array<{ role: 'system' | 'user' | 'assistant'; content: string }> {
return messages
.filter((msg) => msg.role === 'user' || msg.role === 'assistant')
.map((msg) => {
if (msg.role === 'assistant') {
// Assistant messages use JSON array format to serve as few-shot examples
// that match the expected output format from the system prompt
const items: Array<{ type: string; [key: string]: string }> = [];
if (msg.parts) {
for (const part of msg.parts) {
const p = part as Record<string, unknown>;
if (p.type === 'text' && p.text) {
items.push({ type: 'text', content: p.text as string });
} else if ((p.type as string)?.startsWith('action-') && p.state === 'result') {
const actionName = (p.actionName ||
(p.type as string).replace('action-', '')) as string;
const output = p.output as Record<string, unknown> | undefined;
const isSuccess = output?.success === true;
const resultSummary = isSuccess
? output?.data
? `result: ${JSON.stringify(output.data).slice(0, 100)}`
: 'success'
: (output?.error as string) || 'failed';
items.push({
type: 'action',
name: actionName,
result: resultSummary,
});
}
}
}
const content = items.length > 0 ? JSON.stringify(items) : '';
const msgAgentId = msg.metadata?.agentId;
// When currentAgentId is provided and this message is from a DIFFERENT agent,
// convert to user role with agent name attribution
if (currentAgentId && msgAgentId && msgAgentId !== currentAgentId) {
const agentName = msg.metadata?.senderName || msgAgentId;
return {
role: 'user' as const,
content: content ? `[${agentName}]: ${content}` : '',
};
}
return {
role: 'assistant' as const,
content,
};
}
// User messages: keep plain text concatenation
const contentParts: string[] = [];
if (msg.parts) {
for (const part of msg.parts) {
const p = part as Record<string, unknown>;
if (p.type === 'text' && p.text) {
contentParts.push(p.text as string);
} else if ((p.type as string)?.startsWith('action-') && p.state === 'result') {
const actionName = (p.actionName ||
(p.type as string).replace('action-', '')) as string;
const output = p.output as Record<string, unknown> | undefined;
const isSuccess = output?.success === true;
const resultSummary = isSuccess
? output?.data
? `result: ${JSON.stringify(output.data).slice(0, 100)}`
: 'success'
: (output?.error as string) || 'failed';
contentParts.push(`[Action ${actionName}: ${resultSummary}]`);
}
}
}
// Extract speaker name from metadata (e.g. other agents' messages in discussion)
const senderName = msg.metadata?.senderName;
let content = contentParts.join('\n');
if (senderName) {
content = `[${senderName}]: ${content}`;
}
// Annotate interrupted messages so the LLM knows context was cut short
const isInterrupted =
(msg as unknown as Record<string, unknown>).metadata &&
((msg as unknown as Record<string, unknown>).metadata as Record<string, unknown>)
?.interrupted;
return {
role: 'user' as const,
content: isInterrupted
? `${content}\n[This response was interrupted — do NOT continue it. Start a new JSON array response.]`
: content,
};
})
.filter((msg) => {
// Drop empty messages and messages with only dots/ellipsis/whitespace
// (produced by failed agent streams)
const stripped = msg.content.replace(/[.\s…]+/g, '');
return stripped.length > 0;
});
}
|