kar
/
iimtt-api
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects 1 Releases Wiki Activity

only the two most recent messages for context

galleryApi
Kar 2024-11-11 12:27:23 +05:30
parent 5e23f73645
commit 02ec79b262
1 changed files with 65 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

116
src/routes/api/aiFollowupQuestion.js
View File

@ -1,5 +1,4 @@
const aiFollowupQuestion = async (req, res) => { const aiFollowupQuestion = async (req, res) => {
const { MongoClient } = require('mongodb'); const { MongoClient } = require('mongodb');
const fetch = require('node-fetch'); const fetch = require('node-fetch');
const { v4: uuidv4 } = require('uuid'); const { v4: uuidv4 } = require('uuid');
@ -29,63 +28,72 @@ const aiFollowupQuestion = async (req, res) => {
return data; return data;
} }
try {
try { const { prompt, sessionId, model = "gpt-4o-mini", max_tokens = 200 } = req.body;
const { prompt, sessionId, model = "gpt-4o-mini", max_tokens = 200 } = req.body; if (!conversationsCollection) {
if (!conversationsCollection) { return res.status(500).json({
return res.status(500).json({
success: false,
error: 'MongoDB is not connected yet. Please try again later.',
});
}
let conversation;
if (!sessionId) {
const newSessionId = uuidv4();
conversation = {
sessionId: newSessionId,
conversationHistory: [
{ role: 'system', content: 'answer within 50 words' },
{ role: 'user', content: prompt },
],
};
await conversationsCollection.insertOne(conversation);
} else {
conversation = await conversationsCollection.findOne({ sessionId: sessionId });
if (!conversation) {
return res.status(400).json({
success: false,
error: 'Invalid session ID.',
});
}
conversation.conversationHistory.push({ role: 'user', content: prompt });
}
const aiResponse = await fetchOpenAICompletion(prompt, conversation.conversationHistory, model, max_tokens);
conversation.conversationHistory.push({ role: 'assistant', content: aiResponse.choices[0].message.content });
await conversationsCollection.updateOne(
{ sessionId: conversation.sessionId },
{ $set: { conversationHistory: conversation.conversationHistory } }
);
res.json({
success: true,
data: aiResponse.choices[0].message.content,
total_tokens: aiResponse.usage.total_tokens,
sessionId: conversation.sessionId,
});
} catch (error) {
console.error('Error generating response:', error.message);
res.status(500).json({
success: false, success: false,
error: 'Something went wrong. Please try again later.', error: 'MongoDB is not connected yet. Please try again later.',
}); });
} }
let conversation;
if (!sessionId) {
const newSessionId = uuidv4();
// New conversation, start with system and user message
conversation = {
sessionId: newSessionId,
conversationHistory: [
{ role: 'system', content: 'answer within 50 words' },
{ role: 'user', content: prompt },
],
};
await conversationsCollection.insertOne(conversation);
} else {
// Existing conversation, find it by sessionId
conversation = await conversationsCollection.findOne({ sessionId: sessionId });
if (!conversation) {
return res.status(400).json({
success: false,
error: 'Invalid session ID.',
});
}
// Keep only the last 2 messages (1 user and 1 assistant)
conversation.conversationHistory.push({ role: 'user', content: prompt });
// Limit conversation history to the last two entries: one user, one assistant
const lastTwoMessages = conversation.conversationHistory.slice(-2);
conversation.conversationHistory = lastTwoMessages;
}
// Fetch AI response based on the last two messages (user + assistant)
const aiResponse = await fetchOpenAICompletion(prompt, conversation.conversationHistory, model, max_tokens);
// Add the AI response to conversation history
conversation.conversationHistory.push({ role: 'assistant', content: aiResponse.choices[0].message.content });
// Update conversation in the database with new history
await conversationsCollection.updateOne(
{ sessionId: conversation.sessionId },
{ $set: { conversationHistory: conversation.conversationHistory } }
);
res.json({
success: true,
data: aiResponse.choices[0].message.content,
total_tokens: aiResponse.usage.total_tokens,
sessionId: conversation.sessionId,
});
} catch (error) {
console.error('Error generating response:', error.message);
res.status(500).json({
success: false,
error: 'Something went wrong. Please try again later.',
});
} }
}
module.exports = aiFollowupQuestion; module.exports = aiFollowupQuestion;