phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

now use gpt-4o-mini for decreased pricing/better outputs

Browse source
This commit is contained in:
phixxy 2024-07-20 12:58:51 -07:00
parent 7c6c012be6
commit 42ae73030f
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
cogs/chatgpt.py
View file

@ -52,7 +52,7 @@ class ChatGPT(commands.Cog):
self.logger.exception(f"ChatGPT failed to make directories: {e}") self.logger.exception(f"ChatGPT failed to make directories: {e}")
def text_cost_calc(self, model, input_tokens, output_tokens): def text_cost_calc(self, model, input_tokens, output_tokens):
cost_table = {"gpt-3.5-turbo":{"input_tokens":0.0000005,"output_tokens":0.0000015}, cost_table = {"gpt-4o-mini":{"input_tokens":0.00000015,"output_tokens":0.0000006},
"gpt-4-turbo-preview":{"input_tokens":0.00001,"output_tokens":0.00003}, "gpt-4-turbo-preview":{"input_tokens":0.00001,"output_tokens":0.00003},
"gpt-4-vision-preview":{"input_tokens":0.00001,"output_tokens":0.00003}, "gpt-4-vision-preview":{"input_tokens":0.00001,"output_tokens":0.00003},
"gpt-4o":{"input_tokens":0.000005,"output_tokens":0.000015} "gpt-4o":{"input_tokens":0.000005,"output_tokens":0.000015}
@ -287,7 +287,7 @@ class ChatGPT(commands.Cog):
user = self.bot.get_user(reminder_dict["user_id"]) user = self.bot.get_user(reminder_dict["user_id"])
return await user.send(reminder_dict["response"]) return await user.send(reminder_dict["response"])
async def answer_question(self, topic, model="gpt-3.5-turbo"): async def answer_question(self, topic, model="gpt-4o-mini"):
data = { data = {
"model": model, "model": model,
"messages": [{"role": "user", "content": topic}] "messages": [{"role": "user", "content": topic}]
@ -701,7 +701,7 @@ class ChatGPT(commands.Cog):
message = ctx.content[0] message = ctx.content[0]
data = { data = {
"model": "gpt-3.5-turbo", "model": "gpt-4o-mini",
"messages": [{"role": "system", "content": system_msg}, {"role": "user", "content": message}] "messages": [{"role": "system", "content": system_msg}, {"role": "user", "content": message}]
} }