nebu 0.1.37__py3-none-any.whl → 0.1.38__py3-none-any.whl

nebu/processors/consumer.py

@@ -28,9 +28,9 @@ try:
     return_type_name = os.environ.get("RETURN_TYPE_NAME")
     content_type_name = os.environ.get("CONTENT_TYPE_NAME")
 
-    # Get before_func source if provided
-    before_func_source = os.environ.get("BEFORE_FUNC_SOURCE")
-    before_func_name = os.environ.get("BEFORE_FUNC_NAME")
+    # Get init_func source if provided
+    init_func_source = os.environ.get("INIT_FUNC_SOURCE")
+    init_func_name = os.environ.get("INIT_FUNC_NAME")
 
     # Check for generic type arguments
     input_model_args = []
@@ -215,19 +215,19 @@ try:
         traceback.print_exc()
         sys.exit(1)
 
-    # Execute before_func if provided
-    if before_func_source and before_func_name:
-        print(f"Executing before_func: {before_func_name}...")
+    # Execute init_func if provided
+    if init_func_source and init_func_name:
+        print(f"Executing init_func: {init_func_name}...")
         try:
-            exec(before_func_source, local_namespace)
-            before_function = local_namespace[before_func_name]
-            before_function()  # Call the function
-            print(f"Successfully executed before_func: {before_func_name}")
+            exec(init_func_source, local_namespace)
+            init_function = local_namespace[init_func_name]
+            init_function()  # Call the function
+            print(f"Successfully executed init_func: {init_func_name}")
         except Exception as e:
-            print(f"Error executing before_func '{before_func_name}': {e}")
+            print(f"Error executing init_func '{init_func_name}': {e}")
             traceback.print_exc()
             # Decide if failure is critical. For now, let's exit.
-            print("Exiting due to before_func failure.")
+            print("Exiting due to init_func failure.")
             sys.exit(1)
 
 except Exception as e:

nebu/processors/decorate.py

