forked from Archives/langchain
41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
"""Set up the package."""
|
|
from pathlib import Path
|
|
|
|
from setuptools import find_packages, setup
|
|
|
|
with open(Path(__file__).absolute().parents[0] / "langchain" / "VERSION") as _f:
|
|
__version__ = _f.read().strip()
|
|
|
|
with open("README.md", "r") as f:
|
|
long_description = f.read()
|
|
|
|
LLM_DEPENDENCIES = ["cohere", "openai", "nlpcloud", "huggingface_hub"]
|
|
OTHER_DEPENDENCIES = [
|
|
"elasticsearch",
|
|
"google-search-results",
|
|
"wikipedia",
|
|
"faiss-cpu",
|
|
"sentence_transformers",
|
|
"transformers",
|
|
"spacy",
|
|
"nltk",
|
|
]
|
|
|
|
|
|
setup(
|
|
name="langchain",
|
|
version=__version__,
|
|
packages=find_packages(),
|
|
description="Building applications with LLMs through composability",
|
|
install_requires=["pydantic", "sqlalchemy", "numpy", "requests"],
|
|
long_description=long_description,
|
|
license="MIT",
|
|
url="https://github.com/hwchase17/langchain",
|
|
include_package_data=True,
|
|
long_description_content_type="text/markdown",
|
|
extras_require={
|
|
"llms": LLM_DEPENDENCIES,
|
|
"all": LLM_DEPENDENCIES + OTHER_DEPENDENCIES,
|
|
},
|
|
)
|