nebu 0.1.126__py3-none-any.whl → 0.1.128__py3-none-any.whl

nebu/processors/processor.py

@@ -363,19 +363,31 @@ class Processor(Generic[InputType, OutputType]):
         # Poll only if poll=True AND the initial request was configured not to wait (wait=False).
         if poll and not stream_data_wait_param:
             message_id = raw_response_json.get("message_id")
+            return_stream = raw_response_json.get("return_stream")
+
             if not message_id or not isinstance(message_id, str):
                 logger.error(
-                    f"Processor {processor_name}: Polling requested but 'message_id' (string) not found in initial response. Response: {raw_response_json}"
+                    f"Processor {processor_name}: Polling requested but 'message_id' (string) not found or invalid in initial response. Response: {raw_response_json}"
                 )
                 raise ValueError(
                     "Polling failed: 'message_id' (string) missing or invalid in initial server response."
                 )
+            if not return_stream or not isinstance(return_stream, str):
+                logger.error(
+                    f"Processor {processor_name}: Polling requested but 'return_stream' (string) not found or invalid in initial response. Response: {raw_response_json}"
+                )
+                raise ValueError(
+                    "Polling failed: 'return_stream' (string) missing or invalid in initial server response for polling."
+                )
 
             # Polling URL using self.orign_host for consistency
             polling_url = f"{self.orign_host}/v1/processors/{processor_namespace}/{processor_name}/return/{message_id}"
 
+            # Prepare polling payload
+            poll_payload = {"consumer_group": return_stream}
+
             logger.info(
-                f"Processor {processor_name}: Polling for message_id {message_id} at {polling_url}. Overall timeout: {timeout}s, Interval: {poll_interval_seconds}s."
+                f"Processor {processor_name}: Polling for message_id {message_id} at {polling_url} with payload {poll_payload}. Overall timeout: {timeout}s, Interval: {poll_interval_seconds}s."
             )
             polling_start_time = time.time()
 
@@ -401,7 +413,7 @@ class Processor(Generic[InputType, OutputType]):
                         polling_url,
                         headers={"Authorization": f"Bearer {current_op_api_key}"},
                         timeout=individual_poll_timeout,
-                        json={},  # Send an empty JSON body for POST
+                        json=poll_payload,
                     )
 
                     if poll_response.status_code == 200:

{nebu-0.1.126.dist-info → nebu-0.1.128.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.126
+Version: 0.1.128
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown

{nebu-0.1.126.dist-info → nebu-0.1.128.dist-info}/RECORD

@@ -19,11 +19,11 @@ nebu/processors/consumer_process_worker.py,sha256=h--eNFKaLbUayxn88mB8oGGdrU2liE
 nebu/processors/decorate.py,sha256=5p9pQrk_H8-Fj0UjsgSVCYx7Jk7KFuhMZtNhkKvpmkQ,61306
 nebu/processors/default.py,sha256=cy4ETMdbdRGkrvbYec1o60h7mGDlGN5JsuUph0ENtDU,364
 nebu/processors/models.py,sha256=8-TmKha2_QAnPlXcZxYjrCSPDCX7FFcMDMcHK77jK0U,4223
-nebu/processors/processor.py,sha256=k2i2CJxI1MHi0J0dLr1PU0DrlDYmi8HiKbrmXMn-3qc,35402
+nebu/processors/processor.py,sha256=2KAWGki6hfAD9Mg-0phEt6N45lz-cQQp8XQzvD92tF4,36047
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.126.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.126.dist-info/METADATA,sha256=Y-sZBw2xLDusfWLeNfk_0tq_17GWD6xe4VHo8hxDHOg,1798
-nebu-0.1.126.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nebu-0.1.126.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.126.dist-info/RECORD,,
+nebu-0.1.128.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.128.dist-info/METADATA,sha256=iNDy7WL9XqkL00UFNTjheMFFigGJ9cloYvsxOqFGwmI,1798
+nebu-0.1.128.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nebu-0.1.128.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.128.dist-info/RECORD,,