fastrag

1
[options]
2
packages = find:
3

4
install_requires =
5
    farm-haystack==1.23.0
6
    farm-haystack[inference]==1.23.0
7
    transformers>=4.35.2
8
    datasets
9
    evaluate
10
    pandas
11
    nltk
12
    tqdm
13
    numba
14
    openpyxl
15
    numpy
16
    protobuf==3.20.2
17
    ujson
18
    accelerate
19
    fastapi
20
    uvicorn
21
    Pillow==10.1.0
22

23
[options.extras_require]
24
dev =
25
    black
26
    isort
27
    pytest
28
    pre-commit
29

30
benchmark =
31
    beir
32
    kilt @ git+https://github.com/facebookresearch/KILT.git
33

34
ui =
35
    streamlit
36
    st-annotated-text
37
    matplotlib
38
    streamlit_chat
39

40
elastic =
41
    farm-haystack[elasticsearch]
42

43
colbert =
44
    colbert-ai
45

46
faiss-gpu =
47
    faiss-gpu
48

49
faiss-cpu =
50
    faiss-cpu
51

52
qdrant =
53
    qdrant-haystack
54

55
knowledge_graph =
56
    spacy
57
    pyvis
58
    networkx
59

60
quantize_reranker =
61
    opencv-python-headless
62
    intel-extension-for-transformers==1.1
63
    neural_compressor==2.1.1
64
    pytrec_eval
65
    transformers>=4.28.1
66
    datasets >= 1.8.0
67
    sentencepiece !=0.1.92
68
    protobuf==3.20.2
69
    torch==1.13.1
70
    onnx==1.14.0
71
    onnxruntime==1.13.1
72
    onnxruntime-extensions==0.8.0
73
    sentence-transformers==2.2.2
74

75
intel =
76
    intel-extension-for-pytorch
77
    optimum[neural-compressor, ipex]
78

79
llama_cpp =
80
    llama-cpp-python
81

82
[flake8]
83
ignore = E501
84
max-line-length = 100
85