Update tool/browsersearch.py

tool/browsersearch.py

@@ -1,36 +1,44 @@
-from langchain_openai import ChatOpenAI
-from browser_use import Agent
-import asyncio
-from dotenv import load_dotenv
-load_dotenv()
-from langchain.tools import BaseTool
-
-async def main(task):
-    agent = Agent(
-        task=task,
-        llm=ChatOpenAI(model="gpt-4o-2024-11-20",api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
-             base_url="https://www.dmxapi.com/v1"),
-    )
-    result = await agent.run()
-    return result
-
-class browseruse(BaseTool):
-    name: str = "browseruse"
-    description: str = ("Calling the browser to search for information in specific website"
-                        "input query, return the searching results")
-
-    def __init__(
-        self,
-    ):
-        super().__init__()
-
-    def _run(self, task: str) -> str:
-         result = asyncio.run(main(task)) 
-         return result
-    
-    async def _arun(self, smiles: str) -> str:
-        """Use the tool asynchronously."""
-        raise NotImplementedError()
-        
-        
+from langchain_openai import ChatOpenAI
+from browser_use import Agent
+import asyncio
+from dotenv import load_dotenv
+load_dotenv()
+import os
+from langchain.tools import BaseTool
+from browser_use.browser.browser import Browser, BrowserConfig
+from browser_use.controller.service import Controller
+browser = Browser(config=BrowserConfig())
+controller = Controller()
+if os.name == 'nt':
+    asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
+async def main(task):
+    agent = Agent(
+        task=task,
+        llm=ChatOpenAI(model="gpt-4o-2024-11-20",api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
+             base_url="https://www.dmxapi.com/v1"),		controller=controller,
+		browser=browser,
+		use_vision=True,
+    )
+    result = await agent.run()
+    return result
+
+class browseruse(BaseTool):
+    name: str = "browseruse"
+    description: str = ("Calling the browser to search for information in specific website"
+                        "input query, return the searching results")
+
+    def __init__(
+        self,
+    ):
+        super().__init__()
+
+    def _run(self, task: str) -> str:
+         result = asyncio.run(main(task)) 
+         return result
+    
+    async def _arun(self, smiles: str) -> str:
+        """Use the tool asynchronously."""
+        raise NotImplementedError()
+        
+