From 191e9bf04eef0faad88ebc02ad51fdd9aded729a Mon Sep 17 00:00:00 2001
From: ChrisChrome <christophercookman@gmail.com>
Date: Thu, 13 Jul 2023 11:10:24 -0600
Subject: [PATCH] add model to session

---
 index.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/index.js b/index.js
index 7afa0ae..eca786f 100755
--- a/index.js
+++ b/index.js
@@ -178,11 +178,13 @@ client.on('messageCreate', async (message) => {
 	if (!sessions[message.channelId]) {
 		if (message.channel.nsfw) {
 			sessions[message.channelId] = {
+				model: config.discord.authorized_channels[message.channelId] || config.discord.authorized_channels[message.channel.parentId],
 				messages: [basePrompt, nsfwPrompt],
 				started: new Date(),
 			}
 		} else {
 			sessions[message.channelId] = {
+				model: config.discord.authorized_channels[message.channelId] || config.discord.authorized_channels[message.channel.parentId],
 				messages: [basePrompt],
 				started: new Date(),
 			};
@@ -217,9 +219,8 @@ client.on('messageCreate', async (message) => {
 		"role": "user"
 	});
 	// Send the message to OpenAI
-	var model = config.discord.authorized_channels[message.channelId] || config.discord.authorized_channels[message.channel.parentId];
 	await openai.createChatCompletion({
-		model: model,
+		model: sessions[message.channelId].model,
 		messages: sessions[message.channelId].messages
 	}).then((data) => {
 		output = data.data.choices[0].message;