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

initial commit of my fork of the code
Some checks failed
Release / Release (push) Has been cancelled
Details
Release / Docker (push) Has been cancelled
Details
Release / Update develop branch (push) Has been cancelled
Details
Release / Docker pre-release (push) Has been cancelled
Details

Browse Source
This commit is contained in:
cybrneko 2025-05-10 00:19:54 -05:00
parent ba6f93dc96
commit cea0ba39bc
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
bot.py
View File

@ -116,7 +116,7 @@ class Bot:
# pull the model # pull the model
# await self.ollama.pull(self.model) # await self.ollama.pull(self.model)
activity = discord.Activity(name='Status', state=f'Hi, I\'m {self.bot_name.title()}! I only respond to mentions.', type=discord.ActivityType.custom) activity = discord.Activity(name='Status', state=f'Hi, I\'m {self.bot_name.title()}. Don\'t talk to me.', type=discord.ActivityType.custom)
await self.discord.change_presence(activity=activity) await self.discord.change_presence(activity=activity)
logging.info( logging.info(
@ -168,19 +168,19 @@ class Bot:
# Do not respond with llm in private messages # Do not respond with llm in private messages
if isinstance(message.channel, discord.DMChannel): if isinstance(message.channel, discord.DMChannel):
response = DiscordResponse(message) response = DiscordResponse(message)
if self.discord.user.mentioned_in(message): if self.discord.user.mentioned_in(message):
await response.write(message, 'I am sorry, I am unable to respond in private messages.') await response.write(message, 'I am sorry, I am unable to respond in private messages.')
return return
# Do not respond to messages that don't mention us # Do not respond to messages that don't mention us. 0.2 is 20 percent chance to answer anyway.
if not self.discord.user.mentioned_in(message) or message.author.bot or '@everyone' in message.content or '@here' in message.content: if not self.discord.user.mentioned_in(message) or message.author.bot or '@everyone' in message.content or '@here' in message.content:
# don't respond to messages that don't mention us, but save it for context # don't respond to messages that don't mention us, but save it for context
await self.save_message(str(message.channel.id), self.message(message, message.content), 'user') await self.save_message(str(message.channel.id), self.message(message, message.content), 'user')
logging.info('Message saved for context in %s, but it was not for us', (message.channel.id)) logging.info('Message saved for context in %s, but it was not for us', (message.channel.id))
# However, if randomly it does accept the message, and respond. There is a 0.01% chance of it happening. # However, if randomly it does accept the message, and respond. There is a 0.01% chance of it happening.
if (random.random() * 1000) > 0.1: if (random.random()* 1000) > 0.1:
return return
# Clean message # Clean message

2
docker-compose.yaml
View File

@ -44,7 +44,7 @@ services:
image: redis/redis-stack-server:latest image: redis/redis-stack-server:latest
restart: always restart: always
volumes: volumes:
- redis:/data - /portainer/Files/AppData/Config/discordluna/redis:/data
networks: networks:
- redis - redis
expose: expose: