forked from KruxAI/ragbuilder
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
95 lines (94 loc) · 2.73 KB
/
setup.py
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
from setuptools import setup, find_packages
setup(
name='ragbuilder',
version='0.0.1',
author='Ashwin Aravind',
author_email='[email protected]',
description='Library of Rag Methods',
url='https://github.com/kruxai/ragbuilder',
package_dir={'': 'src'},
packages=find_packages(where='src'),
include_package_data=True,
python_requires='>=3.7',
package_data={
'ragbuilder': ['templates/*','static/*', 'eval.db'], # Adjust the package name and template directory as necessary
},
# package_data={'': ['logging_config.ini']},
entry_points={
'console_scripts': [
'ragbuilder=ragbuilder.ragbuilder:main',
],
},
install_requires=[
'Flask',
'pytest==7.2.1',
'pytest-xdist~=3.2.0',
'coverage~=7.1.0',
'black~=23.1.0',
'pytest-timeout~=2.1.0',
'pytest-env~= 0.8.1',
'python-dotenv',
'langchain',
'langchain-community',
'langchainhub',
'langchain-openai',
'langchain-chroma',
'bs4',
'langchain-core',
'unstructured',
'pdf2image',
'pdfminer.six',
'langchain_experimental',
'scikit-learn',
'ragas==0.1.7',
'inquirer',
'llama_index',
'chromadb',
'sentence-transformers',
'llama-index',
'llama-index-vector-stores-chroma',
'llama-index-readers-web',
'IPython',
'llama-index-retrievers-bm25',
'rake_nltk',
'llama-index-embeddings-langchain',
'llama-index-vector-stores-faiss',
'faiss-cpu',
'llama-index-llms-mistralai',
'llama-index-embeddings-mistralai',
'llama-index-embeddings-openai',
'llama-index-postprocessor-longllmlingua',
'llmlingua',
'llama_index-postprocessor-cohere_rerank',
'llama_index-postprocessor-jinaai_rerank',
'llama-index-postprocessor-rankgpt-rerank',
'llama-index-postprocessor-colbert-rerank',
'llama-index-postprocessor-rankllm-rerank',
'llama-index-llms-openai',
'langchain-huggingface',
'rank_bm25',
'ragas',
'flask',
'pandas',
# 'pillow_heif',
# 'opencv-python',
# 'onnx==1.16.0',
# # 'pikepdf==8.0.0',
# # 'unstructured-inference',
# 'pytesseract',
# 'unstructured',
# # 'unstructured[all-docs]',
'mixpanel',
'langchain_mistralai',
'langchain_community',
'huggingface_hub',
'datasets',
'langchain_text_splitters',
'llama-index-core',
'requests',
'markdown',
'singlestoredb',
'langchain_pinecone'
# other dependencies
],
)