nebu 0.1.88__py3-none-any.whl → 0.1.91__py3-none-any.whl

nebu/containers/container.py

@@ -42,13 +42,14 @@ class Container:
         proxy_port: Optional[int] = None,
         authz: Optional[V1AuthzConfig] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         # Fallback to a default config if none is provided
         config = config or GlobalConfig.read()
         current_server = config.get_current_server_config()
         if not current_server:
             raise ValueError("No current server config found")
-        self.api_key = current_server.api_key
+        self.api_key = api_key or current_server.api_key
         self.nebu_host = current_server.server
         self.config = config
 
@@ -250,6 +251,7 @@ class Container:
         name: Optional[str] = None,
         namespace: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ) -> List[V1Container]:
         """
         Get a list of containers that match the optional name and/or namespace filters.
@@ -258,7 +260,7 @@ class Container:
         current_server = config.get_current_server_config()
         if not current_server:
             raise ValueError("No current server config found")
-        api_key = current_server.api_key
+        api_key = api_key or current_server.api_key
         nebu_host = current_server.server
 
         containers_url = f"{nebu_host}/v1/containers"
@@ -292,6 +294,7 @@ class Container:
         name: str,
         namespace: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         """
         Get a container from the remote server.
@@ -307,7 +310,7 @@ class Container:
         current_server = out.config.get_current_server_config()
         if not current_server:
             raise ValueError("No current server config found")
-        out.api_key = current_server.api_key
+        out.api_key = api_key or current_server.api_key
         out.nebu_host = current_server.server
         out.containers_url = f"{out.nebu_host}/v1/containers"
 

nebu/namespaces/namespace.py

@@ -22,6 +22,7 @@ class Namespace:
         labels: Optional[Dict[str, str]] = None,
         owner: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         self.config = config or GlobalConfig.read()
         if not self.config:
@@ -30,7 +31,7 @@ class Namespace:
         if not current_server:
             raise ValueError("No server config found")
         self.current_server = current_server
-        self.api_key = current_server.api_key
+        self.api_key = api_key or current_server.api_key
         self.nebu_host = current_server.server
         self.name = name
         self.labels = labels
@@ -80,11 +81,12 @@ class Namespace:
         cls,
         name: str,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         """
         Get a Namespace from the remote server.
         """
-        namespaces = cls.get(name=name, config=config)
+        namespaces = cls.get(name=name, config=config, api_key=api_key)
         if not namespaces:
             raise ValueError("Namespace not found")
         namespace_v1 = namespaces[0]
@@ -97,7 +99,7 @@ class Namespace:
         out.current_server = out.config.get_current_server_config()
         if not out.current_server:
             raise ValueError("No server config found")
-        out.api_key = out.current_server.api_key
+        out.api_key = api_key or out.current_server.api_key
         out.nebu_host = out.current_server.server
         out.namespaces_url = f"{out.nebu_host}/v1/namespaces"
         out.name = name
@@ -109,6 +111,7 @@ class Namespace:
         cls,
         name: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ) -> List[V1Namespace]:
         """
         Get a list of Namespaces that optionally match the name filter.
@@ -119,11 +122,12 @@ class Namespace:
         current_server = config.get_current_server_config()
         if not current_server:
             raise ValueError("No server config found")
+        api_key = api_key or current_server.api_key
         namespaces_url = f"{current_server.server}/v1/namespaces"
 
         response = requests.get(
             namespaces_url,
-            headers={"Authorization": f"Bearer {current_server.api_key}"},
+            headers={"Authorization": f"Bearer {api_key}"},
         )
         response.raise_for_status()
 

nebu/processors/consumer.py

@@ -919,69 +919,89 @@ consumer_name = f"consumer-{os.getpid()}-{socket.gethostname()}"  # More unique
 MIN_IDLE_TIME_MS = 60000  # Minimum idle time in milliseconds (e.g., 60 seconds)
 CLAIM_COUNT = 10  # Max messages to claim at once
 
+# Check if hot reloading should be disabled
+disable_hot_reload = os.environ.get("NEBU_DISABLE_HOT_RELOAD", "0").lower() in [
+    "1",
+    "true",
+]
+print(
+    f"[Consumer] Hot code reloading is {'DISABLED' if disable_hot_reload else 'ENABLED'}."
+)
+
 try:
     while True:
         print(
             f"[{datetime.now(timezone.utc).isoformat()}] --- Top of main loop ---"
         )  # Added log
         # --- Check for Code Updates ---
-        print(
-            f"[{datetime.now(timezone.utc).isoformat()}] Checking for code updates..."
-        )  # Added log
-        if entrypoint_abs_path:  # Should always be set after init
-            try:
-                current_mtime = os.path.getmtime(entrypoint_abs_path)
-                if current_mtime > last_load_mtime:
-                    print(
-                        f"[Consumer] Detected change in entrypoint file: {entrypoint_abs_path}. Reloading code..."
-                    )
-                    (
-                        reloaded_target_func,
-                        reloaded_init_func,
-                        reloaded_module,
-                        reloaded_namespace,
-                        new_mtime,
-                    ) = load_or_reload_user_code(
-                        _module_path,
-                        _function_name,
-                        entrypoint_abs_path,
-                        _init_func_name,
-                        _included_object_sources,
-                    )
-
-                    if reloaded_target_func is not None and reloaded_module is not None:
-                        print("[Consumer] Code reload successful. Updating functions.")
-                        target_function = reloaded_target_func
-                        init_function = reloaded_init_func  # Update init ref too, though it's already run
-                        imported_module = reloaded_module
-                        local_namespace = (
-                            reloaded_namespace  # Update namespace from includes
-                        )
-                        last_load_mtime = new_mtime
-                    else:
+        if not disable_hot_reload:
+            print(
+                f"[{datetime.now(timezone.utc).isoformat()}] Checking for code updates..."
+            )  # Added log
+            if entrypoint_abs_path:  # Should always be set after init
+                try:
+                    current_mtime = os.path.getmtime(entrypoint_abs_path)
+                    if current_mtime > last_load_mtime:
                         print(
-                            "[Consumer] Code reload failed. Continuing with previously loaded code."
+                            f"[Consumer] Detected change in entrypoint file: {entrypoint_abs_path}. Reloading code..."
+                        )
+                        (
+                            reloaded_target_func,
+                            reloaded_init_func,
+                            reloaded_module,
+                            reloaded_namespace,
+                            new_mtime,
+                        ) = load_or_reload_user_code(
+                            _module_path,
+                            _function_name,
+                            entrypoint_abs_path,
+                            _init_func_name,
+                            _included_object_sources,
                         )
-                        # Optionally: Send an alert/log prominently that reload failed
 
-            except FileNotFoundError:
+                        if (
+                            reloaded_target_func is not None
+                            and reloaded_module is not None
+                        ):
+                            print(
+                                "[Consumer] Code reload successful. Updating functions."
+                            )
+                            target_function = reloaded_target_func
+                            init_function = reloaded_init_func  # Update init ref too, though it's already run
+                            imported_module = reloaded_module
+                            local_namespace = (
+                                reloaded_namespace  # Update namespace from includes
+                            )
+                            last_load_mtime = new_mtime
+                        else:
+                            print(
+                                "[Consumer] Code reload failed. Continuing with previously loaded code."
+                            )
+                            # Optionally: Send an alert/log prominently that reload failed
+
+                except FileNotFoundError:
+                    print(
+                        f"[Consumer] Error: Entrypoint file '{entrypoint_abs_path}' not found during check. Cannot reload."
+                    )
+                    # Mark as non-runnable? Or just log?
+                    target_function = None  # Stop processing until file reappears?
+                    imported_module = None
+                    last_load_mtime = 0  # Reset mtime to force check next time
+                except Exception as e_reload_check:
+                    print(f"[Consumer] Error checking/reloading code: {e_reload_check}")
+                    traceback.print_exc()
+            else:
                 print(
-                    f"[Consumer] Error: Entrypoint file '{entrypoint_abs_path}' not found during check. Cannot reload."
+                    "[Consumer] Warning: Entrypoint absolute path not set, cannot check for code updates."
                 )
-                # Mark as non-runnable? Or just log?
-                target_function = None  # Stop processing until file reappears?
-                imported_module = None
-                last_load_mtime = 0  # Reset mtime to force check next time
-            except Exception as e_reload_check:
-                print(f"[Consumer] Error checking/reloading code: {e_reload_check}")
-                traceback.print_exc()
+            print(
+                f"[{datetime.now(timezone.utc).isoformat()}] Finished checking for code updates."
+            )  # Added log
         else:
+            # Log that hot reload is skipped if it's disabled
             print(
-                "[Consumer] Warning: Entrypoint absolute path not set, cannot check for code updates."
+                f"[{datetime.now(timezone.utc).isoformat()}] Hot reload check skipped (NEBU_DISABLE_HOT_RELOAD=1)."
             )
-        print(
-            f"[{datetime.now(timezone.utc).isoformat()}] Finished checking for code updates."
-        )  # Added log
 
         # --- Claim Old Pending Messages ---
         print(

nebu/processors/decorate.py

@@ -392,6 +392,7 @@ def processor(
     health_check: Optional[V1ContainerHealthCheck] = None,
     execution_mode: str = "inline",
     config: Optional[GlobalConfig] = None,
+    hot_reload: bool = True,
 ):
     def decorator(
         func: Callable[[Any], Any],
@@ -1052,6 +1053,28 @@ def processor(
         all_env.append(V1EnvVar(key="NEBU_EXECUTION_MODE", value=execution_mode))
         print(f"[DEBUG Decorator] Set NEBU_EXECUTION_MODE to: {execution_mode}")
 
+        # Add Hot Reload Configuration
+        if not hot_reload:
+            all_env.append(V1EnvVar(key="NEBU_DISABLE_HOT_RELOAD", value="1"))
+            print(
+                "[DEBUG Decorator] Set NEBU_DISABLE_HOT_RELOAD to: 1 (Hot reload disabled)"
+            )
+        else:
+            # Ensure it's explicitly '0' or unset if enabled (consumer defaults to enabled if var missing)
+            # Setting to "0" might be clearer than removing it if it was added by other means.
+            # Check if it exists and update, otherwise add "0"
+            existing_hot_reload_var = next(
+                (var for var in all_env if var.key == "NEBU_DISABLE_HOT_RELOAD"), None
+            )
+            if existing_hot_reload_var:
+                existing_hot_reload_var.value = "0"
+            else:
+                # Not strictly needed as consumer defaults to enabled, but explicit is good.
+                all_env.append(V1EnvVar(key="NEBU_DISABLE_HOT_RELOAD", value="0"))
+            print(
+                "[DEBUG Decorator] Hot reload enabled (NEBU_DISABLE_HOT_RELOAD=0 or unset)"
+            )
+
         # Add PYTHONPATH
         pythonpath_value = CONTAINER_CODE_DIR
         existing_pythonpath = next(

nebu/processors/processor.py

@@ -97,6 +97,7 @@ class Processor(Generic[InputType, OutputType]):
         max_replicas: Optional[int] = None,
         scale_config: Optional[V1Scale] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
         no_delete: bool = False,
     ):
         self.config = config or GlobalConfig.read()
@@ -106,7 +107,7 @@ class Processor(Generic[InputType, OutputType]):
         if not current_server:
             raise ValueError("No server config found")
         self.current_server = current_server
-        self.api_key = current_server.api_key
+        self.api_key = api_key or current_server.api_key
         self.orign_host = current_server.server
         self.name = name
         self.namespace = namespace
@@ -298,11 +299,14 @@ class Processor(Generic[InputType, OutputType]):
         name: str,
         namespace: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         """
         Get a Processor from the remote server.
         """
-        processors = cls.get(namespace=namespace, name=name, config=config)
+        processors = cls.get(
+            namespace=namespace, name=name, config=config, api_key=api_key
+        )
         if not processors:
             raise ValueError("Processor not found")
         processor_v1 = processors[0]
@@ -315,7 +319,7 @@ class Processor(Generic[InputType, OutputType]):
         out.current_server = out.config.get_current_server_config()
         if not out.current_server:
             raise ValueError("No server config found")
-        out.api_key = out.current_server.api_key
+        out.api_key = api_key or out.current_server.api_key
         out.orign_host = out.current_server.server
         out.processors_url = f"{out.orign_host}/v1/processors"
         out.name = name
@@ -337,6 +341,7 @@ class Processor(Generic[InputType, OutputType]):
         name: Optional[str] = None,
         namespace: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ) -> List[V1Processor]:
         """
         Get a list of Processors that match the optional name and/or namespace filters.
@@ -351,7 +356,7 @@ class Processor(Generic[InputType, OutputType]):
 
         response = requests.get(
             processors_url,
-            headers={"Authorization": f"Bearer {current_server.api_key}"},
+            headers={"Authorization": f"Bearer {api_key or current_server.api_key}"},
         )
         response.raise_for_status()
 

{nebu-0.1.88.dist-info → nebu-0.1.91.dist-info}/METADATA

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

{nebu-0.1.88.dist-info → nebu-0.1.91.dist-info}/RECORD

@@ -8,23 +8,23 @@ nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
 nebu/orign.py,sha256=SkVfHgpadwik58KCZCrjdV5EHY0dhpEhDvijzLxY11Y,2052
 nebu/builders/builder.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nebu/builders/models.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu/containers/container.py,sha256=yb7KaPTVXnEEAlrpdlUi4HNqF6P7z9bmwAILGlq6iqU,13502
+nebu/containers/container.py,sha256=jyqG-WLFMAFzYLW9Bagi74zJ-zUPq_jHmrxF_2HVWjQ,13652
 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/namespaces/models.py,sha256=EqUOpzhVBhvJw2P92ONDUbIgC31M9jMmcaG5vyOrsWg,497
-nebu/namespaces/namespace.py,sha256=Q_EDH7BgQrTkaDh_l4tbo22qpq-uARfIk8ZPBLjITGY,4967
-nebu/processors/consumer.py,sha256=DBlQJjbRrRiSgF5MawyO5vPlu4PEkA7c8lgZ_RQ9rJw,52266
+nebu/namespaces/namespace.py,sha256=LsbGiGBzVtFUtxCRayGrqr2X1tDzRep4RnNklfWCC1k,5160
+nebu/processors/consumer.py,sha256=RNc4DCuAcmFfSDIyprJrACuVfog6V5cWoqWt3zDRGqQ,53141
 nebu/processors/consumer_process_worker.py,sha256=UAmhrR1wilQnRPbbHYZ9jaIrDKs0LKsSHxbj4VFvFcQ,31969
