Skip to content

Commit 3dd203f

Browse files
authored
Merge pull request #382 from hchen2020/master
Fix OnRoutingInstructionRevised
2 parents a245d71 + 15291fa commit 3dd203f

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

src/Infrastructure/BotSharp.Core/Routing/Handlers/RouteToAgentRoutingHandler.cs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
using BotSharp.Abstraction.Infrastructures.Enums;
12
using BotSharp.Abstraction.Routing.Settings;
23

34
namespace BotSharp.Core.Routing.Handlers;
@@ -39,11 +40,8 @@ public RouteToAgentRoutingHandler(IServiceProvider services, ILogger<RouteToAgen
3940

4041
public async Task<bool> Handle(IRoutingService routing, FunctionCallFromLlm inst, RoleDialogModel message)
4142
{
42-
message.FunctionArgs = JsonSerializer.Serialize(inst);
43-
var ret = await routing.InvokeFunction(message.FunctionName, message);
44-
4543
var states = _services.GetRequiredService<IConversationStateService>();
46-
var goalAgent = states.GetState("user_goal_agent");
44+
var goalAgent = states.GetState(StateConst.EXPECTED_GOAL_AGENT);
4745
if (!string.IsNullOrEmpty(goalAgent) && inst.OriginalAgent != goalAgent)
4846
{
4947
inst.OriginalAgent = goalAgent;
@@ -53,6 +51,9 @@ await hook.OnRoutingInstructionRevised(inst, message)
5351
);
5452
}
5553

54+
message.FunctionArgs = JsonSerializer.Serialize(inst);
55+
var ret = await routing.InvokeFunction(message.FunctionName, message);
56+
5657
var agentId = routing.Context.GetCurrentAgentId();
5758

5859
// Update next action agent's name

src/Plugins/BotSharp.Plugin.ChatHub/Hooks/StreamingLogHook.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ public async Task OnRoutingInstructionRevised(FunctionCallFromLlm instruct, Role
326326
{
327327
var conversationId = _state.GetConversationId();
328328
var agent = await _agentService.LoadAgent(message.CurrentAgentId);
329-
var log = $"Revised user goal agent to: {agent?.Name}";
329+
var log = $"Revised user goal agent to {instruct.OriginalAgent}";
330330

331331
var input = new ContentLogInputModel(conversationId, message)
332332
{

0 commit comments

Comments
 (0)