Rene Teigen 1 year ago
parent 61a2a19191
commit deee0b328d

@ -1177,7 +1177,7 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"):
) )
self.redo_users[ctx.author.id] = RedoUser( self.redo_users[ctx.author.id] = RedoUser(
prompt=new_prompt, instruction=instruction, ctx=ctx, message=ctx, response=actual_response_message prompt=new_prompt, instruction=instruction, ctx=ctx, message=ctx, response=actual_response_message, codex=codex
) )
self.redo_users[ctx.author.id].add_interaction( self.redo_users[ctx.author.id].add_interaction(
actual_response_message.id actual_response_message.id
@ -1334,7 +1334,10 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"):
max_value=1, max_value=1,
) )
@discord.option( @discord.option(
name="codex", description="Enable codex version", required=False name="codex",
description="Enable codex version",
required=False,
default=False
) )
@discord.guild_only() @discord.guild_only()
async def edit( async def edit(
@ -1410,11 +1413,13 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"):
name="private", name="private",
description="Converse in a private thread", description="Converse in a private thread",
required=False, required=False,
default=False,
) )
@discord.option( @discord.option(
name="minimal", name="minimal",
description="Use minimal starter text, saves tokens and has a more open personality", description="Use minimal starter text, saves tokens and has a more open personality",
required=False, required=False,
default=False,
) )
@discord.guild_only() @discord.guild_only()
async def converse( async def converse(
@ -1856,6 +1861,7 @@ class RedoButton(discord.ui.Button["ConversationView"]):
instruction = self.converser_cog.redo_users[user_id].instruction instruction = self.converser_cog.redo_users[user_id].instruction
ctx = self.converser_cog.redo_users[user_id].ctx ctx = self.converser_cog.redo_users[user_id].ctx
response_message = self.converser_cog.redo_users[user_id].response response_message = self.converser_cog.redo_users[user_id].response
codex = self.converser_cog.redo_users[user_id].codex
msg = await interaction.response.send_message( msg = await interaction.response.send_message(
"Retrying your original request...", ephemeral=True, delete_after=15 "Retrying your original request...", ephemeral=True, delete_after=15
@ -1868,6 +1874,7 @@ class RedoButton(discord.ui.Button["ConversationView"]):
ctx=ctx, ctx=ctx,
model=self.model, model=self.model,
response_message=response_message, response_message=response_message,
codex=codex,
custom_api_key=self.custom_api_key, custom_api_key=self.custom_api_key,
redo_request=True, redo_request=True,
from_edit_command=self.from_edit_command from_edit_command=self.from_edit_command

@ -409,7 +409,7 @@ class Model:
) )
print(f"The text about to be edited is [{input}] with instructions [{instruction}]") print(f"The text about to be edited is [{input}] with instructions [{instruction}] codex [{codex}]")
print( print(
f"Overrides -> temp:{temp_override}, top_p:{top_p_override}" f"Overrides -> temp:{temp_override}, top_p:{top_p_override}"
) )

@ -5,12 +5,13 @@ history, message count, and the id of the user in order to track them.
class RedoUser: class RedoUser:
def __init__(self, prompt, instruction, message, ctx, response): def __init__(self, prompt, instruction, message, ctx, response, codex):
self.prompt = prompt self.prompt = prompt
self.instruction = instruction self.instruction = instruction
self.message = message self.message = message
self.ctx = ctx self.ctx = ctx
self.response = response self.response = response
self.codex = codex
self.interactions = [] self.interactions = []
def add_interaction(self, interaction): def add_interaction(self, interaction):

Loading…
Cancel
Save