-nebu/processors/decorate.py,sha256=jMh7OMamPdxGn7cMxQsOl5CEEmhZ1TXkMz8nCzBpVaU,54649
+nebu/processors/decorate.py,sha256=w8ZJwe8MNhBoW_LJzUWHOgXvo-39RnPsGvRUXSd9Hk4,55784
 nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
 nebu/processors/models.py,sha256=FnBJFxtaJkp-uIOs90qkJUBvOR80l2cdGnfmOIWIvVA,4058
-nebu/processors/processor.py,sha256=OgEK8Fz0ehSe_VFiNsxweVKZIckhgVvQQ11NNffYZqA,15848
+nebu/processors/processor.py,sha256=h1ZmD1rqTSuC9QpHkobx6VCUZoVkKXoDBLaB_xxfjys,16037
 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.88.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.88.dist-info/METADATA,sha256=aD9p65LkMiOaY0qlJV4wdCjrKaPHb1rwPdQcGCMJntk,1731
-nebu-0.1.88.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-nebu-0.1.88.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.88.dist-info/RECORD,,
+nebu-0.1.91.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.91.dist-info/METADATA,sha256=LVAMxSj83DGy_CAHMuY1RxH3UYAn2wQ5JSoecIxQeY0,1731
+nebu-0.1.91.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+nebu-0.1.91.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.91.dist-info/RECORD,,