30ab8e04b9
Migrate OpenAPI and sample config export to subcommands "export-openapi" and "export-config". Also add a "download" subcommand that passes args to the TabbyAPI downloader. This allows models to be downloaded via the API and CLI args. Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
import argparse
|
|
import asyncio
|
|
import json
|
|
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):
|
|
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,
|
|
)
|
|
)
|
|
|
|
|
|
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
|