PathRAG/setup.py at main · ksmin23/PathRAG · GitHub

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
from setuptools import setup, find_packages

setup(
    name="PathRAG",
    version="0.1.0",
    description="PathRAG: Pruning Graph-based Retrieval Augmented Generation with Relational Paths",
    author="PathRAG Authors",
    license="MIT",
    packages=find_packages(exclude=["web_app*", "examples*", "docs*", "tests*"]),
    python_requires=">=3.10",
    install_requires=[
        "aioboto3",
        "aiohttp",
        "graspologic",
        "nano-vectordb",
        "networkx",
        "numpy",
        "openai",
        "pydantic",
        "litellm",
        "tenacity",
        "tiktoken",
        "tqdm",
    ],
    extras_require={
        "spanner": [
            "google-cloud-spanner",
        ],
        "huggingface": [
            "accelerate",
            "torch",
            "transformers",
        ],
        "ollama": [
            "ollama",
        ],
        "vllm": [
            "vllm",
            "modelscope",
        ],
        "api": [
            "fastapi",
            "uvicorn[standard]",
            "websockets",
            "python-dotenv",
            "python-multipart",
            "python-jose",
            "passlib",
            "bcrypt",
            "pyjwt",
            "aiosqlite",
            "sqlalchemy",
            "psycopg[binary,pool]",
            "PyPDF2",
            "python-docx",
            "docx2txt",
            "python-pptx",
            "openpyxl",
            "striprtf",
            "odfpy",
            "ebooklib",
            "beautifulsoup4",
            "matplotlib",
        ],
        "all": [
            "PathRAG[spanner,huggingface,ollama,vllm,api]",
        ],
    },
)