Compare commits
No commits in common. "d26c025073c157a4c26ddceaed1142dcad556686" and "cc654112b288df91543f898f47bc31993e7d4c80" have entirely different histories.
d26c025073
...
cc654112b2
1
.gitignore
vendored
1
.gitignore
vendored
@ -4,4 +4,3 @@ shared-storage
|
||||
storage_*
|
||||
storage
|
||||
chatBot/.env
|
||||
chatBot/commands/ai.js
|
@ -1,77 +0,0 @@
|
||||
import unirest from 'unirest';
|
||||
|
||||
async function chatWithAI(messages) {
|
||||
const controller = new AbortController();
|
||||
const timeout = setTimeout(() => {
|
||||
controller.abort();
|
||||
}, 900000); // 15 minutes timeout
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const req = unirest.post(`http://${process.env.ROOT_IP}:${process.env.ROOT_PORT}/v1/chat/completions`)
|
||||
.headers({
|
||||
'accept': 'application/json',
|
||||
'Content-Type': 'application/json'
|
||||
})
|
||||
.send({
|
||||
messages,
|
||||
max_tokens: Number(process.env.MAX_TOKENS), // add the max_tokens parameter here
|
||||
repeat_penalty: Number(process.env.REPEAT_PENALTY)
|
||||
})
|
||||
.end(response => {
|
||||
clearTimeout(timeout);
|
||||
if (response.error) {
|
||||
console.error(response.error);
|
||||
reject("An error occurred while processing your request. Please try again later."); // Generic error message
|
||||
} else {
|
||||
const choice = response.body.choices[0];
|
||||
resolve(choice.message.content);
|
||||
}
|
||||
});
|
||||
|
||||
controller.signal.addEventListener('abort', () => {
|
||||
req.abort();
|
||||
reject("Request timed out.");
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function getResponseMessage(content) {
|
||||
const conversation = {
|
||||
messages: [
|
||||
{ role: 'user', content: process.env.INIT_PROMPT },
|
||||
{ role: 'user', content: content }
|
||||
]
|
||||
};
|
||||
|
||||
return chatWithAI(conversation.messages).then(response => {
|
||||
if (response && response.trim()) {
|
||||
const limit = 1980;
|
||||
if (response.length > limit) {
|
||||
const chunks = response.match(new RegExp(`.{1,${limit}}`, 'g'));
|
||||
if (chunks.length >= 15) return "Response chunks too large. Try again.";
|
||||
return chunks.join('\n\n');
|
||||
} else {
|
||||
return response.replace("@", "");
|
||||
}
|
||||
} else {
|
||||
return "I didn't understand that. Can you please rephrase?"; // Generic empty response message
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
export default {
|
||||
handler: function(bot, args, message) {
|
||||
const userMessage = args.join(' ').replace('!ai', '').trim(); // Combine all args to form the user's message and remove !ai
|
||||
console.log(`Processed message: ${userMessage}`);
|
||||
|
||||
// Log environment variables for debugging
|
||||
console.log(`ROOT_IP: ${process.env.ROOT_IP}`);
|
||||
console.log(`ROOT_PORT: ${process.env.ROOT_PORT}`);
|
||||
|
||||
getResponseMessage(userMessage).then(response => {
|
||||
bot.sendMessage(response);
|
||||
}).catch(error => {
|
||||
bot.sendMessage(error);
|
||||
});
|
||||
}
|
||||
};
|
Loading…
Reference in New Issue
Block a user