diff --git a/pkgs/development/python-modules/agentic-threat-hunting-framework/default.nix b/pkgs/development/python-modules/agentic-threat-hunting-framework/default.nix index c9b661eee919..cbce1c8e2635 100644 --- a/pkgs/development/python-modules/agentic-threat-hunting-framework/default.nix +++ b/pkgs/development/python-modules/agentic-threat-hunting-framework/default.nix @@ -5,23 +5,25 @@ setuptools, click, jinja2, + python-dotenv, pyyaml, rich, pytest-cov-stub, pytestCheckHook, scikit-learn, + requests, }: -buildPythonPackage rec { +buildPythonPackage (finalAttrs: { pname = "agentic-threat-hunting-framework"; - version = "0.2.2"; + version = "0.4.0"; pyproject = true; src = fetchFromGitHub { owner = "Nebulock-Inc"; repo = "agentic-threat-hunting-framework"; - tag = "v${version}"; - hash = "sha256-rt7WmBCbSqoZBpwGi7dzh8QDw8Iby3LSdavnCot1Hr0="; + tag = "v${finalAttrs.version}"; + hash = "sha256-WU58wQGlUgbOqcIE7EKtABNvTKtvTiRO9iJLW4gXDlI="; }; build-system = [ setuptools ]; @@ -29,6 +31,7 @@ buildPythonPackage rec { dependencies = [ click jinja2 + python-dotenv pyyaml rich ]; @@ -40,6 +43,7 @@ buildPythonPackage rec { nativeCheckInputs = [ pytest-cov-stub pytestCheckHook + requests ]; pythonImportsCheck = [ "athf" ]; @@ -47,8 +51,8 @@ buildPythonPackage rec { meta = { description = "Framework for agentic threat hunting"; homepage = "https://github.com/Nebulock-Inc/agentic-threat-hunting-framework"; - changelog = "https://github.com/Nebulock-Inc/agentic-threat-hunting-framework/releases/tag/${src.tag}"; + changelog = "https://github.com/Nebulock-Inc/agentic-threat-hunting-framework/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ fab ]; }; -} +})