JakeHillion/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #320049 from GaetanLepage/fastembed

Browse Source 
python311Packages.fastembed: 0.2.7 -> 0.3.0
This commit is contained in:
Yt 2024-06-15 13:56:59 +00:00 committed by GitHub
commit 2839e63e41
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 26 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
pkgs/development/python-modules/fastembed/default.nix
View File

@ -1,23 +1,31 @@
{ {
lib, lib,
buildPythonPackage, buildPythonPackage,
pythonOlder,
fetchFromGitHub, fetchFromGitHub,
pythonRelaxDepsHook,
# build-system
poetry-core,
# dependencies
huggingface-hub, huggingface-hub,
loguru, loguru,
pythonOlder, mmh3,
pythonRelaxDepsHook, numpy,
poetry-core,
onnx, onnx,
onnxruntime, onnxruntime,
pillow,
pystemmer,
requests, requests,
snowballstemmer,
tokenizers, tokenizers,
tqdm, tqdm,
pytestCheckHook,
}: }:
buildPythonPackage rec { buildPythonPackage rec {
pname = "fastembed"; pname = "fastembed";
version = "0.2.7"; version = "0.3.0";
pyproject = true; pyproject = true;
disabled = pythonOlder "3.8"; disabled = pythonOlder "3.8";
@ -26,7 +34,7 @@ buildPythonPackage rec {
owner = "qdrant"; owner = "qdrant";
repo = "fastembed"; repo = "fastembed";
rev = "refs/tags/v${version}"; rev = "refs/tags/v${version}";
hash = "sha256-ArLilvixzpHIGGAom4smX0jZ6lJSZe6tSGreeD+Pzmk="; hash = "sha256-Tfj0YdUW/Nnvn4+RoOWj9l0gDkWbpVgiADA09ht4xxM=";
}; };
build-system = [ poetry-core ]; build-system = [ poetry-core ];
@ -36,30 +44,32 @@ buildPythonPackage rec {
dependencies = [ dependencies = [
huggingface-hub huggingface-hub
loguru loguru
mmh3
numpy
onnx onnx
onnxruntime onnxruntime
pillow
pystemmer
requests requests
snowballstemmer
tokenizers tokenizers
tqdm tqdm
]; ];
pythonImportsCheck = [ "fastembed" ]; pythonImportsCheck = [ "fastembed" ];
pythonRelaxDeps = [ pythonRelaxDeps = [ "onnxruntime" ];
"huggingface-hub"
"onnxruntime"
"tokenizers"
];
nativeCheckInputs = [ pytestCheckHook ];
# there is one test and it requires network # there is one test and it requires network
doCheck = false; doCheck = false;
meta = with lib; { meta = {
description = "Fast, Accurate, Lightweight Python library to make State of the Art Embedding"; description = "Fast, Accurate, Lightweight Python library to make State of the Art Embedding";
homepage = "https://github.com/qdrant/fastembed"; homepage = "https://github.com/qdrant/fastembed";
license = licenses.asl20; changelog = "https://github.com/qdrant/fastembed/releases/tag/v${version}";
maintainers = with maintainers; [ happysalada ]; license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ happysalada ];
# terminate called after throwing an instance of 'onnxruntime::OnnxRuntimeException'
badPlatforms = [ "aarch64-linux" ];
}; };
} }