comfy-env 0.0.46__tar.gz → 0.0.48__tar.gz

{comfy_env-0.0.46 → comfy_env-0.0.48}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.46
+Version: 0.0.48
 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.0.46 → comfy_env-0.0.48}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "comfy-env"
-version = "0.0.46"
+version = "0.0.48"
 description = "Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation"
 readme = "README.md"
 license = {text = "MIT"}

{comfy_env-0.0.46 → comfy_env-0.0.48}/src/comfy_env/cli.py

@@ -42,6 +42,18 @@ def main(args: Optional[List[str]] = None) -> int:
 
     subparsers = parser.add_subparsers(dest="command", help="Available commands")
 
+    # init command
+    init_parser = subparsers.add_parser(
+        "init",
+        help="Create a default comfy-env.toml config file",
+        description="Initialize a new comfy-env.toml configuration file in the current directory",
+    )
+    init_parser.add_argument(
+        "--force", "-f",
+        action="store_true",
+        help="Overwrite existing config file",
+    )
+
     # install command
     install_parser = subparsers.add_parser(
         "install",
@@ -134,7 +146,9 @@ def main(args: Optional[List[str]] = None) -> int:
         return 0
 
     try:
-        if parsed.command == "install":
+        if parsed.command == "init":
+            return cmd_init(parsed)
+        elif parsed.command == "install":
             return cmd_install(parsed)
         elif parsed.command == "info":
             return cmd_info(parsed)
@@ -155,6 +169,42 @@ def main(args: Optional[List[str]] = None) -> int:
         return 1
 
 
+DEFAULT_CONFIG = """\
+# comfy-env.toml - Environment configuration for ComfyUI custom nodes
+# Documentation: https://github.com/PozzettiAndrea/comfy-env
+
+[system]
+# System packages required (apt on Linux, brew on macOS)
+linux = []
+
+[environment]
+python = "3.11"
+cuda_version = "auto"
+pytorch_version = "auto"
+
+[environment.cuda]
+# CUDA packages from comfy-env registry
+# Example: nvdiffrast = "0.4.0"
+
+[environment.packages]
+requirements = []
+"""
+
+
+def cmd_init(args) -> int:
+    """Handle init command."""
+    config_path = Path.cwd() / "comfy-env.toml"
+
+    if config_path.exists() and not args.force:
+        print(f"Config file already exists: {config_path}", file=sys.stderr)
+        print("Use --force to overwrite", file=sys.stderr)
+        return 1
+
+    config_path.write_text(DEFAULT_CONFIG)
+    print(f"Created {config_path}")
+    return 0
+
+
 def cmd_install(args) -> int:
     """Handle install command."""
     from .install import install

{comfy_env-0.0.46 → comfy_env-0.0.48}/src/comfy_env/workers/venv.py

@@ -475,6 +475,12 @@ class VenvWorker(Worker):
             env.update(self.extra_env)
             env["COMFYUI_ISOLATION_WORKER"] = "1"
 
+            # For conda/pixi environments, add lib dir to LD_LIBRARY_PATH
+            lib_dir = self.python.parent.parent / "lib"
+            if lib_dir.is_dir():
+                existing = env.get("LD_LIBRARY_PATH", "")
+                env["LD_LIBRARY_PATH"] = f"{lib_dir}:{existing}" if existing else str(lib_dir)
+
             # Run subprocess
             cmd = [
                 str(self.python),
@@ -843,6 +849,13 @@ class PersistentVenvWorker(Worker):
         env.update(self.extra_env)
         env["COMFYUI_ISOLATION_WORKER"] = "1"
 
+        # For conda/pixi environments, add lib dir to LD_LIBRARY_PATH
+        # This ensures libraries like libstdc++ from the env are used
+        lib_dir = self.python.parent.parent / "lib"
+        if lib_dir.is_dir():
+            existing = env.get("LD_LIBRARY_PATH", "")
+            env["LD_LIBRARY_PATH"] = f"{lib_dir}:{existing}" if existing else str(lib_dir)
+
         # On Windows, pass host Python directory so worker can add it via os.add_dll_directory()
         # This fixes "DLL load failed" errors for packages like opencv-python-headless
         if sys.platform == "win32":