huytofu92 commited on
Commit
cd38811
·
1 Parent(s): 1a04a88

Sync browser only

Browse files
Files changed (2) hide show
  1. browser.py +4 -6
  2. mini_agents.py +1 -3
browser.py CHANGED
@@ -4,7 +4,6 @@ from typing import List, Optional
4
  from threading import Lock
5
  from smolagents.tools import Tool
6
  from langchain_community.tools.playwright.utils import (
7
- create_async_playwright_browser,
8
  create_sync_playwright_browser
9
  )
10
  from langchain_community.agent_toolkits import PlayWrightBrowserToolkit
@@ -34,14 +33,13 @@ class BrowserManager:
34
  if self._browser_tools is None:
35
  with self._lock:
36
  if self._browser_tools is None:
37
- # Create browsers in the current context
38
- async_browser = create_async_playwright_browser()
39
  sync_browser = create_sync_playwright_browser()
40
 
41
- # Create toolkit and tools
42
  browser_toolkit = PlayWrightBrowserToolkit.from_browser(
43
- async_browser=async_browser,
44
- sync_browser=sync_browser
45
  )
46
  self._browser_tools = [
47
  Tool.from_langchain(tool)
 
4
  from threading import Lock
5
  from smolagents.tools import Tool
6
  from langchain_community.tools.playwright.utils import (
 
7
  create_sync_playwright_browser
8
  )
9
  from langchain_community.agent_toolkits import PlayWrightBrowserToolkit
 
33
  if self._browser_tools is None:
34
  with self._lock:
35
  if self._browser_tools is None:
36
+ # Create only sync browser
 
37
  sync_browser = create_sync_playwright_browser()
38
 
39
+ # Create toolkit with only sync browser
40
  browser_toolkit = PlayWrightBrowserToolkit.from_browser(
41
+ sync_browser=sync_browser,
42
+ async_browser=None
43
  )
44
  self._browser_tools = [
45
  Tool.from_langchain(tool)
mini_agents.py CHANGED
@@ -9,8 +9,6 @@ from browser import browser_manager
9
  import os
10
  import logging
11
  import yaml
12
- from typing import List
13
- from smolagents.tools import Tool
14
 
15
  logging.basicConfig(level=logging.DEBUG)
16
 
@@ -178,7 +176,7 @@ class MasterAgentWrapper:
178
  self.master_agent.tools = original_tools + browser_tools
179
 
180
  try:
181
- # Run the agent
182
  result = self.master_agent.run(question)
183
  return result
184
  finally:
 
9
  import os
10
  import logging
11
  import yaml
 
 
12
 
13
  logging.basicConfig(level=logging.DEBUG)
14
 
 
176
  self.master_agent.tools = original_tools + browser_tools
177
 
178
  try:
179
+ # Run the agent (using sync run since we're using sync browser)
180
  result = self.master_agent.run(question)
181
  return result
182
  finally: