Merge pull request #55 from lmnr-ai/dev

Dev

Author: skull8888888

index/browser/browser.py

@@ -307,7 +307,7 @@ async def switch_to_tab(self, page_id: int) -> None:
 		self.current_page = page
 
 		await page.bring_to_front()
-		await page.wait_for_load_state()
+		await page.wait_for_load_state('domcontentloaded')
 
 	async def create_new_tab(self, url: str | None = None) -> None:
 		"""Create a new tab and optionally navigate to a URL"""
@@ -317,7 +317,7 @@ async def create_new_tab(self, url: str | None = None) -> None:
 		new_page = await self.context.new_page()
 		self.current_page = new_page
 
-		await new_page.wait_for_load_state()
+		await new_page.wait_for_load_state('domcontentloaded')
 
 		if url:
 			await new_page.goto(url, wait_until='domcontentloaded')

index/controller/default_actions.py

@@ -50,8 +50,8 @@ async def search_google(query: str, browser: Browser):
         Open google search in new tab and search for the query.
         """
         page = await browser.get_current_page()
-        await page.goto(f'https://www.google.com/search?q={query}&udm=14')
-        await page.wait_for_load_state()
+        await page.goto(f'https://www.google.com/search?q={query}&udm=14', wait_until='domcontentloaded')
+        await asyncio.sleep(1)
         msg = f"Searched for '{query}' in Google"
         logger.info(msg)
         return ActionResult(content=msg)

pyproject.toml

@@ -10,7 +10,7 @@ packages = ["index"]
 
 [project]
 name = "lmnr-index"
-version = "0.1.12"
+version = "0.1.13"
 description = "Index - SOTA browser AI agent for autonomous task execution on the web"
 readme = "README.md"
 requires-python = ">=3.10"