Add files using upload-large-folder tool

24678b9 verified about 1 month ago

4.86 kB

# nix/python.nix — uv2nix virtual environment builder
{
  python312,
  lib,
  callPackage,
  uv2nix,
  pyproject-nix,
  pyproject-build-systems,
  stdenv,
  # Filtered Python source (see lib.nix pythonSrc) — keeps JS/docs/skills
  # edits from invalidating the venv derivation.
  pythonSrc,
  dependency-groups ? [ "all" ],
}:
let
  workspace = uv2nix.lib.workspace.loadWorkspace { workspaceRoot = pythonSrc; };
  hacks = callPackage pyproject-nix.build.hacks { };

  overlay = workspace.mkPyprojectOverlay {
    sourcePreference = "wheel";
  };

  isAarch64Darwin = stdenv.hostPlatform.system == "aarch64-darwin";

  # Keep the workspace locked through uv2nix, but supply the local voice stack
  # from nixpkgs so wheel-only transitive artifacts do not break evaluation.
  mkPrebuiltPassthru = dependencies: {
    inherit dependencies;
    optional-dependencies = { };
    dependency-groups = { };
  };

  mkPrebuiltOverride =
    final: from: dependencies:
    hacks.nixpkgsPrebuilt {
      inherit from;
      prev = {
        nativeBuildInputs = [ final.pyprojectHook ];
        passthru = mkPrebuiltPassthru dependencies;
      };
    };

  # Legacy alibabacloud packages ship only sdists with setup.py/setup.cfg
  # and no pyproject.toml, so setuptools isn't declared as a build dep.
  buildSystemOverrides =
    final: prev:
    builtins.mapAttrs
      (
        name: _:
        prev.${name}.overrideAttrs (old: {
          nativeBuildInputs = (old.nativeBuildInputs or [ ]) ++ [ final.setuptools ];
        })
      )
      (
        lib.genAttrs [
          "alibabacloud-credentials-api"
          "alibabacloud-endpoint-util"
          "alibabacloud-gateway-dingtalk"
          "alibabacloud-gateway-spi"
          "alibabacloud-tea"
        ] (_: null)
      );

  pythonPackageOverrides =
    final: _prev:
    if isAarch64Darwin then
      {
        numpy = mkPrebuiltOverride final python312.pkgs.numpy { };

        pyarrow = mkPrebuiltOverride final python312.pkgs.pyarrow { };

        av = mkPrebuiltOverride final python312.pkgs.av { };

        humanfriendly = mkPrebuiltOverride final python312.pkgs.humanfriendly { };

        coloredlogs = mkPrebuiltOverride final python312.pkgs.coloredlogs {
          humanfriendly = [ ];
        };

        onnxruntime = mkPrebuiltOverride final python312.pkgs.onnxruntime {
          coloredlogs = [ ];
          numpy = [ ];
          packaging = [ ];
        };

        ctranslate2 = mkPrebuiltOverride final python312.pkgs.ctranslate2 {
          numpy = [ ];
          pyyaml = [ ];
        };

        faster-whisper = mkPrebuiltOverride final python312.pkgs.faster-whisper {
          av = [ ];
          ctranslate2 = [ ];
          huggingface-hub = [ ];
          onnxruntime = [ ];
          tokenizers = [ ];
          tqdm = [ ];
        };
      }
    else
      { };

  pythonSet =
    (callPackage pyproject-nix.build.packages {
      python = python312;
    }).overrideScope
      (
        lib.composeManyExtensions [
          pyproject-build-systems.overlays.default
          overlay
          buildSystemOverrides
          pythonPackageOverrides
          # ``setup.py`` permits wheel/sdist creation only from the sealed
          # Hermes derivation. This is deliberately a derivation environment
          # variable, not a devShell variable: ``nix develop -c uv build``
          # must remain blocked.
          (final: prev: {
            hermes-agent = prev.hermes-agent.overrideAttrs (_old: {
              HERMES_NIX_BUILD = "1";
            });
          })
        ]
      );

  # The editable venv points at the live checkout, so it uses an
  # UNFILTERED workspace rooted at a real path — mkEditablePyprojectOverlay
  # computes relative paths via lib.path.splitRoot, which rejects the
  # filtered pythonSrc (a cleanSourceWith set, not a path).  Filtering
  # buys nothing here anyway: the editable install reads from
  # $HERMES_PYTHON_SRC_ROOT at runtime.
  workspaceRoot = ./..;
  editableWorkspace = uv2nix.lib.workspace.loadWorkspace { inherit workspaceRoot; };
  editableOverlay = editableWorkspace.mkEditablePyprojectOverlay {
    root = "$HERMES_PYTHON_SRC_ROOT"; # resolved at shellHook time
  };

  editableSet = pythonSet.overrideScope (
    lib.composeManyExtensions [
      editableOverlay
      (final: prev: {
        hermes-agent = prev.hermes-agent.overrideAttrs (old: {
          # point straight at the real source instead of the filtered nix store copy
          src = workspaceRoot;
          nativeBuildInputs = old.nativeBuildInputs ++ final.resolveBuildSystem { editables = [ ]; };
        });
      })
    ]
  );
