Revamp OpenAI recommendations

1 files changed, 4 insertions, 1 deletions

diff --git a/code/bot.py b/code/bot.py
index 6735522..da23a28 100644
--- a/code/bot.py
+++ b/code/bot.py
@@ -2,6 +2,7 @@ import discord
 from discord.ext import commands, tasks
 import os
 import requests
+import openai
 
 import config
 from tree import Tree
@@ -32,6 +33,8 @@ class MyBot(commands.Bot):
             if ext.endswith(".py"):
                 await self.load_extension(f"cogs.owner.{ext[:-3]}")
 
+        bot.openai = openai.OpenAI(api_key=config.OPENAI_API_KEY)
+
 
 bot = MyBot()
 bot.remove_command("help")
@@ -65,4 +68,4 @@ async def get_access_token():
 
 if __name__ == "__main__":
     config.load_config()
-    bot.run(config.TOKEN)
+    bot.run(config.TOKEN)
\ No newline at end of file