From b6ad54f40ace21d3ded22b3d23671c6092db0c56 Mon Sep 17 00:00:00 2001
From: matty <matty@nicecrew.digital>
Date: Sun, 3 Aug 2025 14:37:23 +0000
Subject: [PATCH] way better responsiveness, better system prompt

---
 src/main.ts | 11 +++++------
 src/util.ts |  6 +++---
 2 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/src/main.ts b/src/main.ts
index 959efdd..4308dd5 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -7,7 +7,7 @@ import {
   OllamaRequest,
   OllamaResponse,
 } from "../types.js";
-import striptags from "striptags";
+// import striptags from "striptags";
 import { PrismaClient } from "../generated/prisma/client.js";
 import {
   getInstanceEmojis,
@@ -62,7 +62,7 @@ const generateOllamaRequest = async (
     envConfig;
   try {
     if (
-      striptags(notification.status.content).includes("!prompt") &&
+      // striptags(notification.status.content).includes("!prompt") &&
       !notification.status.account.bot && // sanity check, sort of
       notification.type === "mention" // &&
       // notification.status.visibility !== "private" // for safety, let's only respond to public messages
@@ -76,7 +76,7 @@ const generateOllamaRequest = async (
       }
       await recordPendingResponse(notification);
       await storeUserData(notification);
-      console.log(trimInputData(notification.status.content));
+      // console.log(trimInputData(notification.status.content));
       const ollamaRequestBody: OllamaRequest = {
         model: ollamaModel,
         // prompt: trimInputData(notification.status.content),
@@ -150,15 +150,14 @@ const createTimelinePost = async () => {
   const {
     bearerToken,
     ollamaModel,
-    // ollamaSystemPrompt,
+    ollamaSystemPrompt,
     ollamaUrl,
     pleromaInstanceUrl,
   } = envConfig;
   const ollamaRequestBody: OllamaRequest = {
     model: ollamaModel,
     prompt: "Say something random.",
-    system:
-      "You are a friendly AI assistant who loves to educate people on random topics, provide words of encouragement. You like to be as detailed as possible.",
+    system: ollamaSystemPrompt,
     stream: false,
     // options: ollamaConfig,
   };
diff --git a/src/util.ts b/src/util.ts
index cd28246..d0fbd81 100644
--- a/src/util.ts
+++ b/src/util.ts
@@ -4,10 +4,10 @@ import { envConfig } from "./main.js";
 import { Notification } from "../types.js";
 
 const trimInputData = (input: string): string => {
-  const strippedInput = striptags(input, [], "\n");
+  const strippedInput = striptags(input);
 
   const split = strippedInput.split(" ");
-  const promptStringIndex = split.indexOf("!prompt");
+  // const promptStringIndex = split.indexOf("!prompt");
   const botFqnIndex = split.indexOf("@nice-ai");
   const botFqnIndexFull = split.indexOf("@nice-ai@nicecrew.digital");
   if (botFqnIndex !== -1) {
@@ -16,7 +16,7 @@ const trimInputData = (input: string): string => {
   if (botFqnIndexFull !== -1) {
     split[botFqnIndexFull] = "Lexi";
   }
-  split.splice(promptStringIndex, 1);
+  // split.splice(promptStringIndex, 1);
   return split.join(" "); // returns everything after the !prompt
 };