aslan/aslobot-matrix
SHA256
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for replied messages

Browse source
This commit is contained in:
Aslan 2025-12-29 00:33:52 +01:00
parent 6264567ec9
commit 791883f580
5 changed files with 46 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/modules/ai/ai.ts
View file

@ -28,7 +28,12 @@ const registerModuleAI = (
});
};
const onAI = async (text: string, roomId: string, sender: string) => {
const onAI = async (
text: string,
roomId: string,
sender: string,
repliedMessage?: string,
) => {
const user = getUserById(sender);
if (text.trim().length < 5) {
@ -66,9 +71,13 @@ const onAI = async (text: string, roomId: string, sender: string) => {
} as IAIInstructions;
const username = getUserName(user);
textMod = `${username}:\n${textMod}`;
textMod = `${username}: ${textMod}`;
const responseAI = await getTextGemini(instructions, textMod);
const responseAI = await getTextGemini(
instructions,
textMod,
repliedMessage,
);
user.aiCost += responseAI.tokens * prices.text;