ChainForge/setup.py at main · gauranshkumar/ChainForge · 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
70
71
72
73
74
75
from setuptools import setup, find_packages

# Dependency groups
rag_deps = [
    # RAGForge dependencies
    "grpcio",
    "numpy<2.0",  # numpy>=2.0 is not compatible with libraries like torch
    "pymupdf",
    "python-docx",
    "tiktoken",
    "nltk>=3.8",
    "transformers",
    "scikit-learn>=1.4.0",
    "sentence-transformers",
    "rank-bm25",
    "whoosh",
    "cohere",
    "chonkie>=1.0",
    "model2vec>=0.5.0",  # required by chonkie
    "pyarrow>=14.0,<=16.0.0",  # newer versions of pyarrow require CMake 3.25 or higher, which is not compatible with all systems
    "lancedb<0.18.0"  # pylance requires pyarrow 14 or higher. Later versions of LanceDB give strange errors with pyarrow<=16.0.0.
]

def readme():
    with open('README.md', encoding='utf-8') as f:
        return f.read()

setup(
    name="chainforge",
    version="0.3.7.0",
    packages=find_packages(),
    author="Ian Arawjo",
    description="A Visual Programming Environment for Prompt Engineering",
    long_description=readme(),
    long_description_content_type="text/markdown",
    keywords="prompt engineering LLM response evaluation",
    license="MIT",
    url="https://github.com/ianarawjo/ChainForge/",
    install_requires=[
        # Core package dependencies (pre-RAGForge)
        "flask>=2.2.3",
        "flask[async]",
        "flask_cors",
        "requests",
        "platformdirs",
        "urllib3==1.26.6",
        "openai",
        "cryptography",
        "mistune>=2.0",  # for LLM response markdown parsing
        "markitdown[pdf, docx, xlsx, xls, pptx]",
    ],
    extras_require={
        # Extra dependencies for functionality like RAGForge,
        # which may not be needed by all users
        "rag": rag_deps,
        "all": rag_deps,
    },
    entry_points={
        "console_scripts": [
            "chainforge = chainforge.app:main",
        ],
    },
    classifiers=[
        # Package classifiers
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3.12",
    ],
    python_requires=">=3.10",
    include_package_data=True,
)