Source code for danoan.llm_assistant.runner.cli.cli

from danoan.llm_assistant.runner.cli.commands.run import run_parser as run
from danoan.llm_assistant.runner.cli.commands.session import session_parser as session
from danoan.llm_assistant.runner.cli.commands.setup import setup

import argparse




def main():
    parser = argparse.ArgumentParser(
        "llm-assistant", description="Collection of LLM applications"
    )
    subcommand_action = parser.add_subparsers()

    list_of_commands = [setup, run, session]
    for command in list_of_commands:
        command.extend_parser(subcommand_action)

    args = parser.parse_args()
    if "func" in args:
        args.func(**vars(args))
    elif "subommand_help" in args:
        args.subommand_help()
    else:
        parser.print_help()



if __name__ == "__main__":
    main()