-
-
Notifications
You must be signed in to change notification settings - Fork 175
/
requirements.txt
60 lines (56 loc) · 1.32 KB
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
numpy<2.0.0 # temporal not using numpy 2.0.0
pandas>=2.1.0
tqdm
tiktoken>=0.7.0 # for counting token
openai>=1.0.0
rank_bm25 # for bm25 retrieval
transformers
pyyaml # for yaml file
pyarrow # for pandas with parquet
fastparquet # for pandas with parquet
sacrebleu # for bleu score
evaluate # for meteor and other scores
rouge_score # for rouge score
rich # for pretty logging
chromadb>=0.5.0 # for vectordb retrieval
click # for cli
Flask # for api server
torch # for monot5 reranker
sentencepiece # for monot5 reranker
cohere>=5.8.0 # for cohere services
tokenlog>=0.0.2 # for token logging
aiohttp # for async http requests
bert_score # for bert score
sentence-transformers # for sentence transformer reranker
FlagEmbedding # for flag embedding reranker
llmlingua # for longllmlingua
peft
### LlamaIndex ###
llama-index>=0.11.0
llama-index-core>=0.11.0
# readers
llama-index-readers-file
# Embeddings
llama-index-embeddings-openai
llama-index-embeddings-huggingface
# LLMs
llama-index-llms-openai>=0.2.7
llama-index-llms-huggingface
llama-index-llms-openai-like
llama-index-llms-ollama
# Retriever
llama-index-retrievers-bm25
# WebUI
streamlit
gradio
### Langchain ###
langchain-core>=0.3.0
langchain-unstructured>=0.1.5
langchain-upstage
langchain-community>=0.3.0
# autorag dashboard
panel
seaborn
ipykernel
ipywidgets
ipywidgets_bokeh