Calculate number of messages after event

This commit is contained in:
Cohee 2024-06-06 00:44:59 +03:00
parent ff241dd0a9
commit 76c35d269b

View File

@ -1284,11 +1284,12 @@ export async function prepareOpenAIMessages({
} }
const chat = chatCompletion.getChat(); const chat = chatCompletion.getChat();
openai_messages_count = chat.filter(x => x?.role === 'user' || x?.role === 'assistant')?.length || 0;
const eventData = { chat, dryRun }; const eventData = { chat, dryRun };
await eventSource.emit(event_types.CHAT_COMPLETION_PROMPT_READY, eventData); await eventSource.emit(event_types.CHAT_COMPLETION_PROMPT_READY, eventData);
openai_messages_count = chat.filter(x => x?.role === 'user' || x?.role === 'assistant')?.length || 0;
return [chat, promptManager.tokenHandler.counts]; return [chat, promptManager.tokenHandler.counts];
} }