Skip to content
This repository was archived by the owner on Feb 7, 2026. It is now read-only.

Add back support for /chatgpt and /openai isobot AI commands #373

Merged
notsniped merged 3 commits into from
Apr 20, 2024
Merged

Add back support for /chatgpt and /openai isobot AI commands #373

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
07fa2c0
Remove locally-added `openai` Python library from client
notsniped Apr 20, 2024
e53c8e1
Add back support for `/chatgpt` and `/openai` isobot AI commands
notsniped Apr 20, 2024
c54eb29
Merge branch 'main' into add-back-support-for-isobot-ai-commands
notsniped Apr 20, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
128 changes: 62 additions & 66 deletions cogs/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import os
import math
import psutil
# import openai # Removed because OpenAI commands were disabled, and OpenAI fails to load due to missing "PyDantic" dependency
import openai
import discord
import json
import time
Expand All @@ -23,7 +23,7 @@
levelling = levelling.Levelling()
_commands = cmds.Commands()
# openai.api_key = os.getenv("chatgpt_API_KEY")
# openai.api_key = auth.ext_token('chatgpt')
openai.api_key = auth.ext_token('chatgpt')
chatgpt_conversation = dict()
_presence = Presence()

Expand Down Expand Up @@ -214,36 +214,34 @@ async def status(self, ctx: ApplicationContext):
@commands.cooldown(1, 1, commands.BucketType.user)
async def chatgpt(self, ctx: ApplicationContext, message: str):
"""Talk to ChatGPT and get a response back."""
# if str(ctx.author.id) not in chatgpt_conversation:
# chatgpt_conversation[str(ctx.author.id)] = [
# {
# "role": "system",
# "content": "You are a intelligent assistant."
# }
# ]
# await ctx.defer()
# try:
# chatgpt_conversation[str(ctx.author.id)].append({"role": "user", "content": message})
# _chat = openai.ChatCompletion.create(
# model="gpt-3.5-turbo",
# messages=chatgpt_conversation[str(ctx.author.id)]
# )
# _reply = _chat.choices[0].message.content
# chatgpt_conversation[str(ctx.author.id)].append({"role": "assistant", "content": _reply})
# except openai.error.RateLimitError as e:
# print(f"Rate limit for OpenAI exceeded: {e}")
# return await ctx.respond("The OpenAI API is currently being rate-limited. Try again after some time.", ephemeral=True)
# except openai.error.ServiceUnavailableError:
# return await ctx.respond("The ChatGPT service is currently unavailable.\nTry again after some time, or check it's status at https://status.openai.com", ephemeral=True)
# except openai.error.APIError:
# return await ctx.respond("ChatGPT encountered an internal error. Please try again.", ephemeral=True)
# except openai.error.Timeout:
# return await ctx.respond("Your request timed out. Please try again, or wait for a while.", ephemeral=True)
# localembed = discord.Embed(description=f"{_reply}", color=discord.Color.random())
# localembed.set_author(name="ChatGPT", icon_url="https://upload.wikimedia.org/wikipedia/commons/thumb/0/04/ChatGPT_logo.svg/1200px-ChatGPT_logo.svg.png")
# localembed.set_footer(text="Powered by OpenAI")
# await ctx.respond(embed=localembed)
localembed = discord.Embed(title="Discontinuation of isobot AI commands", description="Thank you for showing your interest in the isobot AI commands!\nUnfortunately, due to prolonged issues with OpenAI integration, we are temporarily discontinuing all AI-related commands.\nDon't worry, because sometime, in the (not so distant) future, isobot AI commands will be making a sure return for everyone to enjoy.\n\n- NKA Development Team")
if str(ctx.author.id) not in chatgpt_conversation:
chatgpt_conversation[str(ctx.author.id)] = [
{
"role": "system",
"content": "You are a intelligent assistant."
}
]
await ctx.defer()
try:
chatgpt_conversation[str(ctx.author.id)].append({"role": "user", "content": message})
_chat = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=chatgpt_conversation[str(ctx.author.id)]
)
_reply = _chat.choices[0].message.content
chatgpt_conversation[str(ctx.author.id)].append({"role": "assistant", "content": _reply})
except openai.error.RateLimitError as e:
print(f"Rate limit for OpenAI exceeded: {e}")
return await ctx.respond("The OpenAI API is currently being rate-limited. Try again after some time.", ephemeral=True)
except openai.error.ServiceUnavailableError:
return await ctx.respond("The ChatGPT service is currently unavailable.\nTry again after some time, or check it's status at https://status.openai.com", ephemeral=True)
except openai.error.APIError:
return await ctx.respond("ChatGPT encountered an internal error. Please try again.", ephemeral=True)
except openai.error.Timeout:
return await ctx.respond("Your request timed out. Please try again, or wait for a while.", ephemeral=True)
localembed = discord.Embed(description=f"{_reply}", color=discord.Color.random())
localembed.set_author(name="ChatGPT", icon_url="https://upload.wikimedia.org/wikipedia/commons/thumb/0/04/ChatGPT_logo.svg/1200px-ChatGPT_logo.svg.png")
localembed.set_footer(text="Powered by OpenAI")
await ctx.respond(embed=localembed)

