45 lines
1.4 KiB
JavaScript
45 lines
1.4 KiB
JavaScript
|
const OpenAI = require('openai');
|
|||
|
|
|
|||
|
|
class OpenAIClient {
|
|||
|
|
constructor(apiKey) {
|
|||
|
|
this.api = new OpenAI({
|
|||
|
|
apiKey
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 過去のメッセージとシステムプロンプトを使って回答を生成
|
|||
|
|
async generateResponse(conversationHistory, systemPrompt) {
|
|||
|
|
try {
|
|||
|
|
const messages = [
|
|||
|
|
{
|
|||
|
|
role: 'system',
|
|||
|
|
content: systemPrompt // システムプロンプトを最初に追加
|
|||
|
|
},
|
|||
|
|
...conversationHistory // その後に会話履歴を追加
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
const response = await this.api.chat.completions.create({
|
|||
|
|
model: process.env.OPENAI_MODEL, // 使用するモデルは環境変数で定義
|
|||
|
|
messages: messages // システムプロンプトと会話履歴を渡す
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const reply = response.choices[0].message.content;
|
|||
|
|
const interval = this.calculateInterval(reply);
|
|||
|
|
|
|||
|
|
return { reply, interval }; // 返答とインターバルを返す
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('Error generating response:', error);
|
|||
|
|
throw new Error('Failed to generate response from OpenAI');
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// インターバルの計算（例: 1文字につき100ミリ秒）
|
|||
|
|
calculateInterval(reply) {
|
|||
|
|
const length = reply.length;
|
|||
|
|
const baseInterval = 100; // 1文字あたり100ms
|
|||
|
|
return Math.min(length * baseInterval, 5000); // 最大5秒までの遅延
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = OpenAIClient;
|