diff --git a/src/main.ts b/src/main.ts index 95dc74c..3acce59 100644 --- a/src/main.ts +++ b/src/main.ts @@ -72,9 +72,7 @@ const storePromptData = async ( await prisma.response.create({ data: { response: ollamaResponseBody.response, - request: `${notification.status.account.fqn} asks: ${striptags( - notification.status.content - )}`, + request: striptags(notification.status.content), to: notification.account.fqn, pleromaNotificationId: notification.status.id, }, @@ -110,7 +108,9 @@ const generateOllamaRequest = async ( const ollamaRequestBody: OllamaRequest = { model: process.env.OLLAMA_MODEL as string, system: process.env.OLLAMA_SYSTEM_PROMPT as string, - prompt: striptags(notification.status.content), + prompt: `${notification.status.account.fqn} asks: ${striptags( + notification.status.content + )}`, stream: false, }; const response = await fetch(`${process.env.OLLAMA_URL}/api/generate`, {