BotSharp/src/Plugins/BotSharp.Plugin.WebDriver/Functions/ClickHtmlElementFn.cs
2024-01-03 13:40:41 -06:00

33 lines
962 B
C#

using BotSharp.Abstraction.Agents;
using BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;
namespace BotSharp.Plugin.WebDriver.Functions;
public class ClickHtmlElementFn : IFunctionCallback
{
public string Name => "click_html_element";
private readonly IServiceProvider _services;
private readonly PlaywrightWebDriver _driver;
public ClickHtmlElementFn(IServiceProvider services,
PlaywrightWebDriver driver)
{
_services = services;
_driver = driver;
}
public async Task<bool> Execute(RoleDialogModel message)
{
var args = JsonSerializer.Deserialize<BrowsingContextIn>(message.FunctionArgs);
var agentService = _services.GetRequiredService<IAgentService>();
var agent = await agentService.LoadAgent(message.CurrentAgentId);
await _driver.ClickElement(agent, args, message.MessageId);
message.Content = "Executed successfully.";
return true;
}
}