nodebb-plugin-openai 1.4.2 → 1.4.4

package/languages/en-GB/openai.json

@@ -2,6 +2,7 @@
 	"error.need-x-reputation-to-mention": "You need at least %1 reputation for mentioning chatgpt",
 	"summarize-topic": "Summarize Topic",
 	"topic-summary": "Topic Summary",
+	"summary-cooldown-error": "You must wait before summarizing a topic again.",
 
 	"admin.general": "General",
 	"admin.apikey": "API Key",

package/languages/en-US/openai.json

@@ -2,6 +2,7 @@
 	"error.need-x-reputation-to-mention": "You need at least %1 reputation for mentioning chatgpt",
 	"summarize-topic": "Summarize Topic",
 	"topic-summary": "Topic Summary",
+	"summary-cooldown-error": "You must wait before summarizing a topic again.",
 
 	"admin.general": "General",
 	"admin.apikey": "API Key",

package/languages/he/openai.json

@@ -2,6 +2,7 @@
 	"error.need-x-reputation-to-mention": "דרושה לך מוניטין של %1 לפחות כדי לאזכר את chatgpt",
 	"summarize-topic": "סכם נושא",
 	"topic-summary": "סיכום הנושא",
+	"summary-cooldown-error": "עליך להמתין לפני שתסכם את הנושא שוב.",
 
 	"admin.general": "כללי",
 	"admin.apikey": "מפתח API",

package/languages/zh-CN/openai.json

@@ -1,5 +1,6 @@
 {
 	"error.need-x-reputation-to-mention": "提及 chatgpt 至少需要 %1 声望值",
 	"summarize-topic": "总结主题",
-	"topic-summary": "主题摘要"
+	"topic-summary": "主题摘要",
+	"summary-cooldown-error": "在再次总结主题之前，您必须等待。"
 }

package/library.js

@@ -18,6 +18,11 @@ const groups = require.main.require('./src/groups');
 const sockets = require.main.require('./src/socket.io');
 const socketPlugins = require.main.require('./src/socket.io/plugins');
 const summary = require('./lib/summary');
+const cacheCreate = require('../cache/lru');
+const summaryCache = cacheCreate({
+	name: 'openai-summary',
+	max: 200,
+});
 
 const plugin = module.exports;
 
@@ -26,7 +31,7 @@ const defaults = {
 	'chatgpt-username': '',
 	enablePrivateMessages: 'off',
 	model: 'gpt-3.5-turbo',
-	minimumReputation: 0,
+	minimumReputation: 2,
 	allowedGroups: '[]',
 	systemPrompt: 'You are a helpful assistant',
 };
@@ -125,9 +130,13 @@ plugin.actionMessagingSave = async function (hookData) {
 			return;
 		}
 
+		// Check actual room membership directly against the database.
+		// messaging.isUserInRoom() runs filter:messaging.isUserInRoom, which
+		// moderation plugins use to grant virtual access to every room — that
+		// would make the bot react to every private chat on the forum.
 		const [roomData, inRoom] = await Promise.all([
 			messaging.getRoomData(roomId),
-			messaging.isUserInRoom(chatgptUid, roomId),
+			db.isSortedSetMember(`chat:room:${roomId}:uids`, chatgptUid),
 		]);
 
 		if (!roomData || !inRoom) {
@@ -145,24 +154,19 @@ plugin.actionMessagingSave = async function (hookData) {
 		}
 		let conversation = [{ role: 'user', content: message.content }];
 		if (isPrivate) {
-			const mids = await getMessageIds(roomId, chatgptUid, 0, 20);
-
-			// dont allow chats to get too long
-			if (mids.length > 20) {
-				await api.chats.post({ uid: chatgptUid, session: {} }, {
-					roomId,
-					message: 'Conversation too long, please start a new chat',
-					toMid: message.mid,
-				});
-				return;
-			}
+			// keep context bounded to the 20 most recent messages
+			const mids = await getMessageIds(roomId, chatgptUid, 0, 19);
 			let messages = await messaging.getMessagesFields(mids, ['fromuid', 'content', 'system']);
-			messages = messages.filter(m => m && !m.system);
+			messages = messages.filter(m => m && !m.system && m.content);
 			conversation = messages.map(
-				msg => ({ role: msg.fromuid === chatgptUid ? 'assistant' : 'user', content: msg.content })
+				msg => ({ role: parseInt(msg.fromuid, 10) === parseInt(chatgptUid, 10) ? 'assistant' : 'user', content: msg.content })
 			);
 		}
 
+		if (!conversation.length) {
+			return;
+		}
+
 		const response = await chatComplete(conversation);
 
 		if (response) {
@@ -227,17 +231,23 @@ async function checkGroupMembership(uid, settings, silent) {
 	return memberOfAny;
 }
 
+// Returns the newest messages (like core's getMessageIds), in chronological order
 async function getMessageIds(roomId, uid, start, stop) {
 	const isPublic = await db.getObjectField(`chat:room:${roomId}`, 'public');
 	if (parseInt(isPublic, 10) === 1) {
-		return await db.getSortedSetRange(
+		const mids = await db.getSortedSetRevRange(
 			`chat:room:${roomId}:mids`, start, stop
 		);
+		return mids.reverse();
 	}
 	const userjoinTimestamp = await db.sortedSetScore(`chat:room:${roomId}:uids`, uid);
-	return await db.getSortedSetRangeByScore(
-		`chat:room:${roomId}:mids`, start, stop === -1 ? -1 : stop - start + 1, userjoinTimestamp, '+inf'
+	if (userjoinTimestamp === null) {
+		return [];
+	}
+	const mids = await db.getSortedSetRevRangeByScore(
+		`chat:room:${roomId}:mids`, start, stop === -1 ? -1 : stop - start + 1, '+inf', userjoinTimestamp
 	);
+	return mids.reverse();
 }
 
 async function chatComplete(messages) {
@@ -325,16 +335,22 @@ socketPlugins.openai.summarizeTopic = async function (socket, data) {
 	if (!await privileges.topics.can('topics:read', tid, socket.uid)) {
 		throw new Error('[[error:no-privileges]]');
 	}
-	const settings = await getSettings();
-	if (!await canUseOpenAI(socket.uid, settings)) {
-		return;
-	}
 
 	let openaiSummary = await topics.getTopicField(tid, 'openai:summary');
 	if (openaiSummary) {
 		return openaiSummary;
 	}
+	const settings = await getSettings();
+	if (!await canUseOpenAI(socket.uid, settings)) {
+		return;
+	}
+	const cooldown = summaryCache.get(`uid:${socket.uid}`);
+	if (cooldown) {
+		throw new Error('[[openai:summary-cooldown-error]]');
+	}
+
 	openaiSummary = await summary.summarizeTopic(tid, openai, settings);
+	summaryCache.set(`uid:${socket.uid}`, 1, 60000);
 	await topics.setTopicField(tid, 'openai:summary', openaiSummary);
 	return openaiSummary;
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nodebb-plugin-openai",
-  "version": "1.4.2",
+  "version": "1.4.4",
   "description": "A NodeBB OpenAI plugin",
   "main": "library.js",
   "repository": {