Refactor

lib/misskey.ts

@@ -0,0 +1,56 @@
+import { api } from "misskey-js";
+import type { Note } from "misskey-js/entities.js";
+import { sample } from "./util";
+import type { Message } from "./llm";
+
+export const misskey = new api.APIClient({
+  origin: Bun.env["MISSKEY_ORIGIN"] || "https://misskey.cannorin.net",
+  credential: Bun.env["MISSKEY_CREDENTIAL"],
+});
+
+export const me = await misskey.request("i", {});
+
+/** check if a note is suitable as an input */
+export const isSuitableAsInput = (n: Note) =>
+  !n.user.isBot &&
+  !n.replyId &&
+  (!n.mentions || n.mentions.length === 0) &&
+  n.text?.length &&
+  n.text.length > 0;
+
+/** randomly sample some notes from the timeline */
+export async function getNotes() {
+  // randomly sample N local notes
+  const localNotes = (count: number) =>
+    misskey
+      .request("notes/local-timeline", { limit: 100 })
+      .then((xs) => xs.filter(isSuitableAsInput))
+      .then((xs) => sample(xs, count));
+
+  // randomly sample N global notes
+  const globalNotes = (count: number) =>
+    misskey
+      .request("notes/global-timeline", { limit: 100 })
+      .then((xs) => xs.filter(isSuitableAsInput))
+      .then((xs) => sample(xs, count));
+
+  const notes = await Promise.all([localNotes(5), globalNotes(10)]);
+  return sample(notes.flat());
+}
+
+/** fetch the whole reply tree */
+export async function expandReplyTree(
+  note: Note,
+  acc: Note[] = [],
+  cutoff = 5,
+) {
+  if (!note.reply || cutoff < 1) return [...acc, note];
+  const reply = await misskey.request("notes/show", { noteId: note.reply.id });
+  return await expandReplyTree(reply, [...acc, note], cutoff - 1);
+}
+
+/** convert a note to a chat message */
+export const noteToMessage = (note: Note): Message => ({
+  type: note.userId === me.id ? ("model" as const) : ("user" as const),
+  text: note.text?.replaceAll(`@${me.username}`, "") || "",
+});