adds custom name for the AI

Madpeterz · Jan 28, 2025 · 68a71f5 · 68a71f5
1 parent 4148d6d
commit 68a71f5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/SecondBotEvents/Config/ChatGptConfig.cs b/SecondBotEvents/Config/ChatGptConfig.cs
@@ -27,6 +27,12 @@ protected override void MakeSettings()
             settings.Add("ChatHistoryTimeout");
             settings.Add("ShowDebug");
             settings.Add("Provider");
+            settings.Add("CustomName");
+        }
+
+        public string GetCustomName()
+        {
+            return ReadSettingAsString("CustomName", "<!FIRSTNAME!>");
         }
 
         public bool GetShowDebug()

diff --git a/SecondBotEvents/Services/ChatGptService.cs b/SecondBotEvents/Services/ChatGptService.cs
@@ -288,19 +288,24 @@ protected async void GetAiReply(int ratelimiter, UUID replyTo, UUID user, string
                     }
                     // trim history
                     List<KeyValuePair<string, string>> history = chatHistoryAI[user];
+                    string yourname = GetClient().Self.FirstName;
+                    if(myConfig.GetCustomName() != "<!FIRSTNAME!>")
+                    {
+                        yourname = myConfig.GetCustomName();
+                    }
                     if (history.Count() == 0)
                     {
                         if (avatarchat == true)
                         {
-                            history.Add(new KeyValuePair<string, string>("system", "You are " + GetClient().Self.FirstName + ", "+myConfig.GetChatPrompt()+", you are talking to " + name + "."));
+                            history.Add(new KeyValuePair<string, string>("system", "You are " + yourname + ", "+myConfig.GetChatPrompt()+", you are talking to " + name + "."));
                         }
                         else if (groupchat == true)
                         {
-                            history.Add(new KeyValuePair<string, string>("system", "You are " + GetClient().Self.FirstName + ", "+myConfig.GetChatPrompt()+", you are talking to a group of people."));
+                            history.Add(new KeyValuePair<string, string>("system", "You are " + yourname + ", "+myConfig.GetChatPrompt()+", you are talking to a group of people."));
                         }
                         else
                         {
-                            history.Add(new KeyValuePair<string, string>("system", "You are " + GetClient().Self.FirstName + ", "+myConfig.GetChatPrompt()+", you are talking to people in a public place."));
+                            history.Add(new KeyValuePair<string, string>("system", "You are " + yourname + ", "+myConfig.GetChatPrompt()+", you are talking to people in a public place."));
                         }
                     }
                     history.Add(new KeyValuePair<string, string>("user", "" + name + " says " + message));