@aituber-onair/core
Version:
Core library for AITuber OnAir providing voice synthesis and chat processing
59 lines • 2.48 kB
JavaScript
import { ENDPOINT_OPENAI_CHAT_COMPLETIONS_API, MODEL_GPT_4O_MINI, DEFAULT_SUMMARY_PROMPT_TEMPLATE, } from '@aituber-onair/chat';
import { createSummaryContext, summarizeWithFallback, } from '../summarizerUtils';
/**
* Implementation of summarization functionality using OpenAI
*/
export class OpenAISummarizer {
/**
* Constructor
* @param apiKey OpenAI API key
* @param model Name of the model to use
* @param defaultPromptTemplate Default prompt template for summarization
*/
constructor(apiKey, model = MODEL_GPT_4O_MINI, defaultPromptTemplate = DEFAULT_SUMMARY_PROMPT_TEMPLATE) {
this.apiKey = apiKey;
this.model = model;
this.defaultPromptTemplate = defaultPromptTemplate;
}
/**
* Summarize chat messages
* @param messages Array of messages to summarize
* @param maxLength Maximum number of characters (default 256)
* @param customPrompt Custom prompt template for summarization (optional)
* @returns Summarized text
*/
async summarize(messages, maxLength = 256, customPrompt) {
const { systemPrompt, conversationText } = createSummaryContext(messages, maxLength, this.defaultPromptTemplate, customPrompt);
return summarizeWithFallback(messages, async () => {
// API request
const response = await fetch(ENDPOINT_OPENAI_CHAT_COMPLETIONS_API, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${this.apiKey}`,
},
body: JSON.stringify({
model: this.model,
messages: [
{
role: 'system',
content: systemPrompt,
},
{
role: 'user',
content: conversationText,
},
],
max_completion_tokens: maxLength,
}),
});
if (!response.ok) {
const errorData = await response.json();
throw new Error(`OpenAI API error: ${errorData.error?.message || response.statusText}`);
}
const data = await response.json();
return data.choices[0]?.message?.content || '';
});
}
}
//# sourceMappingURL=OpenAISummarizer.js.map