mirror of
https://github.com/neuml/txtai.git
synced 2026-10-12 02:19:52 +00:00
davidmezzetti
46d15a1edb
- Adds id map for all index types to support any id type (Closes #6) - Skip installing Faiss on Windows (Closes #1, Closes #8) - Default to Faiss when available, Annoy otherwise - Switch to faiss-cpu (Closes #2, Closes #9) - Add quantize flag for Faiss (Closes #10)
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
# pylint: disable = C0111
|
|
from setuptools import find_packages, setup
|
|
|
|
with open("README.md", "r") as f:
|
|
DESCRIPTION = f.read()
|
|
|
|
setup(name="txtai",
|
|
version="1.1.0",
|
|
author="NeuML",
|
|
description="AI-powered search engine",
|
|
long_description=DESCRIPTION,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/neuml/txtai",
|
|
project_urls={
|
|
"Documentation": "https://github.com/neuml/txtai",
|
|
"Issue Tracker": "https://github.com/neuml/txtai/issues",
|
|
"Source Code": "https://github.com/neuml/txtai",
|
|
},
|
|
license="Apache 2.0: http://www.apache.org/licenses/LICENSE-2.0",
|
|
packages=find_packages(where="src/python"),
|
|
package_dir={"": "src/python"},
|
|
keywords="search embedding machine-learning nlp",
|
|
python_requires=">=3.6",
|
|
install_requires=[
|
|
"annoy>=1.16.3",
|
|
"faiss-cpu>=1.6.3; os_name != 'nt'",
|
|
"fasttext>=0.9.2",
|
|
"hnswlib>=0.4.0",
|
|
"nltk>=3.5",
|
|
"numpy>=1.18.4",
|
|
"pymagnitude-lite>=0.1.43",
|
|
"regex>=2020.5.14",
|
|
"scikit-learn>=0.23.1",
|
|
"torch>=1.4.0",
|
|
"tqdm>=4.46.0",
|
|
"sentence-transformers>=0.3.3",
|
|
"transformers>=3.0.2"
|
|
],
|
|
classifiers=[
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Operating System :: OS Independent",
|
|
"Programming Language :: Python :: 3",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Topic :: Software Development",
|
|
"Topic :: Text Processing :: Indexing",
|
|
"Topic :: Utilities"
|
|
])
|