gemx 0.2.2__tar.gz → 0.2.3__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: gemx
-Version: 0.2.2
+Version: 0.2.3
 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.3}/pyproject.toml

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

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

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

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

@@ -95,6 +95,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 +181,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,
@@ -573,14 +588,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: