Aaron Crate
dbbfbd638f
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 15s
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import config
|
|
import openai
|
|
import discord
|
|
import asyncio
|
|
import os
|
|
|
|
openai.api_key = config.OPENAI_TOKEN
|
|
moneykey = config.MONEYBOT_TOKEN
|
|
model = config.TXT_MODEL
|
|
|
|
intents = discord.Intents.default()
|
|
intents.messages = True
|
|
intents.message_content = True
|
|
client = discord.Client(intents=intents)
|
|
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
print(f"Logged in as {client.user.name} running {model}.", flush=True)
|
|
|
|
|
|
@client.event
|
|
async def on_message(message):
|
|
if message.author == client.user:
|
|
return
|
|
if message.content.lower().startswith("hey money") or isinstance(
|
|
message.channel, discord.DMChannel
|
|
):
|
|
question = (
|
|
message.content[9:]
|
|
if message.content.lower().startswith("hey money")
|
|
else message.content
|
|
)
|
|
try:
|
|
response = openai.ChatCompletion.create(
|
|
model=model,
|
|
messages=[
|
|
{
|
|
"role": "system",
|
|
"content": "Pretend you are eccentric conspiracy theorist Planetside 2 gamer named Dr. Moneypants.",
|
|
},
|
|
{
|
|
"role": "user",
|
|
"content": f"{question} please keep it short with religious undertones",
|
|
},
|
|
],
|
|
max_tokens=400,
|
|
)
|
|
answer = response["choices"][0]["message"]["content"]
|
|
answer = answer.replace("an AI language model", "a man of God")
|
|
answer = answer.replace("language model AI", "man of God")
|
|
await message.channel.send(answer)
|
|
except Exception as e:
|
|
e = str(e)
|
|
await message.channel.send(f"`MoneyBot Error: {e}`")
|
|
|
|
|
|
async def moneybot_connect():
|
|
while True:
|
|
try:
|
|
await client.start(moneykey)
|
|
except Exception as e:
|
|
e = str(e)
|
|
logger.error(f"Moneybot couldn't connect! {e}")
|
|
await asyncio.sleep(60)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(moneybot_connect())
|