mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-30 22:10:10 +00:00
46 lines
894 B
Python
46 lines
894 B
Python
import sys
|
|
import click
|
|
import os
|
|
import copy
|
|
import logging
|
|
|
|
sys.path.append(
|
|
os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
|
|
)
|
|
|
|
|
|
@click.group()
|
|
@click.option(
|
|
"--log-level",
|
|
required=False,
|
|
type=str,
|
|
default="warn",
|
|
help="Log level",
|
|
)
|
|
@click.version_option()
|
|
def cli(log_level: str):
|
|
# TODO not working now
|
|
logging.basicConfig(level=log_level, encoding="utf-8")
|
|
|
|
|
|
def add_command_alias(command, name: str, hidden: bool = False):
|
|
new_command = copy.deepcopy(command)
|
|
new_command.hidden = hidden
|
|
cli.add_command(new_command, name=name)
|
|
|
|
|
|
try:
|
|
from pilot.model.cli import model_cli_group
|
|
|
|
add_command_alias(model_cli_group, name="model")
|
|
except ImportError as e:
|
|
logging.warning(f"Integrating dbgpt model command line tool failed: {e}")
|
|
|
|
|
|
def main():
|
|
return cli()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|