matty/pleroma-ollama-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

don't reply when not addressed

Browse Source
This commit is contained in:
matty
2025-08-03 18:56:05 +00:00
parent b6ad54f40a
commit aaf4adcf06
2 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.env.example
View File

@ -1,6 +1,7 @@
DATABASE_URL="file:../dev.db" # SQLite database relative to the ./prisma path DATABASE_URL="file:../dev.db" # SQLite database relative to the ./prisma path
PLEROMA_INSTANCE_URL="https://instance.tld" # Pleroma instance full URL including scheme PLEROMA_INSTANCE_URL="https://instance.tld" # Pleroma instance full URL including scheme
PLEROMA_INSTANCE_DOMAIN="instance.tld" # used if you want to only want to respond to people from a particular instance PLEROMA_INSTANCE_DOMAIN="instance.tld" # used if you want to only want to respond to people from a particular instance
PLEROMA_ACCOUNT_ID="" # obtained from /api/v1/accounts/{nickname} - used so we don't spam mentions when not directly addressed
ONLY_WHITELIST="true" # change to "false" if you want to accept prompts from any and all domains - *** USE WITH CAUTION *** ONLY_WHITELIST="true" # change to "false" if you want to accept prompts from any and all domains - *** USE WITH CAUTION ***
WHITELISTED_DOMAINS="" # comma separated list of domains you want to allow the bot to accept prompts from (i.e. poa.st,nicecrew.digital,detroitriotcity.com,decayable.ink) WHITELISTED_DOMAINS="" # comma separated list of domains you want to allow the bot to accept prompts from (i.e. poa.st,nicecrew.digital,detroitriotcity.com,decayable.ink)
OLLAMA_URL="http://localhost:11434" # OLLAMA connection URL OLLAMA_URL="http://localhost:11434" # OLLAMA connection URL

17
src/main.ts
View File

@ -42,6 +42,7 @@ export const envConfig = {
adHocPostInterval: process.env.RANDOM_POST_INTERVAL adHocPostInterval: process.env.RANDOM_POST_INTERVAL
? parseInt(process.env.RANDOM_POST_INTERVAL) ? parseInt(process.env.RANDOM_POST_INTERVAL)
: 3600000, : 3600000,
botAccountId: process.env.PLEROMA_ACCOUNT_ID,
}; };
const ollamaConfig: OllamaConfigOptions = { const ollamaConfig: OllamaConfigOptions = {
@ -58,13 +59,23 @@ const ollamaConfig: OllamaConfigOptions = {
const generateOllamaRequest = async ( const generateOllamaRequest = async (
notification: Notification notification: Notification
): Promise<OllamaResponse | undefined> => { ): Promise<OllamaResponse | undefined> => {
const { whitelistOnly, ollamaModel, ollamaSystemPrompt, ollamaUrl } = const {
envConfig; whitelistOnly,
ollamaModel,
ollamaSystemPrompt,
ollamaUrl,
botAccountId,
} = envConfig;
try { try {
console.log(trimInputData(notification.status.content));
if ( if (
// striptags(notification.status.content).includes("!prompt") && // striptags(notification.status.content).includes("!prompt") &&
!notification.status.account.bot && // sanity check, sort of !notification.status.account.bot && // sanity check, sort of
notification.type === "mention" // && notification.type === "mention" &&
(notification.status.in_reply_to_account_id === botAccountId ||
notification.status.in_reply_to_account_id === null) &&
trimInputData(notification.status.content).split(" ").includes("Lexi")
// only reply to mentions when the bot is the direct recipient or when an @ is at the top level of a conversation chain, or when the AI is @ directly
// notification.status.visibility !== "private" // for safety, let's only respond to public messages // notification.status.visibility !== "private" // for safety, let's only respond to public messages
) { ) {
if (whitelistOnly && !isFromWhitelistedDomain(notification)) { if (whitelistOnly && !isFromWhitelistedDomain(notification)) {