File size: 2,052 Bytes
a2d6347 |
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 76 |
# Copyright (c) 2023 Microsoft
# Licensed under The MIT License [see LICENSE for details]
from setuptools import find_packages, setup
# PEP0440 compatible formatted version, see:
# https://www.python.org/dev/peps/pep-0440/
#
# release markers:
# X.Y
# X.Y.Z # For bugfix releases
#
# pre-release markers:
# X.YaN # Alpha release
# X.YbN # Beta release
# X.YrcN # Release Candidate
# X.Y # Final release
# version.py defines the VERSION and VERSION_SHORT variables.
# We use exec here so we don't import allennlp whilst setting up.
VERSION = {} # type: ignore
with open("promptwizard/version.py", "r") as version_file:
exec(version_file.read(), VERSION)
INSTALL_REQUIRES = [
"datasets",
"tiktoken",
"nltk",
"openai",
"azure-identity",
"azure-search-documents",
"pyyaml~=6.0.1",
"pyarrow==15.0.2",
"llama-index==0.11.10",
"llama-index-core==0.11.10",
"python-dotenv"
]
QUANLITY_REQUIRES = [
"black==21.4b0",
"flake8>=3.8.3",
"isort>=5.5.4",
"pre-commit",
"pytest",
"pytest-xdist",
]
DEV_REQUIRES = INSTALL_REQUIRES + QUANLITY_REQUIRES
setup(
name="promptwizard",
version=VERSION["VERSION"],
author="The PromptWizard team",
author_email="[email protected]",
description="Optimize Prompt",
long_description=open("README.md", encoding="utf8").read(),
long_description_content_type="text/markdown",
keywords="PromptWizard",
license="MIT License",
url="https://github.com/microsoft/PromptWizard",
classifiers=[
"Intended Audience :: Science/Research",
"Development Status :: 3 - Alpha",
"Programming Language :: Python :: 3",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
package_dir={"": "."},
packages=find_packages("."),
extras_require={
"dev": DEV_REQUIRES,
"quality": QUANLITY_REQUIRES,
},
install_requires=INSTALL_REQUIRES,
include_package_data=True,
python_requires=">=3.8.0",
zip_safe=False,
)
|