Merge pull request #12 from h1ddenpr0cess20/dev

improved model switching
This commit is contained in:
Dustin 2024-05-14 13:35:24 -04:00 committed by GitHub
commit 835900c16b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -14,7 +14,8 @@ import asyncio
class ollamarama:
def __init__(self):
#load config file
with open("config.json", "r") as f:
self.config_file = "config.json"
with open(self.config_file, "r") as f:
config = json.load(f)
f.close()
@ -171,7 +172,11 @@ class ollamarama:
if sender_display in self.admins:
#model switching
if message.startswith(".model"):
if message == ".models":
with open(self.config_file, "r") as f:
config = json.load(f)
f.close()
self.models = config[0]['models']
if message == ".models":
current_model = f"Current model: {self.model.removeprefix('ollama/')}\nAvailable models: {', '.join(sorted(list(self.models)))}"
await self.send_message(room_id, current_model)