nebu 0.1.21__py3-none-any.whl → 0.1.23__py3-none-any.whl

nebu/__init__.py

@@ -9,4 +9,5 @@ from .containers.models import *
 from .meta import *
 from .processors.decorate import *
 from .processors.models import *
+from .processors.models import V1StreamMessage as Message
 from .processors.processor import *

nebu/processors/consumer.py

@@ -260,7 +260,7 @@ except ResponseError as e:
 
 
 # Function to process messages
-def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None:
+def process_message(message_id: str, message_data: Dict[str, str]) -> None:
     # Initialize variables that need to be accessible in the except block
     return_stream = None
     user_id = None
@@ -268,14 +268,11 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
     print(f"Message data inner: {message_data}")
 
     try:
-        # Extract the JSON string payload from the b'data' field
-        # Redis keys/fields might be bytes even with decode_responses=True
-        payload_str = message_data.get(b"data")
-
-        # decode_responses=True should handle decoding, but Redis can be tricky.
-        # If errors persist, we might need to re-add explicit decode checks.
-        if not payload_str or not isinstance(payload_str, str):
-            # Add a more specific check if needed later based on runtime errors
+        # Extract the JSON string payload from the 'data' field
+        payload_str = message_data.get("data")
+
+        # decode_responses=True should mean payload_str is a string if found.
+        if not payload_str:
             raise ValueError(
                 f"Missing or invalid 'data' field (expected string): {message_data}"
             )
@@ -308,10 +305,10 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
 
         # --- Health Check Logic based on kind ---
         if kind == "HealthCheck":
-            print(f"Received HealthCheck message {message_id.decode('utf-8')}")
+            print(f"Received HealthCheck message {message_id}")
             health_response = {
                 "kind": "StreamResponseMessage",  # Respond with a standard message kind
-                "id": message_id.decode("utf-8"),
+                "id": message_id,
                 "content": {"status": "healthy", "checked_message_id": msg_id},
                 "status": "success",
                 "created_at": datetime.now().isoformat(),
@@ -327,7 +324,7 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
             assert isinstance(REDIS_STREAM, str)
             assert isinstance(REDIS_CONSUMER_GROUP, str)
             r.xack(REDIS_STREAM, REDIS_CONSUMER_GROUP, message_id)
-            print(f"Acknowledged HealthCheck message {message_id.decode('utf-8')}")
+            print(f"Acknowledged HealthCheck message {message_id}")
             return  # Exit early for health checks
         # --- End Health Check Logic ---
 
@@ -393,13 +390,14 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
             # Otherwise use the param type directly
             try:
                 if param_type_name in local_namespace:
-                    input_obj = local_namespace[param_type_name](**content)
+                    print(f"Validating content against {param_type_name}")
+                    input_obj = local_namespace[param_type_name].model_validate(content)
                 else:
                     # If we can't find the exact type, just pass the content directly
                     input_obj = content
             except Exception as e:
                 print(f"Error creating input model: {e}, using raw content")
-                input_obj = content
+                raise e
 
         print(f"Input object: {input_obj}")
 
@@ -413,7 +411,7 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
         # Prepare the response
         response = {
             "kind": "StreamResponseMessage",
-            "id": message_id.decode("utf-8"),
+            "id": message_id,
             "content": result,
             "status": "success",
             "created_at": datetime.now().isoformat(),
@@ -427,9 +425,7 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
             # Assert type again closer to usage for type checker clarity
             assert isinstance(return_stream, str)
             r.xadd(return_stream, {"data": json.dumps(response)})
-            print(
-                f"Processed message {message_id.decode('utf-8')}, result sent to {return_stream}"
-            )
+            print(f"Processed message {message_id}, result sent to {return_stream}")
 
         # Acknowledge the message
         # Assert types again closer to usage for type checker clarity
@@ -438,13 +434,13 @@ def process_message(message_id: bytes, message_data: Dict[bytes, bytes]) -> None
         r.xack(REDIS_STREAM, REDIS_CONSUMER_GROUP, message_id)
 
     except Exception as e:
-        print(f"Error processing message {message_id.decode('utf-8')}: {e}")
+        print(f"Error processing message {message_id}: {e}")
         traceback.print_exc()
 
         # Prepare the error response
         error_response = {
             "kind": "StreamResponseMessage",
-            "id": message_id.decode("utf-8"),
+            "id": message_id,
             "content": {
                 "error": str(e),
                 "traceback": traceback.format_exc(),

{nebu-0.1.21.dist-info → nebu-0.1.23.dist-info}/METADATA

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

{nebu-0.1.21.dist-info → nebu-0.1.23.dist-info}/RECORD

@@ -1,4 +1,4 @@
-nebu/__init__.py,sha256=Frz4LWVslmGeEE_ZjCIV5riSw8RC4ZJ-gb4iQPKCOSM,299
+nebu/__init__.py,sha256=mXfEfDp3kOi_HBMDGZVbQ26_M1JYgeHccjIIHzgEer0,357
 nebu/auth.py,sha256=rApCd-7_c3GpIb7gjCB79rR7SOcmkG7MmaTE6zMbvr0,1125
 nebu/config.py,sha256=XBY7uKgcJX9d1HGxqqpx87o_9DuF3maUlUnKkcpUrKU,4565
 nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
@@ -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=47ssdvJU7CGur0JoehNrmNGKwAAzVDVp7UTnphApQlI,20080
+nebu/processors/consumer.py,sha256=eB771cVKnjhGxTx2NTz-n4zKkibg9fxLLxCi4gB_FWU,19735
 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.21.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.21.dist-info/METADATA,sha256=SmlvWXJgnKLKph_8UCEtgVtLz80EuKrcxRNm3hdEl_8,1678
-nebu-0.1.21.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nebu-0.1.21.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.21.dist-info/RECORD,,
+nebu-0.1.23.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.23.dist-info/METADATA,sha256=ZOnuBrliRvCmAbndazULon2olbjde0-k33qzfPzcll0,1678
+nebu-0.1.23.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nebu-0.1.23.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.23.dist-info/RECORD,,