diff --git a/functions.py b/functions.py index fb47583..57fd81a 100644 --- a/functions.py +++ b/functions.py @@ -4,8 +4,8 @@ import scraper_functions as sf def processmsg(msg, rcpt): if "youtube.com/watch" in msg: return msg.replace("youtube.com", "iv.datura.network") - elif msg.startswith("!wiki"): - return sf.query_external_website("https://en.wikipedia.org/wiki/", msg.split(" ")[1]) + elif msg.startswith("!"): + return command(msg) elif "good bot" in msg: return "^_^" @@ -20,4 +20,6 @@ def command(msg, rcpt): client = ollama.Client(host='https://ollama.krov.dmz.rs') response = client.chat(model='llama2-uncensored:latest', messages=[{'role':'user','content':f'{msg[4:]}'}]) return(response['message']['content']) + elif msg.startswith("!wiki"): + return sf.query_external_website("https://en.wikipedia.org/wiki/", msg.split(" ")[1])