BotSharp/src/Plugins/BotSharp.Plugin.WebDriver/Functions/ChangeListValueFn.cs
2024-05-31 14:26:27 -05:00

44 lines
1.5 KiB
C#

namespace BotSharp.Plugin.WebDriver.Functions;
public class ChangeListValueFn : IFunctionCallback
{
public string Name => "change_list_value";
private readonly IServiceProvider _services;
private readonly IWebBrowser _browser;
public ChangeListValueFn(IServiceProvider services,
IWebBrowser browser)
{
_services = services;
_browser = browser;
}
public async Task<bool> Execute(RoleDialogModel message)
{
var convService = _services.GetRequiredService<IConversationService>();
var args = JsonSerializer.Deserialize<BrowsingContextIn>(message.FunctionArgs);
var agentService = _services.GetRequiredService<IAgentService>();
var agent = await agentService.LoadAgent(message.CurrentAgentId);
var result = await _browser.ChangeListValue(new BrowserActionParams(agent, args, convService.ConversationId, message.MessageId));
var content = $"Change value to '{args.UpdateValue}' for {args.ElementName}";
message.Content = result.IsSuccess ?
$"{content} successfully" :
$"{content} failed. {result.Message}";
var webDriverService = _services.GetRequiredService<WebDriverService>();
var path = webDriverService.GetScreenshotFilePath(message.MessageId);
message.Data = await _browser.ScreenshotAsync(new MessageInfo
{
AgentId = message.CurrentAgentId,
ContextId = convService.ConversationId,
MessageId = message.MessageId
}, path);
return true;
}
}