Skip to content

Commit 5564287

Browse files
authored
Agent Name is contaminated.
1 parent ed7cabd commit 5564287

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

src/Infrastructure/BotSharp.Core/Routing/Planning/NaivePlanner.cs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,22 @@ private void FixMalformedResponse(FunctionCallFromLlm args)
165165
malformed = true;
166166
}
167167

168+
// Agent Name is contaminated.
169+
if (args.Function == "route_to_agent")
170+
{
171+
// Action agent name
172+
if (!agents.Any(x => x.Name == args.AgentName))
173+
{
174+
args.AgentName = agents.FirstOrDefault(x => args.AgentName.Contains(x.Name))?.Name ?? args.AgentName;
175+
}
176+
177+
// Goal agent name
178+
if (!agents.Any(x => x.Name == args.OriginalAgent))
179+
{
180+
args.OriginalAgent = agents.FirstOrDefault(x => args.OriginalAgent.Contains(x.Name))?.Name ?? args.OriginalAgent;
181+
}
182+
}
183+
168184
if (malformed)
169185
{
170186
_logger.LogWarning($"Captured LLM malformed response");

0 commit comments

Comments
 (0)