comfy-env 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

comfy_env/isolation/wrap.py

@@ -67,6 +67,7 @@ def _get_worker(
     working_dir: Path,
     sys_path: list[str],
     lib_path: Optional[str] = None,
+    env_vars: Optional[dict] = None,
 ):
     """Get or create a persistent worker for the isolated environment."""
     cache_key = str(env_dir)
@@ -106,11 +107,14 @@ def _get_worker(
 
             print(f"[comfy-env] Starting isolated worker (MPWorker)")
             print(f"[comfy-env]   Env: {env_dir}")
+            if env_vars:
+                print(f"[comfy-env]   env_vars: {', '.join(f'{k}={v}' for k, v in env_vars.items())}")
 
             worker = MPWorker(
                 name=working_dir.name,
                 sys_path=sys_path,
                 lib_path=lib_path,
+                env_vars=env_vars,
             )
 
         _workers[cache_key] = worker
@@ -187,6 +191,7 @@ def _wrap_node_class(
     working_dir: Path,
     sys_path: list[str],
     lib_path: Optional[str] = None,
+    env_vars: Optional[dict] = None,
 ) -> type:
     """
     Wrap a node class so its FUNCTION method runs in the isolated environment.
@@ -232,7 +237,7 @@ def _wrap_node_class(
             print(f"[comfy-env] PROXY CALLED: {cls.__name__}.{func_name}", flush=True)
             print(f"[comfy-env]   kwargs keys: {list(kwargs.keys())}", flush=True)
 
-        worker = _get_worker(env_dir, working_dir, sys_path, lib_path)
+        worker = _get_worker(env_dir, working_dir, sys_path, lib_path, env_vars)
         if _DEBUG:
             print(f"[comfy-env]   worker alive: {worker.is_alive()}", flush=True)
 
@@ -322,6 +327,20 @@ def wrap_isolated_nodes(
         print(f"[comfy-env] Warning: No comfy-env.toml in {nodes_dir}")
         return node_class_mappings
 
+    # Read env_vars from comfy-env.toml
+    env_vars = {}
+    try:
+        if sys.version_info >= (3, 11):
+            import tomllib
+        else:
+            import tomli as tomllib
+        with open(config_file, "rb") as f:
+            config = tomllib.load(f)
+        env_vars_data = config.get("env_vars", {})
+        env_vars = {str(k): str(v) for k, v in env_vars_data.items()}
+    except Exception:
+        pass  # Ignore errors reading config
+
     # Find environment directory and paths
     env_dir = _find_env_dir(nodes_dir)
     site_packages, lib_dir = _find_env_paths(nodes_dir)
@@ -342,10 +361,12 @@ def wrap_isolated_nodes(
     print(f"[comfy-env]   site-packages: {site_packages}")
     if lib_path:
         print(f"[comfy-env]   lib: {lib_path}")
+    if env_vars:
+        print(f"[comfy-env]   env_vars: {', '.join(f'{k}={v}' for k, v in env_vars.items())}")
 
     # Wrap all node classes
     for node_name, node_cls in node_class_mappings.items():
         if hasattr(node_cls, "FUNCTION"):
-            _wrap_node_class(node_cls, env_dir, nodes_dir, sys_path, lib_path)
+            _wrap_node_class(node_cls, env_dir, nodes_dir, sys_path, lib_path, env_vars)
 
     return node_class_mappings

comfy_env/workers/mp.py

@@ -40,7 +40,7 @@ _SHUTDOWN = object()
 _CALL_METHOD = "call_method"
 
 
-def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None):
+def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None, env_vars=None):
     """
     Worker process main loop.
 
@@ -56,11 +56,16 @@ def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None):
         queue_out: Output queue for sending results
         sys_path_additions: Paths to add to sys.path
         lib_path: Path to add to LD_LIBRARY_PATH (for conda libraries)
+        env_vars: Environment variables to set (from comfy-env.toml)
     """
     import os
     import sys
     from pathlib import Path
 
+    # Apply env_vars FIRST (before any library imports that might check them)
+    if env_vars:
+        os.environ.update(env_vars)
+
     # Set worker mode env var
     os.environ["COMFYUI_ISOLATION_WORKER"] = "1"
 
@@ -424,7 +429,7 @@ class MPWorker(Worker):
     interpreter without inherited state from the parent.
     """
 
