Refactor (5)

This commit is contained in:
2026-02-24 22:21:29 +09:00
parent c986bb861a
commit 9356ae18bc
2 changed files with 17 additions and 5 deletions

View File

@@ -1,10 +1,16 @@
import { parseArgs } from "node:util"; import { parseArgs } from "node:util";
import { Stream } from "misskey-js"; import { Stream } from "misskey-js";
import type { Note } from "misskey-js/entities.js"; import type { Note } from "misskey-js/entities.js";
import { LlmSession, createGrammar, getModel, parseResponse } from "./lib/llm";
import { expandReplyTree, getNotes, me, misskey } from "./lib/misskey";
import { sleep } from "./lib/util";
import type { ChatHistoryItem, LLamaChatPromptOptions } from "node-llama-cpp"; import type { ChatHistoryItem, LLamaChatPromptOptions } from "node-llama-cpp";
import { LlmSession, createGrammar, getModel, parseResponse } from "./lib/llm";
import {
expandReplyTree,
getNotes,
me,
misskey,
sanitizeText,
} from "./lib/misskey";
import { sleep } from "./lib/util";
const { values } = parseArgs({ const { values } = parseArgs({
args: Bun.argv, args: Bun.argv,
@@ -110,7 +116,7 @@ async function processPostJob() {
if (values.test) return; if (values.test) return;
await misskey.request("notes/create", { await misskey.request("notes/create", {
visibility: "public", visibility: "public",
text: rephrased, text: sanitizeText(rephrased),
}); });
} }
} }
@@ -142,7 +148,7 @@ async function processReplyJob(job: Extract<Job, { type: "reply" }>) {
if (values.test) return; if (values.test) return;
await misskey.request("notes/create", { await misskey.request("notes/create", {
visibility: job.visibility, visibility: job.visibility,
text: rephrased, text: sanitizeText(rephrased),
replyId: job.id, replyId: job.id,
}); });
} }

View File

@@ -72,3 +72,9 @@ export async function expandReplyTree(
} }
return { last: current, history: history.reverse() }; return { last: current, history: history.reverse() };
} }
export const sanitizeText = (text: string) =>
text
.replaceAll(/(\r\n|\r|\n)\s+/g, "\n\n") // remove extra newlines
.replaceAll("@", "") // remove mentions
.replaceAll("#", ""); // remove hashtags