scriptos/ollamarama-matrix-chatbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

various updates

Browse Source
This commit is contained in:
Dustin 2023-12-22 16:19:40 -05:00
parent 3947fa222b
commit a4f7eb5279
1 changed files with 51 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
ollamarama.py
View File

@ -16,6 +16,7 @@ class ollamarama:
self.username = username self.username = username
self.password = password self.password = password
self.channels = channels self.channels = channels
self.default_personality = personality
self.personality = personality self.personality = personality
self.client = AsyncClient(server, username) self.client = AsyncClient(server, username)
@ -139,7 +140,7 @@ class ollamarama:
except Exception as e: except Exception as e:
print(e) print(e)
#Shrink history list for token size management #Shrink history list for token size management
if len(self.messages[channel][sender]) > 30: if len(self.messages[channel][sender]) > 24:
del self.messages[channel][sender][1:3] #delete the first set of question and answers del self.messages[channel][sender][1:3] #delete the first set of question and answers
# change the personality of the bot # change the personality of the bot
@ -219,16 +220,30 @@ Available models: {', '.join(sorted(list(self.models)))}''')
self.admins.remove(nick) self.admins.remove(nick)
await self.send_message(room_id, f"{nick} removed from admins") await self.send_message(room_id, f"{nick} removed from admins")
#set new global personality
if message.startswith(".gpersona "):
m = message.split(" ", 1)[1]
if m != None:
if m == 'reset':
self.personality = self.default_personality
else:
self.personality = m.strip()
await self.send_message(room_id, f"Global personality set to {self.personality}")
#remove personality globally
if message == ".gstock":
pass #i'll figure this out later
# main AI response functionality # main AI response functionality
if message.startswith(".ai ") or message.startswith(self.bot_id): if message.startswith(".ai ") or message.startswith(self.bot_id):
m = message.split(" ", 1) if message != ".ai reset":
try: m = message.split(" ", 1)
m = m[1] + " [your response must be one paragraph or less]" try:
await self.add_history("user", room_id, sender, m) m = m[1] + " [your response must be one paragraph or less]"
await self.respond(room_id, sender, self.messages[room_id][sender]) await self.add_history("user", room_id, sender, m)
except: await self.respond(room_id, sender, self.messages[room_id][sender])
pass except:
pass
# collaborative functionality # collaborative functionality
if message.startswith(".x "): if message.startswith(".x "):
m = message.split(" ", 2) m = message.split(" ", 2)
@ -265,7 +280,7 @@ Available models: {', '.join(sorted(list(self.models)))}''')
await self.respond(room_id, sender, self.messages[room_id][sender]) await self.respond(room_id, sender, self.messages[room_id][sender])
# reset bot to default personality # reset bot to default personality
if message.startswith(".reset"): if message.startswith(".reset") or message == ".ai reset": #some users keep forgetting the correct command
if room_id in self.messages: if room_id in self.messages:
if sender in self.messages[room_id]: if sender in self.messages[room_id]:
self.messages[room_id][sender].clear() self.messages[room_id][sender].clear()
@ -314,7 +329,33 @@ f'''{self.bot_id}, an AI chatbot.
.stock .stock
Remove personality and reset to standard model settings Remove personality and reset to standard model settings
Available at https://github.com/h1ddenpr0cess20/ollamarama-matrix
''') ''')
if sender_display in self.admins:
await self.send_message(room_id, '''Admin commands:
.admins
List of users authorized to use these commands
.models
List available models
.model <model>
Change the model
.gpersona
Change default global personality (bot owner only)
.auth
Add an admin (bot owner only)
.deauth
Remove an admin (bot owner only)
''')
# main loop # main loop
async def main(self): async def main(self):