in
{
  venv = pythonSet.mkVirtualEnv "hermes-agent-env" {
    hermes-agent = dependency-groups;
  };
  editableVenv = editableSet.mkVirtualEnv "hermes-agent-editable-env" {
    hermes-agent = dependency-groups;
  };
}

	# nix/python.nix — uv2nix virtual environment builder
	{
	python312,
	lib,
	callPackage,
	uv2nix,
	pyproject-nix,
	pyproject-build-systems,
	stdenv,
	# Filtered Python source (see lib.nix pythonSrc) — keeps JS/docs/skills
	# edits from invalidating the venv derivation.
	pythonSrc,
	dependency-groups ? [ "all" ],
	}:
	let
	workspace = uv2nix.lib.workspace.loadWorkspace { workspaceRoot = pythonSrc; };
	hacks = callPackage pyproject-nix.build.hacks { };

	overlay = workspace.mkPyprojectOverlay {
	sourcePreference = "wheel";
	};

	isAarch64Darwin = stdenv.hostPlatform.system == "aarch64-darwin";

	# Keep the workspace locked through uv2nix, but supply the local voice stack
	# from nixpkgs so wheel-only transitive artifacts do not break evaluation.
	mkPrebuiltPassthru = dependencies: {
	inherit dependencies;
	optional-dependencies = { };
	dependency-groups = { };
	};

	mkPrebuiltOverride =
	final: from: dependencies:
	hacks.nixpkgsPrebuilt {
	inherit from;
	prev = {
	nativeBuildInputs = [ final.pyprojectHook ];
	passthru = mkPrebuiltPassthru dependencies;
	};
	};

	# Legacy alibabacloud packages ship only sdists with setup.py/setup.cfg
	# and no pyproject.toml, so setuptools isn't declared as a build dep.
	buildSystemOverrides =
	final: prev:
	builtins.mapAttrs
	(
	name: _:
	prev.${name}.overrideAttrs (old: {
	nativeBuildInputs = (old.nativeBuildInputs or [ ]) ++ [ final.setuptools ];
	})
	)
	(
	lib.genAttrs [
	"alibabacloud-credentials-api"
	"alibabacloud-endpoint-util"
	"alibabacloud-gateway-dingtalk"
	"alibabacloud-gateway-spi"
	"alibabacloud-tea"
	] (_: null)
	);

	pythonPackageOverrides =
	final: _prev:
	if isAarch64Darwin then
	{
	numpy = mkPrebuiltOverride final python312.pkgs.numpy { };

	pyarrow = mkPrebuiltOverride final python312.pkgs.pyarrow { };

	av = mkPrebuiltOverride final python312.pkgs.av { };

	humanfriendly = mkPrebuiltOverride final python312.pkgs.humanfriendly { };

	coloredlogs = mkPrebuiltOverride final python312.pkgs.coloredlogs {
	humanfriendly = [ ];
	};

	onnxruntime = mkPrebuiltOverride final python312.pkgs.onnxruntime {
	coloredlogs = [ ];
	numpy = [ ];
	packaging = [ ];
	};

	ctranslate2 = mkPrebuiltOverride final python312.pkgs.ctranslate2 {
	numpy = [ ];
	pyyaml = [ ];
	};

	faster-whisper = mkPrebuiltOverride final python312.pkgs.faster-whisper {
	av = [ ];
	ctranslate2 = [ ];
	huggingface-hub = [ ];
	onnxruntime = [ ];
	tokenizers = [ ];
	tqdm = [ ];
	};
	}
	else
	{ };

	pythonSet =
	(callPackage pyproject-nix.build.packages {
	python = python312;
	}).overrideScope
	(
	lib.composeManyExtensions [
	pyproject-build-systems.overlays.default
	overlay
	buildSystemOverrides
	pythonPackageOverrides
	# ``setup.py`` permits wheel/sdist creation only from the sealed
	# Hermes derivation. This is deliberately a derivation environment
	# variable, not a devShell variable: ``nix develop -c uv build``
	# must remain blocked.
	(final: prev: {
	hermes-agent = prev.hermes-agent.overrideAttrs (_old: {
	HERMES_NIX_BUILD = "1";
	});
	})
	]
	);

	# The editable venv points at the live checkout, so it uses an
	# UNFILTERED workspace rooted at a real path — mkEditablePyprojectOverlay
	# computes relative paths via lib.path.splitRoot, which rejects the
	# filtered pythonSrc (a cleanSourceWith set, not a path). Filtering
	# buys nothing here anyway: the editable install reads from
	# $HERMES_PYTHON_SRC_ROOT at runtime.
	workspaceRoot = ./..;
	editableWorkspace = uv2nix.lib.workspace.loadWorkspace { inherit workspaceRoot; };
	editableOverlay = editableWorkspace.mkEditablePyprojectOverlay {
	root = "$HERMES_PYTHON_SRC_ROOT"; # resolved at shellHook time
	};

	editableSet = pythonSet.overrideScope (
	lib.composeManyExtensions [
	editableOverlay
	(final: prev: {
	hermes-agent = prev.hermes-agent.overrideAttrs (old: {
	# point straight at the real source instead of the filtered nix store copy
	src = workspaceRoot;
	nativeBuildInputs = old.nativeBuildInputs ++ final.resolveBuildSystem { editables = [ ]; };
	});
	})
	]
	);
	in
	{
	venv = pythonSet.mkVirtualEnv "hermes-agent-env" {
	hermes-agent = dependency-groups;
	};
	editableVenv = editableSet.mkVirtualEnv "hermes-agent-editable-env" {
	hermes-agent = dependency-groups;
	};
	}