llama_index/setup.py at asif_bagel · hmmhmmhm/llama_index

Repository navigation

"""Set up the package."""

from pathlib import Path

import os

from setuptools import find_packages, setup

DEFAULT_PACKAGE_NAME = "llama_index"

PACKAGE_NAME = os.environ.get("PACKAGE_NAME_OVERRIDE", DEFAULT_PACKAGE_NAME)

with open(Path(__file__).absolute().parents[0] / "llama_index" / "VERSION") as _f:

__version__ = _f.read().strip()

with open("README.md", "r", encoding="utf-8") as f:

long_description = f.read()

install_requires = [

"tiktoken",

"dataclasses_json",

"langchain>=0.0.262,<=0.0.266",

"sqlalchemy>=2.0.15",

"numpy",

"tenacity>=8.2.0,<9.0.0",

"openai>=0.26.4",

"pandas",

"urllib3<2",

"fsspec>=2023.5.0",

"typing-inspect>=0.8.0",

"typing_extensions>=4.5.0",

"beautifulsoup4", # hotfix for langchain 0.0.212 bug

"nest_asyncio",

]

setup(

author="Jerry Liu",

name=PACKAGE_NAME,

version=__version__,

packages=find_packages(),

description="Interface between LLMs and your data",

install_requires=install_requires,

long_description=long_description,

license="MIT",

url="https://github.com/jerryjliu/llama_index",

include_package_data=True,

long_description_content_type="text/markdown",

)