intuned-runtime 1.3.5__tar.gz → 1.3.6__tar.gz

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intuned-runtime
-Version: 1.3.5
+Version: 1.3.6
 Summary: Runtime SDK that powers browser automation projects running on Intuned
 License: Elastic-2.0
 License-File: LICENSE

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "intuned-runtime"
-version = "1.3.5"
+version = "1.3.6"
 description = "Runtime SDK that powers browser automation projects running on Intuned"
 authors = [ "Intuned Developers <engineering@intunedhq.com>",]
 readme = "README.md"

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/runtime/browser/extensions/intuned_extension.py

@@ -51,13 +51,7 @@ async def get_intuned_worker(context: BrowserContext):
     return None
 
 
-async def get_intuned_extension_settings() -> dict[str, Any]:
-    intuned_json = await load_intuned_json()
-    captcha_settings: CaptchaSolverSettings = (
-        intuned_json.captcha_solver
-        if intuned_json and intuned_json.captcha_solver is not None
-        else CaptchaSolverSettings()
-    )
+async def get_intuned_extension_settings(captcha_settings: CaptchaSolverSettings) -> dict[str, Any]:
     context = IntunedContext.current()
     return {
         **captcha_settings.model_dump(mode="json"),
@@ -71,14 +65,20 @@ async def get_intuned_extension_settings() -> dict[str, Any]:
 async def setup_intuned_extension():
     if not is_intuned_extension_enabled():
         return
-    await setup_intuned_extension_server()
+    intuned_json = await load_intuned_json()
+    captcha_settings: CaptchaSolverSettings = (
+        intuned_json.captcha_solver
+        if intuned_json and intuned_json.captcha_solver is not None
+        else CaptchaSolverSettings()
+    )
+    await setup_intuned_extension_server(captcha_settings)
     intuned_extension_path = get_intuned_extension_path()
     if intuned_extension_path is None:
         logger.warning("Intuned extension path not found, intuned extension might not work properly")
         return
 
     settings_path = intuned_extension_path / "intunedSettings.json"
-    settings_data = await get_intuned_extension_settings()
+    settings_data = await get_intuned_extension_settings(captcha_settings)
 
     try:
         with open(settings_path, "w") as f:

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/runtime/browser/extensions/intuned_extension_server.py

@@ -17,7 +17,6 @@ from waitress.server import create_server
 from waitress.server import MultiSocketServer
 
 from runtime.types import CaptchaSolverSettings
-from runtime.utils.config_loader import load_intuned_json
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -158,14 +157,10 @@ event_emitter: Optional[AsyncIOEventEmitter] = None
 extension_server: Optional[ExtensionServer] = None
 
 
-async def setup_intuned_extension_server():
+async def setup_intuned_extension_server(captcha_settings: Optional[CaptchaSolverSettings] = None):
     global event_emitter, extension_server
-    intuned_json = await load_intuned_json()
-    captcha_settings: CaptchaSolverSettings = (
-        intuned_json.captcha_solver
-        if intuned_json and intuned_json.captcha_solver is not None
-        else CaptchaSolverSettings()
-    )
+    if captcha_settings is None:
+        captcha_settings = CaptchaSolverSettings()
     extension_server = ExtensionServer()
     event_emitter = AsyncIOEventEmitter()
     await extension_server.start(port=captcha_settings.port)

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/runtime/browser/launch_chromium.py

@@ -13,6 +13,7 @@ from runtime.browser.extensions.intuned_extension import get_intuned_worker
 from runtime.browser.extensions.intuned_extension import is_intuned_extension_enabled
 from runtime.browser.extensions.intuned_extension import setup_intuned_extension
 from runtime.browser.extensions.intuned_extension_server import clean_intuned_extension_server
+from runtime.browser.extensions.intuned_extension_server import setup_intuned_extension_server
 
 from .helpers import get_local_cdp_address
 from .helpers import get_proxy_env
@@ -68,6 +69,8 @@ async def launch_chromium(
     extra_args: list[str] = []
     async with async_playwright() as playwright:
         if cdp_address is not None:
+            if is_intuned_extension_enabled():
+                await setup_intuned_extension_server()
             browser: Browser = await playwright.chromium.connect_over_cdp(cdp_address)
             context = browser.contexts[0]
             user_preferences_dir = None

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/runtime/helpers/extensions.py

@@ -202,25 +202,25 @@ async def _wait_for_captcha_solve_core(
 
     logger.debug(f"Page object: {page}")
 
-    # Execute function if provided
     result = None
     if func is not None:
         result = await func()
 
-    # Wait for network to settle if requested
     if wait_for_network_settled:
         try:
             await page.wait_for_load_state("networkidle", timeout=timeout)
         except Exception as e:
             logger.debug(f"Network idle wait failed: {e}")
 
-    # Check for captcha detection
     detection_event: EventRequest
     try:
         detection_event = await wait_for_captcha_event("CAPTCHA_DETECTED", timeout=timeout)
-    except:
-        logger.debug("CAPTCHA Not detected, skipping...")
+    except asyncio.TimeoutError:
+        logger.info("CAPTCHA Detection timed out")
         return result
+    except Exception as e:
+        logger.error(f"Error while waiting for captcha: {e}")
+        raise e
 
     logger.info("CAPTCHA Detected, awaiting result...")
     if detection_event.session_id is None:
@@ -263,9 +263,9 @@ async def _wait_for_captcha_solve_core(
 
         logger.info("CAPTCHA solved successfully")
 
-    except asyncio.TimeoutError:
+    except asyncio.TimeoutError as e:
         logger.error("CAPTCHA Result timeout")
-        raise
+        raise RuntimeError("CAPTCHA Solving timeout") from e
     except Exception as e:
         logger.error(f"CAPTCHA solve error: {e}")
         raise

{intuned_runtime-1.3.5 → intuned_runtime-1.3.6}/runtime/types/settings_types.py

@@ -1,3 +1,4 @@
+import os
 import socket
 from typing import List
 from typing import Literal
@@ -6,7 +7,9 @@ from pydantic import BaseModel
 from pydantic import Field
 
 
-def get_random_port():
+def get_intuned_captcha_extension_port():
+    if "INTUNED_CAPTCHA_EXTENSION_PORT" in os.environ:
+        return int(os.environ["INTUNED_CAPTCHA_EXTENSION_PORT"])
     with socket.socket() as s:
         s.bind(("", 0))
         return s.getsockname()[1]
@@ -54,7 +57,7 @@ class CaptchaSolverSettings(BaseModel):
     }
 
     enabled: bool = Field(default=False)
-    port: int = Field(default_factory=get_random_port)
+    port: int = Field(default_factory=get_intuned_captcha_extension_port)
     cloudflare: CaptchaSettings = Field(default_factory=CaptchaSettings)
     google_recaptcha_v2: CaptchaSettings = Field(alias="googleRecaptchaV2", default_factory=CaptchaSettings)
     google_recaptcha_v3: CaptchaSettings = Field(alias="googleRecaptchaV3", default_factory=CaptchaSettings)