nebu 0.1.17__py3-none-any.whl → 0.1.19__py3-none-any.whl

nebu/processors/consumer.py

@@ -262,15 +262,23 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
     user_id = None
 
     try:
-        # Get the message content from field 'data'
-        if b"data" not in message_data:
-            print(f"Message {message_id} has no 'data' field")
-            return
+        # Check for and decode the 'data' field
+        data_bytes = message_data.get(b"data")
+        if data_bytes is None:
+            # Raise an error if 'data' field is missing
+            raise ValueError("Message missing 'data' field")
 
-        # Parse the message data
-        raw_payload = json.loads(message_data[b"data"].decode("utf-8"))
+        try:
+            # Decode bytes to string and parse JSON
+            raw_payload_str = data_bytes.decode("utf-8")
+            raw_payload = json.loads(raw_payload_str)
+        except (json.JSONDecodeError, UnicodeDecodeError) as decode_error:
+            # Raise a specific error for decoding/parsing issues
+            raise ValueError(f"Invalid JSON payload: {decode_error}") from decode_error
+
+        print(f"Raw payload: {raw_payload}")
 
-        # Extract fields from the Rust structure
+        # Extract fields from the parsed payload
         # These fields are extracted for completeness and potential future use
         _ = raw_payload.get("kind", "")  # kind
         msg_id = raw_payload.get("id", "")  # msg_id
@@ -291,6 +299,8 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
         else:
             content = content_raw
 
+        print(f"Content: {content}")
+
         # For StreamMessage, construct the proper input object
         if is_stream_message and "V1StreamMessage" in local_namespace:
             # If we have a content type, try to construct it
@@ -298,6 +308,7 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
                 # Try to create the content type model first
                 try:
                     content_model = local_namespace[content_type_name](**content)
+                    print(f"Content model: {content_model}")
                     input_obj = local_namespace["V1StreamMessage"](
                         kind=_,
                         id=msg_id,
@@ -325,6 +336,7 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
                     )
             else:
                 # Just use the raw content
+                print(f"Using raw content")
                 input_obj = local_namespace["V1StreamMessage"](
                     kind=_,
                     id=msg_id,
@@ -348,9 +360,11 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
                 print(f"Error creating input model: {e}, using raw content")
                 input_obj = content
 
+        print(f"Input object: {input_obj}")
+
         # Execute the function
         result = target_function(input_obj)
-
+        print(f"Result: {result}")
         # If the result is a Pydantic model, convert to dict
         if hasattr(result, "model_dump"):
             result = result.model_dump()
@@ -365,6 +379,8 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
             "user_id": user_id,
         }
 
+        print(f"Response: {response}")
+
         # Send the result to the return stream
         if return_stream:
             r.xadd(return_stream, {"data": json.dumps(response)})
@@ -421,6 +437,8 @@ while True:
         stream_name, stream_messages = messages[0]
 
         for message_id, message_data in stream_messages:
+            print(f"Processing message {message_id}")
+            print(f"Message data: {message_data}")
             process_message(message_id, message_data)
 
     except redis.exceptions.ConnectionError as e:

nebu/processors/decorate.py

@@ -769,7 +769,7 @@ def processor(
         metadata = V1ResourceMetaRequest(
             name=processor_name, namespace=namespace, labels=labels
         )
-        consumer_command = f"{python_cmd} -m nebu.processors.consumer"
+        consumer_command = f"{python_cmd} -u -m nebu.processors.consumer"
         setup_commands = [
             f"{python_cmd} -m pip install dill pydantic redis nebu",  # Base deps
         ]

{nebu-0.1.17.dist-info → nebu-0.1.19.dist-info}/METADATA

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

{nebu-0.1.17.dist-info → nebu-0.1.19.dist-info}/RECORD

@@ -6,15 +6,15 @@ nebu/containers/container.py,sha256=yb7KaPTVXnEEAlrpdlUi4HNqF6P7z9bmwAILGlq6iqU,
 nebu/containers/decorator.py,sha256=uFtzlAXRHYZECJ-NPusY7oN9GXvdHrHDd_JNrIGr8aQ,3244
 nebu/containers/models.py,sha256=0j6NGy4yto-enRDh_4JH_ZTbHrLdSpuMOqNQPnIrwC4,6815
 nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,2513
-nebu/processors/consumer.py,sha256=0HJxRLoeRdN4xY6bjIxqr5bD5JpFSyKb5s-eS5oTy9s,16063
-nebu/processors/decorate.py,sha256=BlpRF9u-1q9IS-UDn0XFzoYIBqEU4Xkjkoge394exJs,34786
+nebu/processors/consumer.py,sha256=EbOhoFogS3W91BmJjqrFeipX_2G7lgAuUVGOpemcswM,16844
+nebu/processors/decorate.py,sha256=AeG1c1n8JtcexxAEf2sF2L2eKwVDaNQ5gvPs6EpazKo,34789
 nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
 nebu/processors/models.py,sha256=GvnI8UJrQSjHo2snP07cPfisCH90cEGTY-PZV5_AtXI,3654
 nebu/processors/processor.py,sha256=oy2YdI-cy6qQWxrZhpZahJV46oWZlu_Im-jm811R_oo,9667
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.17.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.17.dist-info/METADATA,sha256=m9oiMDrMt5x1x7wa51dsxwgzj-tUcX_hZrQrGN8NtyE,1678
-nebu-0.1.17.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nebu-0.1.17.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.17.dist-info/RECORD,,
+nebu-0.1.19.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.19.dist-info/METADATA,sha256=soUNeXLW9Ikz2heD7RSSeiUW8Rut2Q9bK-P9rAuwnes,1678
+nebu-0.1.19.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nebu-0.1.19.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.19.dist-info/RECORD,,