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

Merge pull request #18 from h1ddenpr0cess20/dev

Browse Source 
cleanup
This commit is contained in:
Dustin 2024-07-30 22:06:29 -04:00 committed by GitHub
commit ab2b6c802c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ollamarama.py
View File

@ -39,7 +39,6 @@ class ollamarama:
self.default_model = self.models[config[0]['default_model']] self.default_model = self.models[config[0]['default_model']]
self.model = self.default_model self.model = self.default_model
#no idea if optimal, change if necessary
self.temperature, self.top_p, self.repeat_penalty = config[2]['options'].values() self.temperature, self.top_p, self.repeat_penalty = config[2]['options'].values()
self.defaults = { self.defaults = {
"temperature": self.temperature, "temperature": self.temperature,
@ -96,7 +95,6 @@ class ollamarama:
#generate Ollama model response #generate Ollama model response
async def respond(self, channel, sender, message, sender2=None): async def respond(self, channel, sender, message, sender2=None):
try: try:
# #Generate response
data = { data = {
"model": self.model, "model": self.model,
"messages": message, "messages": message,
@ -135,7 +133,7 @@ class ollamarama:
#Shrink history list for token size management #Shrink history list for token size management
if len(self.messages[channel][sender]) > 24: if len(self.messages[channel][sender]) > 24:
if self.messages[channel][sender][0]['role'] == 'system': if self.messages[channel][sender][0]['role'] == 'system':
del self.messages[channel][sender][1:3] #delete the first set of question and answers del self.messages[channel][sender][1:3]
else: else:
del self.messages[channel][sender][0:2] del self.messages[channel][sender][0:2]
@ -186,7 +184,7 @@ class ollamarama:
f.close() f.close()
self.models = config[0]['models'] self.models = config[0]['models']
if message == ".models": if message == ".models":
current_model = f"Current model: {self.model.removeprefix('ollama/')}\nAvailable models: {', '.join(sorted(list(self.models)))}" current_model = f"Current model: {self.model}\nAvailable models: {', '.join(sorted(list(self.models)))}"
await self.send_message(room_id, current_model) await self.send_message(room_id, current_model)
if message.startswith(".model "): if message.startswith(".model "):
@ -196,7 +194,7 @@ class ollamarama:
self.model = self.models[m] self.model = self.models[m]
elif m == 'reset': elif m == 'reset':
self.model = self.default_model self.model = self.default_model
await self.send_message(room_id, f"Model set to {self.model.removeprefix('ollama/')}") await self.send_message(room_id, f"Model set to {self.model}")
#bot owner commands #bot owner commands
if sender_display == self.admins[0]: if sender_display == self.admins[0]: