2024-02-06 01:00:41 +00:00
|
|
|
import ollama
|
2024-02-06 01:04:47 +00:00
|
|
|
|
2024-02-05 23:40:23 +00:00
|
|
|
def processmsg(msg, rcpt):
|
2024-02-06 01:04:47 +00:00
|
|
|
if "youtube.com/watch" in msg:
|
2024-02-06 00:03:54 +00:00
|
|
|
return msg.replace("youtube.com", "iv.datura.network")
|
2024-02-06 01:04:47 +00:00
|
|
|
elif msg.startswith("!wiki"):
|
|
|
|
|
return sf.query_external_website("https://en.wikipedia.org/wiki/", msg.split(" ")[1])
|
2024-02-06 00:03:54 +00:00
|
|
|
|
|
|
|
|
def command(msg, rcpt):
|
|
|
|
|
if msg.startswith("!help"):
|
2024-02-06 01:07:47 +00:00
|
|
|
response = "chatbot commands:" + "\n"
|
|
|
|
|
response += "!help Show this help page" + "\n"
|
2024-02-06 01:05:31 +00:00
|
|
|
response += "!ai [message] Ask llama2"
|
|
|
|
|
return response
|
2024-02-06 01:00:41 +00:00
|
|
|
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'])
|
|
|
|
|
|
