@aituber-onair/core
Version:
Core library for AITuber OnAir providing voice synthesis and chat processing
70 lines • 2.9 kB
JavaScript
import { ENDPOINT_OPENAI_CHAT_COMPLETIONS_API, MODEL_GPT_4O_MINI, } from '../../../../constants';
import { DEFAULT_SUMMARY_PROMPT_TEMPLATE } from '../../../../constants';
/**
* 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) {
try {
// Create system prompt
const promptTemplate = customPrompt || this.defaultPromptTemplate;
const systemPrompt = promptTemplate.replace('{maxLength}', maxLength.toString());
// Join message content
const conversationText = messages
.map((msg) => `${msg.role}: ${msg.content}`)
.join('\n');
// 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_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 || '';
}
catch (error) {
console.error('Error in summarize:', error);
// Error fallback - simple summary
return `${messages.length} messages. Latest topic: ${messages[messages.length - 1]?.content.substring(0, 50) || 'none'}...`;
}
}
}
//# sourceMappingURL=OpenAISummarizer.js.map