Merge branch 'main' of https://github.com/Kav-K/GPT3Discord

Kav-K · Jan 1, 2023 · 53fd818 · 53fd818
2 parents 4f9006b + 5a79188
commit 53fd818
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cogs/gpt_3_commands_and_converser.py b/cogs/gpt_3_commands_and_converser.py
@@ -518,6 +518,12 @@ async def encapsulated_send(
 
         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)