@@ -352,7 +352,7 @@ def processor(
     python_cmd: str = "python",
     no_delete: bool = False,
     include: Optional[List[Any]] = None,
-    before_func: Optional[Callable[[], None]] = None,
+    init_func: Optional[Callable[[], None]] = None,
 ):
     def decorator(
         func: Callable[[Any], Any],
@@ -636,71 +636,71 @@ def processor(
         # --- End Function Source ---
 
         # --- Get Before Function Source (if provided) ---
-        before_func_source = None
-        before_func_name = None
-        if before_func:
-            print(f"[DEBUG Decorator] Processing before_func: {before_func.__name__}")
-            before_func_name = before_func.__name__
+        init_func_source = None
+        init_func_name = None
+        if init_func:
+            print(f"[DEBUG Decorator] Processing init_func: {init_func.__name__}")
+            init_func_name = init_func.__name__
             # Validate signature (must take no arguments)
-            before_sig = inspect.signature(before_func)
+            before_sig = inspect.signature(init_func)
             if len(before_sig.parameters) != 0:
                 raise TypeError(
-                    f"before_func '{before_func_name}' must take zero parameters"
+                    f"init_func '{init_func_name}' must take zero parameters"
                 )
 
             # Try to get source using similar logic as the main function
             before_explicit_source = getattr(
-                before_func, _NEBU_EXPLICIT_SOURCE_ATTR, None
+                init_func, _NEBU_EXPLICIT_SOURCE_ATTR, None
             )
             if before_explicit_source:
                 print(
-                    f"[DEBUG Decorator] Using explicit source (@include) for before_func {before_func_name}"
+                    f"[DEBUG Decorator] Using explicit source (@include) for init_func {init_func_name}"
                 )
-                before_func_source = before_explicit_source
+                init_func_source = before_explicit_source
             elif in_jupyter and notebook_code:
                 print(
-                    f"[DEBUG Decorator] Attempting notebook history extraction for before_func '{before_func_name}'..."
+                    f"[DEBUG Decorator] Attempting notebook history extraction for init_func '{init_func_name}'..."
                 )
-                before_func_source = extract_definition_source_from_string(
-                    notebook_code, before_func_name, ast.FunctionDef
+                init_func_source = extract_definition_source_from_string(
+                    notebook_code, init_func_name, ast.FunctionDef
                 )
-                if before_func_source:
+                if init_func_source:
                     print(
-                        f"[DEBUG Decorator] Found before_func '{before_func_name}' source in notebook history."
+                        f"[DEBUG Decorator] Found init_func '{init_func_name}' source in notebook history."
                     )
                 else:
                     print(
-                        f"[DEBUG Decorator] Failed to find before_func '{before_func_name}' in notebook history, falling back to dill."
+                        f"[DEBUG Decorator] Failed to find init_func '{init_func_name}' in notebook history, falling back to dill."
                     )
 
-            if before_func_source is None:
+            if init_func_source is None:
                 print(
-                    f"[DEBUG Decorator] Using dill fallback for before_func '{before_func_name}'..."
+                    f"[DEBUG Decorator] Using dill fallback for init_func '{init_func_name}'..."
                 )
                 try:
-                    raw_before_func_source = dill.source.getsource(before_func)
-                    before_func_source = textwrap.dedent(raw_before_func_source)
+                    raw_init_func_source = dill.source.getsource(init_func)
+                    init_func_source = textwrap.dedent(raw_init_func_source)
                     print(
-                        f"[DEBUG Decorator] Successfully got source via dill for '{before_func_name}'."
+                        f"[DEBUG Decorator] Successfully got source via dill for '{init_func_name}'."
                     )
                 except (IOError, TypeError, OSError) as e:
                     print(
-                        f"[DEBUG Decorator] Dill fallback failed for '{before_func_name}': {e}"
+                        f"[DEBUG Decorator] Dill fallback failed for '{init_func_name}': {e}"
                     )
                     # Raise error if we couldn't get the source by any means
                     raise ValueError(
-                        f"Could not retrieve source for before_func '{before_func_name}': {e}"
+                        f"Could not retrieve source for init_func '{init_func_name}': {e}"
                     ) from e
 
-            if before_func_source is None:  # Final check
+            if init_func_source is None:  # Final check
                 raise ValueError(
-                    f"Failed to obtain source code for before_func '{before_func_name}' using any method."
+                    f"Failed to obtain source code for init_func '{init_func_name}' using any method."
                 )
             print(
-                f"[DEBUG Decorator] Final before_func source obtained for '{before_func_name}'."
+                f"[DEBUG Decorator] Final init_func source obtained for '{init_func_name}'."
             )
         else:
-            print("[DEBUG Decorator] No before_func provided.")
+            print("[DEBUG Decorator] No init_func provided.")
         # --- End Before Function Source ---
 
         # --- Get Model Sources ---
@@ -815,13 +815,11 @@ def processor(
         add_source_to_env("CONTENT_TYPE", content_type_source)
         add_source_to_env("STREAM_MESSAGE", stream_message_source)
 
-        # Add before_func source if available
-        if before_func_source and before_func_name:
-            print(
-                f"[DEBUG Decorator] Adding BEFORE_FUNC env vars for {before_func_name}"
-            )
-            all_env.append(V1EnvVar(key="BEFORE_FUNC_SOURCE", value=before_func_source))
-            all_env.append(V1EnvVar(key="BEFORE_FUNC_NAME", value=before_func_name))
+        # Add init_func source if available
+        if init_func_source and init_func_name:
+            print(f"[DEBUG Decorator] Adding INIT_FUNC env vars for {init_func_name}")
+            all_env.append(V1EnvVar(key="INIT_FUNC_SOURCE", value=init_func_source))
+            all_env.append(V1EnvVar(key="INIT_FUNC_NAME", value=init_func_name))
 
         print("[DEBUG Decorator] Adding type info env vars...")
         all_env.append(V1EnvVar(key="PARAM_TYPE_STR", value=param_type_str_repr))

{nebu-0.1.37.dist-info → nebu-0.1.38.dist-info}/METADATA

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

{nebu-0.1.37.dist-info → nebu-0.1.38.dist-info}/RECORD

@@ -11,16 +11,16 @@ 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=GxU-IM-U7B9Tq6eTwW2lBVe7jm-JfK0ux4ZR2HN3sfU,20693
-nebu/processors/decorate.py,sha256=Jn7rJfhJFDlbFGxLDNJzL_GOthX_MWDl1rtktl1kYFM,38286
+nebu/processors/consumer.py,sha256=cRYf8j7nwjQTzUT6Rtrsvso7SqzItkV9F104OqGAsPQ,20655
+nebu/processors/decorate.py,sha256=KZ43vOPzl3HVPB9P3QX6U94yTBnVi7ISoXQhUuEMO7g,38156
 nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
 nebu/processors/models.py,sha256=y40HoW-MEzDWB2dm_tsYlUy3Nf3s6eiLC0iGO9BoNog,3956
 nebu/processors/processor.py,sha256=068hLQKapWabNlhb_DtzqAJ7N7MGdr5UcjfZrb_MkFo,9732
 nebu/processors/remote.py,sha256=TeAIPGEMqnDIb7H1iett26IEZrBlcbPB_-DSm6jcH1E,1285
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.37.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.37.dist-info/METADATA,sha256=gcjQbyKOX5ZvQuZ1QnJEtSEugrQIXiUuDjH4e4PJKPI,1786
-nebu-0.1.37.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nebu-0.1.37.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.37.dist-info/RECORD,,
+nebu-0.1.38.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.38.dist-info/METADATA,sha256=8UJuoJ3UEfjvP9pmdyP_BmYRP-zHPNnei3tr3eUkPlE,1786
+nebu-0.1.38.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nebu-0.1.38.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.38.dist-info/RECORD,,