FineTurnX/scripts/start_api.py

"""
启动 API 服务器
"""

import argparse
import sys
import os

# 添加项目路径
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))

from finetunex.api.server import run_server


def main():
    parser = argparse.ArgumentParser(description="FineTuneX API 服务器")
    parser.add_argument("--host", type=str, default="0.0.0.0", help="主机地址")
    parser.add_argument("--port", type=int, default=8000, help="端口号")
    parser.add_argument("--reload", action="store_true", help="自动重载")
    
    args = parser.parse_args()
    
    print(f"启动 FineTuneX API 服务器...")
    print(f"主机：{args.host}:{args.port}")
    print(f"文档：http://{args.host}:{args.port}/docs")
    
    run_server(host=args.host, port=args.port, reload=args.reload)


if __name__ == "__main__":
    main()