From 81233011b0a09d00ee4e443d214e72f4c9bdf5f8 Mon Sep 17 00:00:00 2001
From: crate <crateimage@gmail.com>
Date: Mon, 21 Oct 2024 01:39:42 -0500
Subject: [PATCH] fix jonbot and moneybot

---
 jonbot.py   | 21 ++++++++++++++-------
 moneybot.py | 17 ++++++++++++++---
 2 files changed, 28 insertions(+), 10 deletions(-)

diff --git a/jonbot.py b/jonbot.py
index cc09161..dbe8101 100644
--- a/jonbot.py
+++ b/jonbot.py
@@ -1,11 +1,12 @@
 import config
 import openai
 import discord
+import asyncio
 import os
 
 openai.api_key = config.OPENAI_TOKEN
 jonkey = config.JONBOT_TOKEN
-model = "gpt-3.5-turbo"
+model = config.TXT_MODEL
 
 intents = discord.Intents.default()
 intents.messages = True
@@ -14,7 +15,7 @@ client = discord.Client(intents=intents)
 
 @client.event
 async def on_ready():
-    print(f"Logged in as {client.user.name} running gpt-3.5-turbo.", flush=True)
+    print(f"Logged in as {client.user.name} running {model}.", flush=True)
 
 @client.event
 async def on_message(message):
@@ -39,8 +40,14 @@ async def on_message(message):
             e = str(e)
             await message.channel.send(f"`JonBot Error: {e}`")
 
-try:
-    client.run(jonkey)
-except Exception as e:
-        e = str(e)
-        print(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())
diff --git a/moneybot.py b/moneybot.py
index 2525555..428aef1 100644
--- a/moneybot.py
+++ b/moneybot.py
@@ -1,11 +1,12 @@
 import config
 import openai
 import discord
+import asyncio
 import os
 
 openai.api_key = config.OPENAI_TOKEN
 moneykey = config.MONEYBOT_TOKEN
-model = "gpt-4"
+model = config.TXT_MODEL
 
 intents = discord.Intents.default()
 intents.messages = True
@@ -14,7 +15,7 @@ client = discord.Client(intents=intents)
 
 @client.event
 async def on_ready():
-    print(f"Logged in as {client.user.name} running gpt-4.", flush=True)
+    print(f"Logged in as {client.user.name} running {model}.", flush=True)
 
 @client.event
 async def on_message(message):
@@ -39,4 +40,14 @@ async def on_message(message):
             e = str(e)
             await message.channel.send(f"`MoneyBot Error: {e}`")
 
-client.run(moneykey)
+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())