From bb49b39e82562f21866c46cd320b762c5738bf95 Mon Sep 17 00:00:00 2001 From: Jicheng Lu Date: Wed, 4 Sep 2024 20:12:55 -0500 Subject: [PATCH] minor change --- .../BotSharp.Plugin.Planner/Functions/SecondaryStagePlanFn.cs | 4 ++-- .../BotSharp.Plugin.Planner/Functions/SummaryPlanFn.cs | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/Plugins/BotSharp.Plugin.Planner/Functions/SecondaryStagePlanFn.cs b/src/Plugins/BotSharp.Plugin.Planner/Functions/SecondaryStagePlanFn.cs index cc5aaa9b..9ff3a002 100644 --- a/src/Plugins/BotSharp.Plugin.Planner/Functions/SecondaryStagePlanFn.cs +++ b/src/Plugins/BotSharp.Plugin.Planner/Functions/SecondaryStagePlanFn.cs @@ -71,8 +71,8 @@ public class SecondaryStagePlanFn : IFunctionCallback var agentService = _services.GetRequiredService(); var render = _services.GetRequiredService(); - var planner = await agentService.GetAgent(message.CurrentAgentId); - var template = planner.Templates.FirstOrDefault(x => x.Name == "two_stage.2nd.plan")?.Content ?? string.Empty; + var agent = await agentService.GetAgent(message.CurrentAgentId); + var template = agent.Templates.FirstOrDefault(x => x.Name == "two_stage.2nd.plan")?.Content ?? string.Empty; var responseFormat = JsonSerializer.Serialize(new SecondStagePlan { Tool = "tool name if task solution provided", diff --git a/src/Plugins/BotSharp.Plugin.Planner/Functions/SummaryPlanFn.cs b/src/Plugins/BotSharp.Plugin.Planner/Functions/SummaryPlanFn.cs index 8320713c..bba623b9 100644 --- a/src/Plugins/BotSharp.Plugin.Planner/Functions/SummaryPlanFn.cs +++ b/src/Plugins/BotSharp.Plugin.Planner/Functions/SummaryPlanFn.cs @@ -67,8 +67,8 @@ public class SummaryPlanFn : IFunctionCallback var agentService = _services.GetRequiredService(); var render = _services.GetRequiredService(); - var aiAssistant = await agentService.GetAgent(BuiltInAgentId.Planner); - var template = aiAssistant.Templates.FirstOrDefault(x => x.Name == "two_stage.summarize")?.Content ?? string.Empty; + var agent = await agentService.GetAgent(BuiltInAgentId.Planner); + var template = agent.Templates.FirstOrDefault(x => x.Name == "two_stage.summarize")?.Content ?? string.Empty; var responseFormat = JsonSerializer.Serialize(new FirstStagePlan { Parameters = [JsonDocument.Parse("{}")],