setup.py

#!/usr/bin/env python

"""The setup script."""

from setuptools import setup, find_packages
from glob import glob

with open("README.md", encoding="utf8") as readme_file:
    readme = readme_file.read()

with open("HISTORY.rst", encoding="utf8") as history_file:
    history = history_file.read()

requirements = [
    "Click>=7.0",
    "requests>=2.28",
    "langchain>=0.0.232",
    "llama-hub",
    "llama-index",
    "uvicorn==0.23.0",
    "fastapi>=0.95.2",
    "weaviate-client",
    "transformers",
    "gpt4all>=1.0.8",
    "jsonschema==4.18.6",
    "Jinja2==3.1.2",
    "torch",
    "chromadb==0.4.5",
    "pypdf==3.14.0",
    "sentence-transformers==2.2.2",
]

test_requirements = []

setup(
    author="AIM by DPhi",
    author_email="support@aiplanet.com",
    python_requires=">=3.8",
    classifiers=[
        "Intended Audience :: Developers",
        "Intended Audience :: Education",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: Apache Software License",
        "Operating System :: OS Independent",
        "Topic :: Scientific/Engineering",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Topic :: Software Development",
        "Topic :: Software Development :: Libraries",
        "Topic :: Software Development :: Libraries :: Python Modules",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
    ],
    description="An end to end GenAI framework",
    entry_points={
        "console_scripts": [
            "genaistack=genai_stack.cli:main",
            "genai-stack=genai_stack.cli:main",
            "genaistk=genai_stack.cli:main",
        ],
    },
    install_requires=requirements,
    license="Apache 2.0",
    long_description=readme + "\n\n" + history,
    include_package_data=True,
    keywords=[
        "MLOps",
        "AI",
        "GenAI Stack",
        "Model Serving",
        "Model Deployment",
        "LLMOps",
        "Large Language Model",
        "Generative AI",
        "gpt4all",
        "Transformers",
        "chatgpt",
        "LLM Chat",
    ],
    package_data={"genai_stack": ["genai_stack/install/templates/**/*.json", "genai_stack/install/templates/**/*.j2"]},
    data_files=[
        ("json", glob("genai_stack/install/templates/**/*.json", recursive=True)),
        ("j2", glob("genai_stack/install/templates/**/*.j2", recursive=True)),
    ],
    name="genai_stack",
    packages=find_packages(include=["genai_stack", "genai_stack.*"]),
    test_suite="tests",
    tests_require=test_requirements,
    url="https://github.com/aiplanethub/genai-stack",
    version="0.2.1",
    zip_safe=False,
)