@commands.slash_command(
Expand All @@ -255,40 +253,38 @@ async def chatgpt(self, ctx: ApplicationContext, message: str):
@commands.cooldown(1, 10, commands.BucketType.user)
async def generate_image(self, ctx: ApplicationContext, prompt: str, resolution: str = "512x512"):
"""Generate an image of your choice using the DALL-E modal."""
# parsed_resolution: list = resolution.split("x")
# max_index: int = 0
# for index in parsed_resolution:
# max_index += 1
# if max_index < 2 or max_index > 2:
# return await ctx.respond("Your resolution format is malformed. Please check it and try again.", ephemeral=True)
# res_width = int(parsed_resolution[0])
# res_height = int(parsed_resolution[1])
# if res_width < 256 or res_height < 256:
# return await ctx.respond("Your custom resolution needs to be at least 256p or higher.", ephermeral=True)
# if res_width > 1024 or res_height > 1024:
# return await ctx.respond("Your image output resolution cannot exceed 1024p.", ephemeral=True)
# await ctx.defer()
# try:
# response = openai.Image.create(
# prompt=prompt,
# n=1,
# size=resolution
# )
# generated_image_url = response['data'][0]['url']
# except openai.error.RateLimitError:
# return await ctx.respond("The OpenAI API is currently being rate-limited. Try again after some time.", ephemeral=True)
# except openai.error.ServiceUnavailableError:
# return await ctx.respond("The OpenAI service is currently unavailable.\nTry again after some time, or check it's status at https://status.openai.com", ephemeral=True)
# except openai.error.APIError:
# return await ctx.respond("DALL-E encountered an internal error. Please try again.", ephemeral=True)
# except openai.error.Timeout:
# return await ctx.respond("Your request timed out. Please try again, or wait for a while.", ephemeral=True)
# localembed = discord.Embed(title="Here's an image generated using your prompt.", color=discord.Color.random())
# localembed.set_image(url=generated_image_url)
# localembed.set_author(name="DALL-E", icon_url="https://upload.wikimedia.org/wikipedia/commons/thumb/0/04/ChatGPT_logo.svg/1200px-ChatGPT_logo.svg.png")
# localembed.set_footer(text="Powered by OpenAI")
# await ctx.respond(embed=localembed)
localembed = discord.Embed(title="Discontinuation of isobot AI commands", description="Thank you for showing your interest in the isobot AI commands!\nUnfortunately, due to prolonged issues with OpenAI integration, we are temporarily discontinuing all AI-related commands.\nDon't worry, because sometime, in the (not so distant) future, isobot AI commands will be making a sure return for everyone to enjoy.\n\n- NKA Development Team")
parsed_resolution: list = resolution.split("x")
max_index: int = 0
for index in parsed_resolution:
max_index += 1
if max_index < 2 or max_index > 2:
return await ctx.respond("Your resolution format is malformed. Please check it and try again.", ephemeral=True)
res_width = int(parsed_resolution[0])
res_height = int(parsed_resolution[1])
if res_width < 256 or res_height < 256:
return await ctx.respond("Your custom resolution needs to be at least 256p or higher.", ephermeral=True)
if res_width > 1024 or res_height > 1024:
return await ctx.respond("Your image output resolution cannot exceed 1024p.", ephemeral=True)
await ctx.defer()
try:
response = openai.Image.create(
prompt=prompt,
n=1,
size=resolution
)
generated_image_url = response['data'][0]['url']
except openai.error.RateLimitError:
return await ctx.respond("The OpenAI API is currently being rate-limited. Try again after some time.", ephemeral=True)
except openai.error.ServiceUnavailableError:
return await ctx.respond("The OpenAI service is currently unavailable.\nTry again after some time, or check it's status at https://status.openai.com", ephemeral=True)
except openai.error.APIError:
return await ctx.respond("DALL-E encountered an internal error. Please try again.", ephemeral=True)
except openai.error.Timeout:
return await ctx.respond("Your request timed out. Please try again, or wait for a while.", ephemeral=True)
localembed = discord.Embed(title="Here's an image generated using your prompt.", color=discord.Color.random())
localembed.set_image(url=generated_image_url)
localembed.set_author(name="DALL-E", icon_url="https://upload.wikimedia.org/wikipedia/commons/thumb/0/04/ChatGPT_logo.svg/1200px-ChatGPT_logo.svg.png")
localembed.set_footer(text="Powered by OpenAI")
await ctx.respond(embed=localembed)

@commands.slash_command(
Expand Down
Loading