From 5564287871efe44226eef406898162d98b402c80 Mon Sep 17 00:00:00 2001
From: "C. Oceania" <haiping008@gmail.com>
Date: Wed, 10 Apr 2024 15:02:19 -0500
Subject: [PATCH] Agent Name is contaminated.

---
 .../Routing/Planning/NaivePlanner.cs             | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/Infrastructure/BotSharp.Core/Routing/Planning/NaivePlanner.cs b/src/Infrastructure/BotSharp.Core/Routing/Planning/NaivePlanner.cs
index eefb88f7..aeed1a78 100644
--- a/src/Infrastructure/BotSharp.Core/Routing/Planning/NaivePlanner.cs
+++ b/src/Infrastructure/BotSharp.Core/Routing/Planning/NaivePlanner.cs
@@ -165,6 +165,22 @@ public class NaivePlanner : IPlaner
             malformed = true;
         }
 
+        // Agent Name is contaminated.
+        if (args.Function == "route_to_agent")
+        {
+            // Action agent name
+            if (!agents.Any(x => x.Name == args.AgentName))
+            {
+                args.AgentName = agents.FirstOrDefault(x => args.AgentName.Contains(x.Name))?.Name ?? args.AgentName;
+            }
+
+            // Goal agent name
+            if (!agents.Any(x => x.Name == args.OriginalAgent))
+            {
+                args.OriginalAgent = agents.FirstOrDefault(x => args.OriginalAgent.Contains(x.Name))?.Name ?? args.OriginalAgent;
+            }
+        }
+
         if (malformed)
         {
             _logger.LogWarning($"Captured LLM malformed response");