nixpkgs/pkgs/servers/unstructured-api/default.nix
2023-10-17 07:51:06 +00:00

69 lines
1.7 KiB
Nix

{
lib,
stdenvNoCC,
fetchFromGitHub,
python3,
makeWrapper,
nix-update-script,
symlinkJoin,
nltk-data,
}:
let
pythonEnv = python3.withPackages (packages: with packages; [
unstructured-api-tools
unstructured
pydantic
click
ratelimit
requests
pypdf
pycryptodome
safetensors
uvicorn
] ++ packages.unstructured.optional-dependencies.local-inference);
version = "0.0.53";
unstructured_api_nltk_data = symlinkJoin {
name = "unstructured_api_nltk_data";
paths = [ nltk-data.punkt nltk-data.averaged_perceptron_tagger ];
};
in stdenvNoCC.mkDerivation {
pname = "unstructured-api";
inherit version;
src = fetchFromGitHub {
owner = "Unstructured-IO";
repo = "unstructured-api";
rev = version;
hash = "sha256-lD3f6ws3xGFZsC1wHoWydIkWOgIwXuZencle+jJ8T3Y=";
};
nativeBuildInputs = [ makeWrapper ];
installPhase = ''
runHook preInstall
mkdir -p $out $out/bin $out/lib
cp -r . $out/lib
makeWrapper ${pythonEnv}/bin/uvicorn $out/bin/unstructured-api \
--set NLTK_DATA ${unstructured_api_nltk_data} \
--prefix PYTHONPATH : $out/lib \
--add-flags "prepline_general.api.app:app"
runHook postInstall
'';
passthru = {
updateScript = nix-update-script { };
};
meta = with lib; {
description = "open-source toolkit designed to make it easy to prepare unstructured data like PDFs, HTML and Word Documents for downstream data science tasks";
homepage = "https://github.com/Unstructured-IO/unstructured-api";
changelog = "https://github.com/Unstructured-IO/unstructured-api/releases/tag/${version}";
license = licenses.asl20;
maintainers = with maintainers; [ happysalada ];
};
}