CodeConCat/setup.py at main · biostochastics/CodeConCat · GitHub

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
77
78
79
80
81
82
83
84
85
86
87
88
89
import os
import re

from setuptools import find_packages, setup


def get_version():
    """
    Extracts the CodeConCat version from codeconcat/version.py
    """
    version_file = os.path.join(os.path.dirname(__file__), "codeconcat", "version.py")
    with open(version_file, encoding="utf-8") as f:
        content = f.read()
        match = re.search(r'__version__\s*=\s*"([^"]+)"', content)
        if match:
            return match.group(1)
    return "0.0.0"


def get_long_description():
    try:
        with open("README.md", encoding="utf-8") as f:
            return f.read()
    except FileNotFoundError:
        return ""


extras_require = {
    "web": [
        "fastapi>=0.68.0",
        "uvicorn>=0.15.0",
        "pydantic>=1.8.0",
    ],
    "test": [
        "pytest>=7.4.0",
        "pytest-cov>=4.1.0",
        "pytest-asyncio>=0.21.1",
        "pytest-mock>=3.11.1",
    ],
    "token": [
        "tiktoken>=0.5.1",
    ],
    "security": [
        "transformers>=4.36.0",
    ],
    "all": [
        "fastapi>=0.68.0",
        "uvicorn>=0.15.0",
        "pydantic>=1.8.0",
        "pytest>=7.4.0",
        "pytest-cov>=4.1.0",
        "pytest-asyncio>=0.21.1",
        "pytest-mock>=3.11.1",
        "tiktoken>=0.5.1",
        "transformers>=4.36.0",
    ],
}

setup(
    name="codeconcat",
    version=get_version(),
    author="Sergey Kornilov",
    author_email="sergey.kornilov@biostochastics.com",
    description="An LLM-friendly code parser, aggregator and doc extractor",
    long_description=get_long_description(),
    long_description_content_type="text/markdown",
    packages=find_packages(),
    install_requires=[
        "pyyaml>=5.0",
        "pyperclip>=1.8.0",
    ],
    extras_require=extras_require,
    python_requires=">=3.8",
    entry_points={"console_scripts": ["codeconcat=codeconcat.cli:app"]},
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Operating System :: OS Independent",
    ],
    keywords="llm code parser documentation",
    project_urls={
        "Source": "https://github.com/biostochastics/codeconcat",
    },
)