diff --git a/.gitignore b/.gitignore index 948a4f4..7cfbc19 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ config.ini +venv/ __pycache__ diff --git a/functions.py b/functions.py index 6a2fc5c..348f71d 100644 --- a/functions.py +++ b/functions.py @@ -1,3 +1,5 @@ +import ollama + def processmsg(msg, rcpt): if msg.startswith("!"): return command(msg, rcpt) @@ -7,3 +9,8 @@ def processmsg(msg, rcpt): def command(msg, rcpt): if msg.startswith("!help"): return "chatbot commands: \n" + "!help Show this help page" + elif msg.startswith("!ai"): + 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']) + diff --git a/requirements.txt b/requirements.txt index 5593b66..e1e7e30 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,2 @@ slixmpp +ollama