code-puppy 0.0.350__py3-none-any.whl → 0.0.351__py3-none-any.whl

code_puppy/plugins/antigravity_oauth/antigravity_model.py

@@ -215,9 +215,39 @@ class AntigravityModel(GoogleModel):
         response = await client.post(url, json=body)
 
         if response.status_code != 200:
-            raise RuntimeError(
-                f"Antigravity API Error {response.status_code}: {response.text}"
-            )
+            # Check for corrupted thought signature error and retry
+            # Error 400: { error: { code: 400, message: Corrupted thought signature., status: INVALID_ARGUMENT } }
+            error_text = response.text
+            if (
+                response.status_code == 400
+                and "Corrupted thought signature" in error_text
+            ):
+                logger.warning(
+                    "Received 400 Corrupted thought signature. Backfilling signatures and retrying."
+                )
+                _backfill_thought_signatures(messages)
+
+                # Re-map messages
+                system_instruction, contents = await self._map_messages(
+                    messages, model_request_parameters
+                )
+
+                # Update body
+                body["contents"] = contents
+                if system_instruction:
+                    body["systemInstruction"] = system_instruction
+
+                # Retry request
+                response = await client.post(url, json=body)
+                # Check error again after retry
+                if response.status_code != 200:
+                    raise RuntimeError(
+                        f"Antigravity API Error {response.status_code}: {response.text}"
+                    )
+            else:
+                raise RuntimeError(
+                    f"Antigravity API Error {response.status_code}: {error_text}"
+                )
 
         data = response.json()
 
@@ -318,24 +348,56 @@ class AntigravityModel(GoogleModel):
 
         # Create async generator for SSE events
         async def stream_chunks() -> AsyncIterator[dict[str, Any]]:
-            async with client.stream("POST", url, json=body) as response:
-                if response.status_code != 200:
-                    text = await response.aread()
-                    raise RuntimeError(
-                        f"Antigravity API Error {response.status_code}: {text.decode()}"
-                    )
+            retry_count = 0
+            while retry_count < 2:
+                should_retry = False
+                async with client.stream("POST", url, json=body) as response:
+                    if response.status_code != 200:
+                        text = await response.aread()
+                        error_msg = text.decode()
+                        if (
+                            response.status_code == 400
+                            and "Corrupted thought signature" in error_msg
+                            and retry_count == 0
+                        ):
+                            should_retry = True
+                        else:
+                            raise RuntimeError(
+                                f"Antigravity API Error {response.status_code}: {error_msg}"
+                            )
 
-                async for line in response.aiter_lines():
-                    line = line.strip()
-                    if not line:
-                        continue
-                    if line.startswith("data: "):
-                        json_str = line[6:]  # Remove 'data: ' prefix
-                        if json_str:
-                            try:
-                                yield json.loads(json_str)
-                            except json.JSONDecodeError:
+                    if not should_retry:
+                        async for line in response.aiter_lines():
+                            line = line.strip()
+                            if not line:
                                 continue
