commit

libs/mysql.js

@@ -167,4 +167,44 @@ async function isNewUser(id, username) {
     });
 }
 
-module.exports = { addUserToDb, incrementQuota, usersInDb, getQuota, addConv, delConv, getConvs, addMessage, getMessages, isNewUser };
+async function listchannels() {
+    return new Promise((resolve, reject) => {
+        connection.query('SELECT id FROM channels', (error, results, fields) => {
+            if (error) {
+                reject(error);
+            } else {
+                channels = [];
+                results.forEach(element => {
+                    channels.push(element.id);
+                });
+                resolve(channels);
+            }
+        });
+    });
+}
+
+async function addChannel(id) {
+    return new Promise((resolve, reject) => {
+        connection.query('INSERT INTO channels (id) VALUES (' + id + ')', (error, results, fields) => {
+            if (error) {
+                reject(error);
+            } else {
+                resolve();
+            }
+        });
+    });
+}
+
+async function deleteChannel(id) {
+    return new Promise((resolve, reject) => {
+        connection.query('DELETE FROM channels WHERE id = ' + id, (error, results, fields) => {
+            if (error) {
+                reject(error);
+            } else {
+                resolve();
+            }
+        });
+    });
+}
+
+module.exports = { addUserToDb, incrementQuota, usersInDb, getQuota, addConv, delConv, getConvs, addMessage, getMessages, isNewUser, listchannels, addChannel, deleteChannel };

libs/openAi.js

@@ -60,4 +60,17 @@ async function sendConv (messages) {
   return response;
 }
 
-module.exports = { generateImage, answerQuestion, sendConv, quickAnswer };
+async function sendQuickConv (messages) {
+  response = await openai.createChatCompletion({
+    model: "gpt-3.5-turbo",
+    messages: messages,
+    temperature: 0.9,
+  }).catch((err) => {
+    console.log(err);
+    addToLogs("--> error : " + err);
+  })
+  
+  return response;
+}
+
+module.exports = { generateImage, answerQuestion, sendConv, quickAnswer, sendQuickConv };