Update PizzaBotConversationHook.cs

This commit is contained in:
geffzhang 2025-03-28 09:49:54 +08:00
parent 451e7802eb
commit 71336f9ecf

View file

@ -32,18 +32,20 @@ public class PizzaBotConversationHook : ConversationHookBase
return base.OnTaskCompleted(message);
}
#if USE_BOTSHARP
public override async Task OnResponseGenerated(RoleDialogModel message)
{
var agentService = _services.GetRequiredService<IAgentService>();
var state = _services.GetRequiredService<IConversationStateService>();
var agent = await agentService.LoadAgent(message.CurrentAgentId);
#if USE_BOTSHARP
if (agent.McpTools.Any(item => item.Functions.Any(x => x.Name == message.FunctionName)))
{
var data = JsonDocument.Parse(JsonSerializer.Serialize(message.Data));
state.SaveStateByArgs(data);
}
#endif
await base.OnResponseGenerated(message);
}
#endif
}