Update webdriver code

This commit is contained in:
vguruparan 2025-03-24 15:33:46 -05:00
parent 0ed8ecab23
commit 9f0995b15b
6 changed files with 15 additions and 10 deletions

View file

@ -1,5 +1,3 @@
using BotSharp.Abstraction.Browsing.Settings;
namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;
public partial class PlaywrightWebDriver
@ -7,8 +5,7 @@ public partial class PlaywrightWebDriver
public async Task<BrowserActionResult> ScreenshotAsync(MessageInfo message, string path)
{
var result = new BrowserActionResult();
var _webDriver = _services.GetRequiredService<WebBrowsingSettings>();
if (_webDriver.IsEnableScreenshot)
if (_webBrowsingSettings.IsEnableScreenshot)
{
await _instance.Wait(message.ContextId, waitNetworkIdle: false);
var page = _instance.GetPage(message.ContextId);

View file

@ -1,4 +1,6 @@
using BotSharp.Abstraction.Browsing.Settings;
namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;
public partial class PlaywrightWebDriver : IWebBrowser
@ -6,17 +8,19 @@ public partial class PlaywrightWebDriver : IWebBrowser
private IServiceProvider _services => _instance.Services;
private readonly PlaywrightInstance _instance;
private readonly ILogger _logger;
private readonly WebBrowsingSettings _webBrowsingSettings;
public PlaywrightInstance Instance => _instance;
public Agent Agent => _agent;
private Agent _agent;
public PlaywrightWebDriver(IServiceProvider services, PlaywrightInstance instance, ILogger<PlaywrightWebDriver> logger)
public PlaywrightWebDriver(IServiceProvider services, PlaywrightInstance instance, ILogger<PlaywrightWebDriver> logger, WebBrowsingSettings webBrowsingSettings)
{
_instance = instance;
_logger = logger;
_instance.SetServiceProvider(services);
}
_webBrowsingSettings = webBrowsingSettings;
}
public void SetAgent(Agent agent)
{

View file

@ -34,7 +34,8 @@ public class UtilWebActionOnElementFn : IFunctionCallback
var conv = _services.GetRequiredService<IConversationService>();
var browser = _services.GetRequiredService<IWebBrowser>();
var services = _services.CreateScope().ServiceProvider;
var browser = services.GetRequiredService<IWebBrowser>();
var webDriverService = _services.GetRequiredService<WebDriverService>();
var msg = new MessageInfo
{

View file

@ -19,7 +19,8 @@ public class UtilWebCloseBrowserFn : IFunctionCallback
{
var conv = _services.GetRequiredService<IConversationService>();
var webDriverService = _services.GetRequiredService<WebDriverService>();
var browser = _services.GetRequiredService<IWebBrowser>();
var services = _services.CreateScope().ServiceProvider;
var browser = services.GetRequiredService<IWebBrowser>();
var msg = new MessageInfo
{
AgentId = message.CurrentAgentId,

View file

@ -30,7 +30,8 @@ public class UtilWebGoToPageFn : IFunctionCallback
var conv = _services.GetRequiredService<IConversationService>();
var webDriverService = _services.GetRequiredService<WebDriverService>();
var browser = _services.GetRequiredService<IWebBrowser>();
var services = _services.CreateScope().ServiceProvider;
var browser = services.GetRequiredService<IWebBrowser>();
var msg = new MessageInfo
{
AgentId = message.CurrentAgentId,

View file

@ -21,7 +21,8 @@ public class UtilWebLocateElementFn : IFunctionCallback
var conv = _services.GetRequiredService<IConversationService>();
locatorArgs.Highlight = true;
var browser = _services.GetRequiredService<IWebBrowser>();
var services = _services.CreateScope().ServiceProvider;
var browser = services.GetRequiredService<IWebBrowser>();
var webDriverService = _services.GetRequiredService<WebDriverService>();
var msg = new MessageInfo
{