verifiers 0.3.2.dev161__py3-none-any.whl → 0.3.2.dev162__py3-none-any.whl

verifiers/v1/serve/pool.py

@@ -185,57 +185,60 @@ class EnvServerPool:
             while True:
                 events = dict(await self._poller.poll())
                 if self.frontend in events:
-                    (
-                        client_id,
-                        request_id,
-                        method,
-                        payload,
-                    ) = await self.frontend.recv_multipart()
-                    if method == b"health":
-                        await self.frontend.send_multipart(
-                            [client_id, request_id, b"reply", _HEALTH]
+                    frames = await self.frontend.recv_multipart()
+                    # Drop malformed requests without skipping ready worker replies.
+                    if len(frames) != 4:
+                        logger.warning(
+                            "frontend: dropping malformed request (%d frames, expected 4)",
+                            len(frames),
                         )
-                    elif method == b"cancel":
-                        # Route the cancel to the worker holding the target run;
-                        # an unknown/finished target (or an unparseable payload
-                        # — one bad frame must not tear down the broker) is
-                        # answered inline
-                        try:
-                            body = msgpack.unpackb(payload, raw=False)
-                            target = str(body.get("request_id", "")).encode()
-                        except Exception:  # noqa: BLE001 - one bad frame must not kill the broker
-                            target = b""
-                        target_entry = pending.get(target)
-                        if target_entry is None:
+                    else:
+                        client_id, request_id, method, payload = frames
+                        if method == b"health":
                             await self.frontend.send_multipart(
-                                [client_id, request_id, b"reply", _CANCEL_MISS]
+                                [client_id, request_id, b"reply", _HEALTH]
                             )
+                        elif method == b"cancel":
+                            # Route the cancel to the worker holding the target run;
+                            # an unknown/finished target (or an unparseable payload
+                            # — one bad frame must not tear down the broker) is
+                            # answered inline
+                            try:
+                                body = msgpack.unpackb(payload, raw=False)
+                                target = str(body.get("request_id", "")).encode()
+                            except Exception:  # noqa: BLE001 - one bad frame must not kill the broker
+                                target = b""
+                            target_entry = pending.get(target)
+                            if target_entry is None:
+                                await self.frontend.send_multipart(
+                                    [client_id, request_id, b"reply", _CANCEL_MISS]
+                                )
+                            else:
+                                worker = target_entry["worker"]
+                                worker["active"] += 1
+                                pending[request_id] = {
+                                    "client_id": client_id,
+                                    "worker": worker,
+                                }
+                                in_flight += 1
+                                await worker["dealer"].send_multipart(
+                                    [request_id, method, payload]
+                                )
                         else:
-                            worker = target_entry["worker"]
+                            worker = min(self.workers, key=lambda w: w["active"])
                             worker["active"] += 1
                             pending[request_id] = {
                                 "client_id": client_id,
                                 "worker": worker,
                             }
                             in_flight += 1
+                            # forward without client_id — the DEALER identity is the worker's
+                            # `client_id`; we hold the real one in `pending`.
                             await worker["dealer"].send_multipart(
                                 [request_id, method, payload]
                             )
-                    else:
-                        worker = min(self.workers, key=lambda w: w["active"])
-                        worker["active"] += 1
-                        pending[request_id] = {
-                            "client_id": client_id,
-                            "worker": worker,
-                        }
-                        in_flight += 1
-                        # forward without client_id — the DEALER identity is the worker's
-                        # `client_id`; we hold the real one in `pending`.
-                        await worker["dealer"].send_multipart(
-                            [request_id, method, payload]
-                        )
-                        if self.elastic:
-                            self._maybe_scale_up(in_flight)
+                            if self.elastic:
+                                self._maybe_scale_up(in_flight)
                 for w in self.workers:
                     if w["dealer"] in events:
                         request_id, kind, data = await w["dealer"].recv_multipart(

{verifiers-0.3.2.dev161.dist-info → verifiers-0.3.2.dev162.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: verifiers
-Version: 0.3.2.dev161
+Version: 0.3.2.dev162
 Summary: Verifiers: Environments for LLM Reinforcement Learning
 Project-URL: Homepage, https://github.com/primeintellect-ai/verifiers
 Project-URL: Documentation, https://github.com/primeintellect-ai/verifiers

{verifiers-0.3.2.dev161.dist-info → verifiers-0.3.2.dev162.dist-info}/RECORD

@@ -154,7 +154,7 @@ verifiers/v1/serve/__init__.py,sha256=wOKuwfugzmDT4yIniwzCXlf4G6tv6Cg8Kc--WMa38B
 verifiers/v1/serve/client.py,sha256=E2y2x4qqYveWRWC9VZYQEu19aMEryPWTdXnYzbq5uTA,9829
 verifiers/v1/serve/delta.py,sha256=wSQnfrm57K2bxZLbOsIiR0u161oCtpH3fykYelXwwPo,13633
 verifiers/v1/serve/encoding.py,sha256=hBZFucAZK9riXOV3DaHcskq9zHTT8gVGkoFVOPfnr30,2499
-verifiers/v1/serve/pool.py,sha256=bT-FOlIuOiBtOasuPB-p9887tkpEqvQUJBgn1Q45ph0,15912
+verifiers/v1/serve/pool.py,sha256=pUvUhAvcTkLinFuTJJOp6pqi_sQdWsjJS4clLzgkRhc,16362
 verifiers/v1/serve/server.py,sha256=kaQFh_tVRcmRaQoB5o6CU4y8Fh9X-zFg-cmcvU2GExg,9913
 verifiers/v1/serve/types.py,sha256=Qw5IjiJhsZz4giSXUBi9dfInBnQZ5_xiUmgH2qp8jPI,1741
 verifiers/v1/tasksets/__init__.py,sha256=2ijE3qIlLoUA02g6iysOwWQylyBBQL1GhSpjv8_bp0w,521
@@ -193,8 +193,8 @@ verifiers/v1/utils/scope.py,sha256=bzUEiWDOVdbj7dXOwGOlzsQ_-Xu0j9NR5TalQB-8TS8,8
 verifiers/v1/utils/score.py,sha256=493yJVMw8teCu9JxapxMFPFzI0hNUqdo0Y2nGW4kckk,6200
 verifiers/v1/utils/trace_store.py,sha256=RsCDonGR-bs-tvEstcppiJm3jFmAdIiG83fA5KtCbAw,2801
 verifiers/v1/utils/version.py,sha256=-obEo_-l9-D8FLef4hYxncOe-uJpxrM1g2Hig_37Sgs,1607
-verifiers-0.3.2.dev161.dist-info/METADATA,sha256=YLQztvhTrnWx-C4jdpEGMn9mEdCzpXkBmBpFAMiiO4A,4161
-verifiers-0.3.2.dev161.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
-verifiers-0.3.2.dev161.dist-info/entry_points.txt,sha256=iugElcdWPKbQM7uFF0lZ8iUpHsNr17-BwEAAjJWxV3U,259
-verifiers-0.3.2.dev161.dist-info/licenses/LICENSE,sha256=v0RrUsdV3IDoZhrRce297IXS3xMHNJ-_LdLpFAUWb9k,1072
-verifiers-0.3.2.dev161.dist-info/RECORD,,
+verifiers-0.3.2.dev162.dist-info/METADATA,sha256=FocAQk6L9uycD720na7FpwMyDf47X7gEqG0E4lzrBUA,4161
+verifiers-0.3.2.dev162.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+verifiers-0.3.2.dev162.dist-info/entry_points.txt,sha256=iugElcdWPKbQM7uFF0lZ8iUpHsNr17-BwEAAjJWxV3U,259
+verifiers-0.3.2.dev162.dist-info/licenses/LICENSE,sha256=v0RrUsdV3IDoZhrRce297IXS3xMHNJ-_LdLpFAUWb9k,1072
+verifiers-0.3.2.dev162.dist-info/RECORD,,