diff --git a/pkgs/development/python-modules/private-gpt/default.nix b/pkgs/development/python-modules/private-gpt/default.nix index 2824587d7c25..4642f2dd3221 100644 --- a/pkgs/development/python-modules/private-gpt/default.nix +++ b/pkgs/development/python-modules/private-gpt/default.nix @@ -1,51 +1,56 @@ { lib, buildPythonPackage, - python, fetchFromGitHub, + fetchurl, + + # build-system poetry-core, + + # dependencies + docx2txt, fastapi, injector, llama-index-core, llama-index-readers-file, - huggingface-hub, python-multipart, pyyaml, transformers, uvicorn, watchdog, + + # optional-dependencies + python, + huggingface-hub, gradio, - fetchurl, - fetchpatch, + + # tests nixosTests, }: buildPythonPackage rec { pname = "private-gpt"; - version = "0.5.0"; + version = "0.6.2"; pyproject = true; src = fetchFromGitHub { owner = "zylon-ai"; repo = "private-gpt"; - rev = "v${version}"; - hash = "sha256-bjydzJhOJjmbflcJbuMyNsmby7HtNPFW3MY2Tw12cHw="; + rev = "refs/tags/v${version}"; + hash = "sha256-IYTysU3W/NrtBuLe3ZJkztVSK+gzjkGIg0qcBYzB3bs="; }; - patches = [ - # Fix a vulnerability, to be removed in the next bump version - # See https://github.com/zylon-ai/private-gpt/pull/1890 - (fetchpatch { - url = "https://github.com/zylon-ai/private-gpt/commit/86368c61760c9cee5d977131d23ad2a3e063cbe9.patch"; - hash = "sha256-4ysRUuNaHW4bmNzg4fn++89b430LP6AzYDoX2HplVH0="; - }) - ]; - build-system = [ poetry-core ]; - pythonRelaxDeps = [ "fastapi" ]; + pythonRelaxDeps = [ + "cryptography" + "fastapi" + "llama-index-core" + "llama-index-readers-file" + ]; dependencies = [ + docx2txt fastapi injector llama-index-core @@ -55,7 +60,7 @@ buildPythonPackage rec { transformers uvicorn watchdog - ] ++ lib.flatten (builtins.attrValues passthru.optional-dependencies); + ] ++ lib.flatten (builtins.attrValues optional-dependencies); # This is needed for running the tests and the service in offline mode, # See related issue at https://github.com/zylon-ai/private-gpt/issues/1870 @@ -64,7 +69,7 @@ buildPythonPackage rec { hash = "sha256-Ijkht27pm96ZW3/3OFE+7xAPtR0YyTWXoRO8/+hlsqc="; }; - passthru.optional-dependencies = with python.pkgs; { + optional-dependencies = with python.pkgs; { embeddings-huggingface = [ huggingface-hub llama-index-embeddings-huggingface @@ -98,6 +103,6 @@ buildPythonPackage rec { homepage = "https://github.com/zylon-ai/private-gpt"; license = lib.licenses.asl20; mainProgram = "private-gpt"; - maintainers = [ ]; + maintainers = with lib.maintainers; [ GaetanLepage ]; }; }