+                            if line.startswith("data: "):
+                                json_str = line[6:]  # Remove 'data: ' prefix
+                                if json_str:
+                                    try:
+                                        yield json.loads(json_str)
+                                    except json.JSONDecodeError:
+                                        continue
+                        return
+
+                # Handle retry outside the context manager
+                if should_retry:
+                    logger.warning(
+                        "Received 400 Corrupted thought signature in stream. Backfilling and retrying."
+                    )
+                    _backfill_thought_signatures(messages)
+
+                    # Re-map messages
+                    system_instruction, contents = await self._map_messages(
+                        messages, model_request_parameters
+                    )
+
+                    # Update body in place
+                    body["contents"] = contents
+                    if system_instruction:
+                        body["systemInstruction"] = system_instruction
+
+                    retry_count += 1
 
         # Create streaming response
         streamed = AntigravityStreamingResponse(
@@ -666,3 +728,12 @@ def _antigravity_process_response_from_parts(
         provider_details=vendor_details,
         provider_name=provider_name,
     )
+
+
+def _backfill_thought_signatures(messages: list[ModelMessage]) -> None:
+    """Backfill all thinking parts with the bypass signature."""
+    for m in messages:
+        if isinstance(m, ModelResponse):
+            for part in m.parts:
+                if isinstance(part, ThinkingPart):
+                    object.__setattr__(part, "signature", BYPASS_THOUGHT_SIGNATURE)

{code_puppy-0.0.350.dist-info → code_puppy-0.0.351.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.350
+Version: 0.0.351
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.350.dist-info → code_puppy-0.0.351.dist-info}/RECORD

@@ -124,7 +124,7 @@ code_puppy/plugins/__init__.py,sha256=gWgrXWoFpl-3Mxz2DAvxKW6SkCWrOnw-hKsY9O7nHc
 code_puppy/plugins/oauth_puppy_html.py,sha256=Wpa-V_NlRiBAvo_OXHuR7wvOH_jSt8L9HSFGiab6xI0,13058
 code_puppy/plugins/antigravity_oauth/__init__.py,sha256=1miHihSqRNXO20Vh_Gn9M3Aa2szh0gtdSCaKKj9nq0Q,362
 code_puppy/plugins/antigravity_oauth/accounts.py,sha256=GQit2-K24bsopmTZyscFUq3M0cAEO5WutHWnipVdgz8,14304
-code_puppy/plugins/antigravity_oauth/antigravity_model.py,sha256=g0_nXnMg288CvBE48CFgZ-iAqlYbtbi1dcB4Up6cYYc,26115
+code_puppy/plugins/antigravity_oauth/antigravity_model.py,sha256=T81dp_j59BAKxcE8lIt3XKJQVB9Wb0IFugcNeR8mdVQ,29195
 code_puppy/plugins/antigravity_oauth/config.py,sha256=BoQgqf5I2XoHWnBBo9vhCIc_XwPj9Mbp0Z95ygWwt78,1362
 code_puppy/plugins/antigravity_oauth/constants.py,sha256=qsrA10JJvzNuY0OobvvwCQcoGpILBninllcUUMKkUrQ,4644
 code_puppy/plugins/antigravity_oauth/oauth.py,sha256=ZHXJtZP63l6brOpX1WdLfuUClIleA79-4y36YUJc6Wo,15137
@@ -177,10 +177,10 @@ code_puppy/tools/browser/browser_scripts.py,sha256=sNb8eLEyzhasy5hV4B9OjM8yIVMLV
 code_puppy/tools/browser/browser_workflows.py,sha256=nitW42vCf0ieTX1gLabozTugNQ8phtoFzZbiAhw1V90,6491
 code_puppy/tools/browser/camoufox_manager.py,sha256=RZjGOEftE5sI_tsercUyXFSZI2wpStXf-q0PdYh2G3I,8680
 code_puppy/tools/browser/vqa_agent.py,sha256=DBn9HKloILqJSTSdNZzH_PYWT0B2h9VwmY6akFQI_uU,2913
-code_puppy-0.0.350.data/data/code_puppy/models.json,sha256=FMQdE_yvP_8y0xxt3K918UkFL9cZMYAqW1SfXcQkU_k,3105
-code_puppy-0.0.350.data/data/code_puppy/models_dev_api.json,sha256=wHjkj-IM_fx1oHki6-GqtOoCrRMR0ScK0f-Iz0UEcy8,548187
-code_puppy-0.0.350.dist-info/METADATA,sha256=vAuuWdLBDGW2HGexfUhPLdUeoyXpabrNvtyNgZV57sQ,27572
-code_puppy-0.0.350.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-code_puppy-0.0.350.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
-code_puppy-0.0.350.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.350.dist-info/RECORD,,
+code_puppy-0.0.351.data/data/code_puppy/models.json,sha256=FMQdE_yvP_8y0xxt3K918UkFL9cZMYAqW1SfXcQkU_k,3105
+code_puppy-0.0.351.data/data/code_puppy/models_dev_api.json,sha256=wHjkj-IM_fx1oHki6-GqtOoCrRMR0ScK0f-Iz0UEcy8,548187
+code_puppy-0.0.351.dist-info/METADATA,sha256=CeEJbr8iSJB0kJ-s-PqQ-Qcef7Bkr0leg-cWB8OI-FA,27572
+code_puppy-0.0.351.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+code_puppy-0.0.351.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.351.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.351.dist-info/RECORD,,