-    def __init__(self, name: Optional[str] = None, sys_path: Optional[list] = None, lib_path: Optional[str] = None):
+    def __init__(self, name: Optional[str] = None, sys_path: Optional[list] = None, lib_path: Optional[str] = None, env_vars: Optional[dict] = None):
         """
         Initialize the worker.
 
@@ -432,10 +437,12 @@ class MPWorker(Worker):
             name: Optional name for logging/debugging.
             sys_path: Optional list of paths to add to sys.path in worker process.
             lib_path: Optional path to add to LD_LIBRARY_PATH (for conda libraries).
+            env_vars: Optional environment variables to set in worker process.
         """
         self.name = name or "MPWorker"
         self._sys_path = sys_path or []
         self._lib_path = lib_path
+        self._env_vars = env_vars or {}
         self._process = None
         self._queue_in = None
         self._queue_out = None
@@ -504,7 +511,7 @@ class MPWorker(Worker):
             self._queue_out = ctx.Queue()
             self._process = ctx.Process(
                 target=_worker_loop,
-                args=(self._queue_in, self._queue_out, self._sys_path, self._lib_path),
+                args=(self._queue_in, self._queue_out, self._sys_path, self._lib_path, self._env_vars),
                 daemon=True,
             )
             self._process.start()

{comfy_env-0.1.2.dist-info → comfy_env-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.1.2
+Version: 0.1.3
 Summary: Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation
 Project-URL: Homepage, https://github.com/PozzettiAndrea/comfy-env
 Project-URL: Repository, https://github.com/PozzettiAndrea/comfy-env

{comfy_env-0.1.2.dist-info → comfy_env-0.1.3.dist-info}/RECORD

@@ -8,7 +8,7 @@ comfy_env/config/__init__.py,sha256=4Guylkb-FV8QxhFwschzpzbr2eu8y-KNgNT3_JOm9jc,
 comfy_env/config/parser.py,sha256=dA1lX5ExBEfCqUJwe4V5i_jn2NJ69bMq3c3ji3lMSV8,4295
 comfy_env/config/types.py,sha256=Sb8HO34xsSZu5YAc2K4M7Gb3QNevJlngf12hHiwuU0w,2140
 comfy_env/isolation/__init__.py,sha256=vw9a4mpJ2CFjy-PLe_A3zQ6umBQklgqWNxwn9beNw3g,175
-comfy_env/isolation/wrap.py,sha256=9bXxK9h4FMrFL9k4EAdILWxkjXYytVCJV68MuwwufK8,11485
+comfy_env/isolation/wrap.py,sha256=NACfz-EmKj6X5Ko6dshYjS-jUm-q8xm7uZWpr52858Y,12307
 comfy_env/pixi/__init__.py,sha256=BUrq7AQf3WDm0cHWh72B2xZbURNnDu2dCuELWiQCUiM,997
 comfy_env/pixi/core.py,sha256=BAaO-6a4XmnnNXZbfMla7egBNKSqgX9cSXHVQsQP6xI,12752
 comfy_env/pixi/cuda_detection.py,sha256=sqB3LjvGNdV4eFqiARQGfyecBM3ZiUmeh6nG0YCRYQw,9751
@@ -22,11 +22,11 @@ comfy_env/templates/comfy-env-instructions.txt,sha256=ve1RAthW7ouumU9h6DM7mIRX1M
 comfy_env/templates/comfy-env.toml,sha256=ROIqi4BlPL1MEdL1VgebfTHpdwPNYGHwWeigI9Kw-1I,4831
 comfy_env/workers/__init__.py,sha256=TMVG55d2XLP1mJ3x1d16H0SBDJZtk2kMC5P4HLk9TrA,1073
 comfy_env/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,2307
-comfy_env/workers/mp.py,sha256=pn_1RNyPTCq-oE_pTHWuMMROCkY-cwQnAhezJraT9OI,27013
+comfy_env/workers/mp.py,sha256=lKCdFJ0-OE5-ItJxcfXxoBTiESGU-5mUcTerU1WnUbk,27404
 comfy_env/workers/subprocess.py,sha256=UMhKhaJoSjv2-FWnwQq9TeMWtaDLIs3ajAFTq1ngdJw,57844
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env-0.1.2.dist-info/METADATA,sha256=y9VJ1ErAv5lWCtTbruKtEwwc1bWd1xe7W-2txfKzyXM,6945
-comfy_env-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.1.2.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.1.2.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.1.2.dist-info/RECORD,,
+comfy_env-0.1.3.dist-info/METADATA,sha256=EeO4REOwJBW0g-SOD4-LNl2XzxMzcE4WED50Yoje7zg,6945
+comfy_env-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.1.3.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.1.3.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.1.3.dist-info/RECORD,,