stash extras

etc/jonbot.py

@@ -0,0 +1,53 @@
+import config
+import openai
+import discord
+import asyncio
+import os
+
+openai.api_key = config.OPENAI_TOKEN
+jonkey = config.JONBOT_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 jon") or isinstance(message.channel, discord.DMChannel):
+        question = message.content[7:] if message.content.lower().startswith("hey jon") else message.content
+        try:
+            response = openai.ChatCompletion.create(
+                model=model,
+                messages=[
+                        {"role": "system", "content": "Pretend you are friendly Jon Arbuckle."},
+                        {"role": "user", "content": f"{question}"}
+                    ],
+                max_tokens=400
+            )
+            answer = response['choices'][0]['message']['content']
+            answer = answer.replace("AI language model", "American citizen")
+            answer = answer.replace("language model AI", "citizen of the United States")
+            await message.channel.send(answer)
+        except Exception as e:
+            e = str(e)
+            await message.channel.send(f"`JonBot Error: {e}`")
+
+async def jonbot_connect():
+    while True:
+        try:
+            await client.start(jonkey)
+        except Exception as e:
+                e = str(e)
+                logger.error(f"Jonbot couldn't connect! {e}")
+                await asyncio.sleep(60)
+
+if __name__ == "__main__":
+    asyncio.run(jonbot_connect())