c580893054
If an error is returned from HuggingFace, raise it to the calling function. Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
import argparse
|
|
import asyncio
|
|
import json
|
|
import traceback
|
|
from loguru import logger
|
|
|
|
from common.downloader import hf_repo_download
|
|
from common.tabby_config import generate_config_file
|
|
from common.utils import unwrap
|
|
from endpoints.server import export_openapi
|
|
|
|
|
|
def download_action(args: argparse.Namespace):
|
|
try:
|
|
asyncio.run(
|
|
hf_repo_download(
|
|
repo_id=args.repo_id,
|
|
folder_name=args.folder_name,
|
|
revision=args.revision,
|
|
token=args.token,
|
|
include=args.include,
|
|
exclude=args.exclude,
|
|
)
|
|
)
|
|
except Exception:
|
|
exception = traceback.format_exc()
|
|
logger.error(exception)
|
|
|
|
|
|
def config_export_action(args: argparse.Namespace):
|
|
export_path = unwrap(args.export_path, "config_sample.yml")
|
|
generate_config_file(filename=export_path)
|
|
|
|
|
|
def openapi_export_action(args: argparse.Namespace):
|
|
export_path = unwrap(args.export_path, "openapi.json")
|
|
openapi_json = export_openapi()
|
|
|
|
with open(export_path, "w") as f:
|
|
f.write(json.dumps(openapi_json))
|
|
logger.info("Successfully wrote OpenAPI spec to " + f"{export_path}")
|
|
|
|
|
|
def run_subcommand(args: argparse.Namespace) -> bool:
|
|
match args.actions:
|
|
case "download":
|
|
download_action(args)
|
|
return True
|
|
case "export-config":
|
|
config_export_action(args)
|
|
return True
|
|
case "export-openapi":
|
|
openapi_export_action(args)
|
|
return True
|
|
case _:
|
|
return False
|