gemx 0.2.2__tar.gz → 0.2.4__tar.gz

{gemx-0.2.2 → gemx-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: gemx
-Version: 0.2.2
+Version: 0.2.4
 Summary: Drive the Gemini web UI from Python via Playwright
 Project-URL: Homepage, https://github.com/ra0x3/gemx
 Project-URL: Repository, https://github.com/ra0x3/gemx

{gemx-0.2.2 → gemx-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gemx"
-version = "0.2.2"
+version = "0.2.4"
 description = "Drive the Gemini web UI from Python via Playwright"
 readme = "README.md"
 requires-python = ">=3.12"

{gemx-0.2.2 → gemx-0.2.4}/src/gemx/__init__.py

@@ -28,4 +28,4 @@ __all__ = [
     "parse_output",
 ]
 
-__version__ = "0.2.2"
+__version__ = "0.2.4"

{gemx-0.2.2 → gemx-0.2.4}/src/gemx/client.py

@@ -26,6 +26,7 @@ from pathlib import Path
 from typing import Any, AsyncIterator
 
 from playwright.async_api import Page, async_playwright
+from playwright.async_api import TimeoutError as PlaywrightTimeoutError
 
 from .errors import InputError, ProfileBusyError, ResponseTimeoutError
 from .formats import OutputFormat, format_instruction, parse_output
@@ -95,6 +96,7 @@ CANNED_REPLIES = (
     "i'm having a hard time fulfilling your request",
     "i seem to be encountering an error",
     "i encountered an error doing what you asked",
+    "something went wrong. please try your request again",
 )
 
 
@@ -180,6 +182,20 @@ _LONGEST_RESPONSE_JS = """(args) => {
     return { text: best, selector, length: best.length };
 }"""
 
+_LIVE_RESPONSES_JS = """(args) => {
+    const [selectors, initialCounts] = args;
+    const texts = [];
+    selectors.forEach(sel => {
+        const els = Array.from(document.querySelectorAll(sel));
+        const initial = initialCounts[sel] || 0;
+        els.slice(initial).forEach(el => {
+            const t = el.innerText || el.textContent || '';
+            if (t) texts.push(t);
+        });
+    });
+    return texts;
+}"""
+
 _PAGE_ELEMENTS_JS = """() => {
     return {
         hasTextarea: document.querySelector('textarea') !== null,
@@ -337,7 +353,7 @@ class Gemx:
                 for attempt in range(1, attempts + 1):
                     try:
                         return await self._ask_raw_once(prompt, expected_format)
-                    except ResponseTimeoutError:
+                    except (ResponseTimeoutError, PlaywrightTimeoutError):
                         if attempt >= attempts:
                             raise
                         logger.warning(
@@ -421,7 +437,10 @@ class Gemx:
         logger.debug("navigating to %s", GEMINI_URL)
         await page.goto(GEMINI_URL, wait_until="load", timeout=cfg.nav_timeout_ms)
         await page.wait_for_load_state("domcontentloaded")
-        await page.wait_for_load_state("networkidle", timeout=30_000)
+        try:
+            await page.wait_for_load_state("networkidle", timeout=30_000)
+        except PlaywrightTimeoutError:
+            logger.debug("network never went idle; continuing")
         await page.wait_for_timeout(5_000)
 
         await page.wait_for_timeout(cfg.post_load_settle_ms)
@@ -573,14 +592,20 @@ class Gemx:
                     current_length,
                     len(best),
                 )
-                if (
-                    stable >= cfg.stable_ticks
-                    and expected_format in (OutputFormat.JSON, OutputFormat.XML)
-                    and is_canned_reply(best)
+                if stable >= cfg.stable_ticks and expected_format in (
+                    OutputFormat.JSON,
+                    OutputFormat.XML,
                 ):
-                    raise ResponseTimeoutError(
-                        f"Gemini returned a canned reply: {best[:200]!r}"
+                    live = await page.evaluate(
+                        _LIVE_RESPONSES_JS, [list(RESPONSE_SELECTORS), initial_counts]
+                    )
+                    canned = next(
+                        (t for t in [best, *live] if is_canned_reply(t)), None
                     )
+                    if canned is not None:
+                        raise ResponseTimeoutError(
+                            f"Gemini returned a canned reply: {canned[:200]!r}"
+                        )
                 if stable >= cfg.stable_ticks and len(best) > cfg.min_response_chars:
                     if expected_format in (OutputFormat.JSON, OutputFormat.XML):
                         try: