108 lines
2.6 KiB
Nix
108 lines
2.6 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
fetchurl,
|
|
|
|
# build-system
|
|
poetry-core,
|
|
|
|
# dependencies
|
|
docx2txt,
|
|
fastapi,
|
|
injector,
|
|
llama-index-core,
|
|
llama-index-readers-file,
|
|
python-multipart,
|
|
pyyaml,
|
|
transformers,
|
|
uvicorn,
|
|
watchdog,
|
|
|
|
# optional-dependencies
|
|
python,
|
|
huggingface-hub,
|
|
gradio,
|
|
|
|
# tests
|
|
nixosTests,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "private-gpt";
|
|
version = "0.6.2";
|
|
pyproject = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "zylon-ai";
|
|
repo = "private-gpt";
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-IYTysU3W/NrtBuLe3ZJkztVSK+gzjkGIg0qcBYzB3bs=";
|
|
};
|
|
|
|
build-system = [ poetry-core ];
|
|
|
|
pythonRelaxDeps = [
|
|
"cryptography"
|
|
"fastapi"
|
|
"llama-index-core"
|
|
"llama-index-readers-file"
|
|
];
|
|
|
|
dependencies = [
|
|
docx2txt
|
|
fastapi
|
|
injector
|
|
llama-index-core
|
|
llama-index-readers-file
|
|
python-multipart
|
|
pyyaml
|
|
transformers
|
|
uvicorn
|
|
watchdog
|
|
] ++ 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
|
|
passthru.cl100k_base.tiktoken = fetchurl {
|
|
url = "https://openaipublic.blob.core.windows.net/encodings/cl100k_base.tiktoken";
|
|
hash = "sha256-Ijkht27pm96ZW3/3OFE+7xAPtR0YyTWXoRO8/+hlsqc=";
|
|
};
|
|
|
|
optional-dependencies = with python.pkgs; {
|
|
embeddings-huggingface = [
|
|
huggingface-hub
|
|
llama-index-embeddings-huggingface
|
|
];
|
|
embeddings-ollama = [ llama-index-embeddings-ollama ];
|
|
embeddings-openai = [ llama-index-embeddings-openai ];
|
|
embeddings-sagemaker = [ boto3 ];
|
|
llms-ollama = [ llama-index-llms-ollama ];
|
|
llms-openai = [ llama-index-llms-openai ];
|
|
llms-openai-like = [ llama-index-llms-openai-like ];
|
|
llms-sagemaker = [ boto3 ];
|
|
ui = [ gradio ];
|
|
vector-stores-chroma = [ llama-index-vector-stores-chroma ];
|
|
vector-stores-postgres = [ llama-index-vector-stores-postgres ];
|
|
vector-stores-qdrant = [ llama-index-vector-stores-qdrant ];
|
|
};
|
|
|
|
postInstall = ''
|
|
cp settings*.yaml $out/${python.sitePackages}/private_gpt/
|
|
'';
|
|
|
|
pythonImportsCheck = [ "private_gpt" ];
|
|
|
|
passthru.tests = {
|
|
inherit (nixosTests) private-gpt;
|
|
};
|
|
|
|
meta = {
|
|
changelog = "https://github.com/zylon-ai/private-gpt/blob/${src.rev}/CHANGELOG.md";
|
|
description = "Interact with your documents using the power of GPT, 100% privately, no data leaks";
|
|
homepage = "https://github.com/zylon-ai/private-gpt";
|
|
license = lib.licenses.asl20;
|
|
mainProgram = "private-gpt";
|
|
maintainers = with lib.maintainers; [ GaetanLepage ];
|
|
};
|
|
}
|