setup.py

from setuptools import setup, find_packages, Command
import os


class RunSystemInfo(Command):
    description = "Run init script: Gather System Details to Help AI Helper"
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        from setup import noizu_help_init
        si = noizu_help_init.system_info()
        noizu_help_init.update_config(si)

setup(
    name="noizu_ops",
    version="0.1.1",
    url='https://github.com/noizu-labs-ml/noizu-ops',
    packages=find_packages(),
    package_data={
        'noizu_ops': ['config/*.yml']
    },
    cmdclass={
     'run_system_info': RunSystemInfo,
    },
    install_requires=[
        "openai",
        "rich",
        "MarkupSafe",
        "pyyaml",
        "python-slugify",
        "psutil",
        "argcomplete"
    ],
    entry_points={
        "console_scripts": [
            "noizu-ops = noizu_ops.bin.noizu_ops:main",
        ],
        'argcomplete.completers': [
            'noizu-ops = noizu_ops.bin.noizu_ops:main'
        ]
    },
    author="Keith Brings",
    author_email="keith.brings@noizu.com",
    description="Command line LLM interface. Generate how to/answer system queries. Command passing in your system info to better target response, GPT self reflection used to improve final results. (but it's slow as heck.)",
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3.10",
    ],
)