promptflow

1
# ---------------------------------------------------------
2
# Copyright (c) Microsoft Corporation. All rights reserved.
3
# ---------------------------------------------------------
4

5
import os
6
import re
7
from io import open
8
from typing import Any, List, Match, cast
9

10
from setuptools import find_namespace_packages, setup
11

12
PACKAGE_NAME = "promptflow-rag"
13
PACKAGE_FOLDER_PATH = "promptflow"
14

15

16
def parse_requirements(file_name: str) -> List[str]:
17
    with open(file_name) as f:
18
        return [
19
            require.strip() for require in f
20
            if require.strip() and not require.startswith('#')
21
        ]
22

23

24
# Version extraction inspired from 'requests'
25
with open(os.path.join(PACKAGE_FOLDER_PATH, "version.txt"), "r") as fd:
26
    version_content = fd.read()
27
    print(version_content)
28
    version = cast(Match[Any], re.search(r'^VERSION\s*=\s*[\'"]([^\'"]*)[\'"]', version_content, re.MULTILINE)).group(1)
29
if not version:
30
    raise RuntimeError("Cannot find version information")
31

32
with open("README.md", encoding="utf-8") as f:
33
    readme = f.read()
34

35
with open("CHANGELOG.md", encoding="utf-8") as f:
36
    changelog = f.read()
37

38
setup(
39
    name=PACKAGE_NAME,
40
    version=version,
41
    description="Prompt flow index build",
42
    long_description_content_type="text/markdown",
43
    long_description=readme + "\n\n" + changelog,
44
    author="Microsoft Corporation",
45
    author_email="aml-pt-eng@microsoft.com",
46
    url="https://github.com/microsoft/promptflow",
47
    classifiers=[
48
        "Development Status :: 4 - Beta",
49
        "Programming Language :: Python",
50
        "Programming Language :: Python :: 3",
51
        "Programming Language :: Python :: 3 :: Only",
52
        "Programming Language :: Python :: 3.8",
53
        "Programming Language :: Python :: 3.9",
54
        "Programming Language :: Python :: 3.10",
55
        "Programming Language :: Python :: 3.11",
56
        "Programming Language :: Python :: 3.12",
57
        "License :: OSI Approved :: MIT License",
58
        "Operating System :: OS Independent",
59
    ],
60
    python_requires="<4.0,>=3.8",
61
    install_requires=parse_requirements('requirements.txt'),
62
    packages=find_namespace_packages(include=[f"{PACKAGE_FOLDER_PATH}.*"]),
63
    include_package_data=True,
64
    project_urls={
65
        "Bug Reports": "https://github.com/microsoft/promptflow/issues",
66
        "Source": "https://github.com/microsoft/promptflow",
67
    },
68
)
69