Replace Human: in prompt with actual username

2 years ago · 7ba231473a
parent a0089d6613
commit 7ba231473a
1 changed files with 6 additions and 0 deletions
--- a/cogs/gpt_3_commands_and_converser.py
+++ b/cogs/gpt_3_commands_and_converser.py
@ -503,6 +503,12 @@ class GPT3ComCon(commands.Cog, name="GPT3ComCon"):
        new_prompt = prompt + "\nGPTie: "
        from_context = isinstance(ctx, discord.ApplicationContext)

+        # Replace 'Human:' with the user's name
+        try:
+             new_prompt = new_prompt.replace("Human:", ctx.author.name + ":")
+        except AttributeError:
+            pass
+
        try:
            tokens = self.usage_service.count_tokens(new_prompt)