mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-07-24 09:26:08 +00:00
792232dcc5
This also follows what I have seen as the recommend way to define a file package like this. Also bumps minor versions from pip compile Testing: `pip install -e .` Everything should build as normal `❯ pip install -e . Obtaining file:///Users/trevor/dev/unstructured Installing build dependencies ... done Checking if build backend supports build_editable ... done Getting requirements to build editable ... done Preparing editable metadata (pyproject.toml) ... done Collecting scarf@ https://packages.unstructured.io/scarf.tgz (from unstructured==0.10.17.dev16) Using cached https://packages.unstructured.io/scarf.tgz (1.1 kB) Installing build dependencies ... done Getting requirements to build wheel ... done Preparing metadata (pyproject.toml) ... done` When new release goes out, I will test just plain pip install to verify that functionality still works
170 lines
6.7 KiB
Python
170 lines
6.7 KiB
Python
"""
|
|
setup.py
|
|
|
|
unstructured - pre-processing tools for unstructured data
|
|
|
|
Copyright 2022 Unstructured Technologies, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
from typing import List, Optional, Union
|
|
|
|
from setuptools import find_packages, setup
|
|
|
|
from unstructured.__version__ import __version__
|
|
|
|
|
|
def load_requirements(file_list: Optional[Union[str, List[str]]] = None) -> List[str]:
|
|
if file_list is None:
|
|
file_list = ["requirements/base.in"]
|
|
if isinstance(file_list, str):
|
|
file_list = [file_list]
|
|
requirements: List[str] = []
|
|
for file in file_list:
|
|
with open(file, encoding="utf-8") as f:
|
|
requirements.extend(f.readlines())
|
|
requirements = [
|
|
req for req in requirements if not req.startswith("#") and not req.startswith("-")
|
|
]
|
|
return requirements
|
|
|
|
|
|
csv_reqs = load_requirements("requirements/extra-csv.in")
|
|
doc_reqs = load_requirements("requirements/extra-docx.in")
|
|
docx_reqs = load_requirements("requirements/extra-docx.in")
|
|
epub_reqs = load_requirements("requirements/extra-epub.in")
|
|
image_reqs = load_requirements("requirements/extra-pdf-image.in")
|
|
markdown_reqs = load_requirements("requirements/extra-markdown.in")
|
|
msg_reqs = load_requirements("requirements/extra-msg.in")
|
|
odt_reqs = load_requirements("requirements/extra-odt.in")
|
|
org_reqs = load_requirements("requirements/extra-pandoc.in")
|
|
pdf_reqs = load_requirements("requirements/extra-pdf-image.in")
|
|
ppt_reqs = load_requirements("requirements/extra-pptx.in")
|
|
pptx_reqs = load_requirements("requirements/extra-pptx.in")
|
|
rtf_reqs = load_requirements("requirements/extra-pandoc.in")
|
|
rst_reqs = load_requirements("requirements/extra-pandoc.in")
|
|
tsv_reqs = load_requirements("requirements/extra-csv.in")
|
|
xlsx_reqs = load_requirements("requirements/extra-xlsx.in")
|
|
|
|
all_doc_reqs = list(
|
|
set(
|
|
csv_reqs
|
|
+ docx_reqs
|
|
+ epub_reqs
|
|
+ image_reqs
|
|
+ markdown_reqs
|
|
+ msg_reqs
|
|
+ odt_reqs
|
|
+ org_reqs
|
|
+ pdf_reqs
|
|
+ pptx_reqs
|
|
+ rtf_reqs
|
|
+ rst_reqs
|
|
+ tsv_reqs
|
|
+ xlsx_reqs,
|
|
),
|
|
)
|
|
|
|
|
|
setup(
|
|
name="unstructured",
|
|
description="A library that prepares raw documents for downstream ML tasks.",
|
|
long_description=open("README.md", encoding="utf-8").read(), # noqa: SIM115
|
|
long_description_content_type="text/markdown",
|
|
keywords="NLP PDF HTML CV XML parsing preprocessing",
|
|
url="https://github.com/Unstructured-IO/unstructured",
|
|
python_requires=">=3.7.0",
|
|
classifiers=[
|
|
"Development Status :: 4 - Beta",
|
|
"Intended Audience :: Developers",
|
|
"Intended Audience :: Education",
|
|
"Intended Audience :: Science/Research",
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Operating System :: OS Independent",
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: Python :: 3.8",
|
|
"Programming Language :: Python :: 3.9",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.11",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
],
|
|
author="Unstructured Technologies",
|
|
author_email="devops@unstructuredai.io",
|
|
license="Apache-2.0",
|
|
packages=find_packages(),
|
|
version=__version__,
|
|
entry_points={
|
|
"console_scripts": ["unstructured-ingest=unstructured.ingest.main:main"],
|
|
},
|
|
install_requires=[
|
|
# (Trevor): This is a simple hello world package that is used to track
|
|
# download count for this package using scarf.
|
|
'scarf @ https://packages.unstructured.io/scarf.tgz',
|
|
load_requirements()
|
|
],
|
|
extras_require={
|
|
# Document specific extra requirements
|
|
"all-docs": all_doc_reqs,
|
|
"csv": csv_reqs,
|
|
"doc": doc_reqs,
|
|
"docx": docx_reqs,
|
|
"epub": epub_reqs,
|
|
"image": image_reqs,
|
|
"md": markdown_reqs,
|
|
"msg": msg_reqs,
|
|
"odt": odt_reqs,
|
|
"org": org_reqs,
|
|
"pdf": pdf_reqs,
|
|
"ppt": ppt_reqs,
|
|
"pptx": pptx_reqs,
|
|
"rtf": rtf_reqs,
|
|
"rst": rst_reqs,
|
|
"tsv": tsv_reqs,
|
|
"xlsx": xlsx_reqs,
|
|
# Extra requirements for data connectors
|
|
"s3": load_requirements("requirements/ingest-s3.in"),
|
|
"azure": load_requirements("requirements/ingest-azure.in"),
|
|
"azure-cognitive-search": load_requirements(
|
|
"requirements/ingest-azure-cognitive-search.in",
|
|
),
|
|
"biomed": load_requirements("requirements/ingest-biomed.in"),
|
|
"discord": load_requirements("requirements/ingest-discord.in"),
|
|
"github": load_requirements("requirements/ingest-github.in"),
|
|
"gitlab": load_requirements("requirements/ingest-gitlab.in"),
|
|
"reddit": load_requirements("requirements/ingest-reddit.in"),
|
|
"notion": load_requirements("requirements/ingest-notion.in"),
|
|
"slack": load_requirements("requirements/ingest-slack.in"),
|
|
"wikipedia": load_requirements("requirements/ingest-wikipedia.in"),
|
|
"google-drive": load_requirements("requirements/ingest-google-drive.in"),
|
|
"gcs": load_requirements("requirements/ingest-gcs.in"),
|
|
"elasticsearch": load_requirements("requirements/ingest-elasticsearch.in"),
|
|
"dropbox": load_requirements("requirements/ingest-dropbox.in"),
|
|
"box": load_requirements("requirements/ingest-box.in"),
|
|
"onedrive": load_requirements("requirements/ingest-onedrive.in"),
|
|
"outlook": load_requirements("requirements/ingest-outlook.in"),
|
|
"confluence": load_requirements("requirements/ingest-confluence.in"),
|
|
"airtable": load_requirements("requirements/ingest-airtable.in"),
|
|
"sharepoint": load_requirements("requirements/ingest-sharepoint.in"),
|
|
"delta-table": load_requirements("requirements/ingest-delta-table.in"),
|
|
"salesforce": load_requirements("requirements/ingest-salesforce.in"),
|
|
"jira": load_requirements("requirements/ingest-jira.in"),
|
|
# Legacy extra requirements
|
|
"huggingface": load_requirements("requirements/huggingface.in"),
|
|
"local-inference": all_doc_reqs,
|
|
"paddleocr": load_requirements("requirements/extra-paddleocr.in"),
|
|
"openai": load_requirements("requirements/ingest-openai.in"),
|
|
},
|
|
package_dir={"unstructured": "unstructured"},
|
|
package_data={"unstructured": ["nlp/*.txt"]},
|
|
)
|