ru-rag

1
from setuptools import setup, find_packages
2

3
from ru_rag import __version__
4

5

6
NAME = "ru-rag"
7
ROOT_PACKAGE = NAME.replace("-", "_")
8

9

10
setup(
11
    name=NAME,
12
    platforms="any",
13
    version=__version__,
14
    packages=find_packages(
15
        exclude=[f"{ROOT_PACKAGE}.tests", f"{ROOT_PACKAGE}.tests.*"]
16
    ),
17
    entry_points={
18
        "console_scripts": [
19
            f"populate_db = {ROOT_PACKAGE}.serve:populate_db",
20
            f"find_similar = {ROOT_PACKAGE}.serve:find_similar",
21
            f"answer = {ROOT_PACKAGE}.serve:answer",
22
        ],
23
    },
24
    install_requires=[
25
        # prod
26
        "chromadb==0.3.23",
27
        "huggingface-hub==0.14.1",
28
        "langchain==0.0.174",
29
        "llama-cpp-python==0.1.53",
30
        "sentence-transformers",
31

32
        # dev
33
        "autopep8",
34
        "black",
35
    ],
36
    classifiers=[
37
        "Programming Language :: Python",
38
        "Programming Language :: Python :: 3",
39
        "Programming Language :: Python :: 3.9",
40
    ],
41
)
42