tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchPypi,
5
6 # build-system
7 setuptools,
8
9 # dependencies
10 bitarray,
11 datasets,
12 flask,
13 python-dotenv,
14 ninja,
15 scipy,
16 tqdm,
17 transformers,
18 ujson,
19 gitpython,
20 torch,
21 faiss,
22}:
23
24buildPythonPackage rec {
25 pname = "colbert-ai";
26 version = "0.2.21";
27 pyproject = true;
28
29 src = fetchPypi {
30 inherit version;
31 pname = "colbert_ai";
32 hash = "sha256-qNb9tOInLysI7Tf45QlgchYNhBXR5AWFdRiYt35iW6s=";
33 };
34
35 pythonRemoveDeps = [ "git-python" ];
36
37 build-system = [
38 setuptools
39 ];
40
41 dependencies = [
42 bitarray
43 datasets
44 faiss
45 flask
46 gitpython
47 python-dotenv
48 ninja
49 scipy
50 torch
51 tqdm
52 transformers
53 ujson
54 ];
55
56 pythonImportsCheck = [ "colbert" ];
57
58 # There is no tests
59 doCheck = false;
60
61 meta = {
62 description = "Fast and accurate retrieval model, enabling scalable BERT-based search over large text collections in tens of milliseconds";
63 homepage = "https://github.com/stanford-futuredata/ColBERT";
64 license = lib.licenses.mit;
65 maintainers = with lib.maintainers; [
66 bachp
67 ];
68 };
69}