Merge pull request #44 from rolanddeboer/main

Replace 'Human:' in prompt with actual username
Kav-K · Jan 1, 2023 · ca8e6b0 · ca8e6b0
2 parents d0be447 + 4ecf299
commit ca8e6b0
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)