Coverage for kwai/kwai_api.py: 0%

1"""Module for starting the api server.""" 

2import argparse 

3 

4import uvicorn 

5 

6 

7def create_args(): 

8 """Parse and create cli arguments.""" 

9 parser = argparse.ArgumentParser(description="kwai backend") 

10 parser.add_argument( 

11 "--reload", 

12 action=argparse.BooleanOptionalAction, 

13 help="Watch for code changes or not", 

14 ) 

15 parser.add_argument( 

16 "--port", type=int, default=8000, help="The port of the api server." 

17 ) 

18 return parser.parse_args() 

19 

20 

21if __name__ == "__main__": 

22 args = create_args() 

23 uvicorn.run( 

24 "kwai.api.app:create_app", 

25 host="0.0.0.0", 

26 port=args.port, 

27 factory=True, 

28 reload=args.reload, 

29 )