Ability to like messages

Author: dahlia

README.md

@@ -6,7 +6,7 @@ FediChatBot
 [![GitHub Actions][GitHub Actions badge]][GitHub Actions]
 
 FediChatBot is an LLM-powered chatbot for fediverse, powered by [BotKit] and
-[Gemini 2.0 Flash (experimental)].  It consists of only about 300 lines of code
+[Gemini 2.0 Flash (experimental)].  It consists of only about 350 lines of code
 (except for prompts)!
 
 Please follow `@FediChatBot@fedichatbot.deno.dev` on your fediverse instance to

bot.ts

@@ -47,6 +47,7 @@ import { join } from "@std/path/join";
 import { AsyncLocalStorage } from "node:async_hooks";
 import { detect } from "tinyld";
 import { FilterXSS } from "xss";
+import { z } from "zod";
 import metadata from "./deno.json" with { type: "json" };
 
 await configure({
@@ -118,13 +119,27 @@ bot.onFollow = async (session, followRequest) => {
   );
 };
 
+const reactionResponseSchema = z.object({
+  whetherToLike: z.boolean().describe(
+    "Whether to press the like button on the message.",
+  ),
+});
+
 bot.onMention = async (session, msg) => {
   if (msg.replyTarget != null) return;
   const actor = msg.actor;
+  const llmWithStruct = llm.withStructuredOutput(reactionResponseSchema);
+  const whetherToLike = await llmWithStruct.invoke([
+    getSystemMessage(session),
+    await getIntroMessage(session, actor, await getMentionPrompt(actor)),
+    await getHumanMessage(msg, true),
+  ]);
+  logger.debug("Whether to like: {whetherToLike}", whetherToLike);
+  if (whetherToLike.whetherToLike) await msg.like();
   const response = await llm.invoke([
     getSystemMessage(session),
     await getIntroMessage(session, actor, await getMentionPrompt(actor)),
-    await getHumanMessage(msg),
+    await getHumanMessage(msg, false),
   ]);
   const message = response.content.toString();
   const language = detect(message);
@@ -153,9 +168,16 @@ bot.onReply = async (session, msg) => {
   for (const msg of thread) {
     const message = msg.actor?.id?.href === session.actorId.href
       ? new AIMessage(msg.text)
-      : await getHumanMessage(msg);
+      : await getHumanMessage(msg, false);
     messages.push(message);
   }
+  const llmWithStruct = llm.withStructuredOutput(reactionResponseSchema);
+  const whetherToLike = await llmWithStruct.invoke([
+    ...messages.slice(0, messages.length - 1),
+    await getHumanMessage(thread[thread.length - 1], true),
+  ]);
+  if (whetherToLike.whetherToLike) await msg.like();
+  logger.debug("Whether to like: {whetherToLike}", whetherToLike);
   const response = await llm.invoke(messages);
   const message = response.content.toString();
   const language = detect(message);
@@ -183,6 +205,9 @@ const FOLLOW_PROMPT_TEMPLATE = await Deno.readTextFile(
 const MENTION_PROMPT_TEMPLATE = await Deno.readTextFile(
   join(import.meta.dirname!, "prompts", "mention.txt"),
 );
+const REACTION_PROMPT_TEMPLATE = await Deno.readTextFile(
+  join(import.meta.dirname!, "prompts", "reaction.txt"),
+);
 
 const template = new Template({ isEscape: false });
 
@@ -224,6 +249,15 @@ async function getFollowPrompt(actor: Actor): Promise<string> {
   });
 }
 
+function getReactionPrompt(message: Message<MessageClass, void>): string {
+  const msg = message.text;
+  return template.render(REACTION_PROMPT_TEMPLATE, {
+    quotedMessage: msg == null
+      ? "Not available."
+      : `> ${msg.replaceAll("\n", "\n> ")}`,
+  });
+}
+
 async function getIntroMessage(
   session: Session<void>,
   actor: Actor,
@@ -281,6 +315,7 @@ async function getMentionPrompt(actor: Actor): Promise<string> {
 
 async function getHumanMessage<T extends MessageClass>(
   msg: Message<T, void>,
+  reaction: boolean,
 ): Promise<HumanMessage> {
   const attachments = msg.attachments.map(async (doc) => {
     if (!doc.mediaType?.startsWith("image/")) return null;
@@ -293,7 +328,7 @@ async function getHumanMessage<T extends MessageClass>(
   });
   return new HumanMessage({
     content: [
-      { type: "text", text: msg.text },
+      { type: "text", text: reaction ? getReactionPrompt(msg) : msg.text },
       ...(await Promise.all(attachments)).filter((a) => a != null),
     ],
   });

deno.json

@@ -5,7 +5,7 @@
   "exports": "./bot.ts",
   "imports": {
     "@deno-library/template": "jsr:@deno-library/template@^0.2.1",
-    "@fedify/botkit": "jsr:@fedify/botkit@^0.1.0-dev.63+f54fee0a",
+    "@fedify/botkit": "jsr:@fedify/botkit@^0.1.0-dev.67+94b4841c",
     "@fedify/fedify": "jsr:@fedify/fedify@^1.3.3",
     "@kitsonk/kv-toolbox": "jsr:@kitsonk/kv-toolbox@^0.25.0",
     "@langchain/core": "npm:@langchain/core@^0.3.28",
@@ -15,7 +15,8 @@
     "@std/encoding": "jsr:@std/encoding@^1.0.6",
     "@std/path": "jsr:@std/path@^1.0.8",
     "tinyld": "npm:tinyld@^1.3.4",
-    "xss": "npm:xss@^1.0.15"
+    "xss": "npm:xss@^1.0.15",
+    "zod": "npm:zod@^3.24.1"
   },
   "unstable": [
     "kv",

deno.lock

@@ -0,0 +1,3 @@
+Would you like to press the like button to the following message?
+
+{{quotedMessage}}