presenton/servers/fastapi/server.py

21 lines
584 B
Python

import uvicorn
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Run the FastAPI server")
parser.add_argument(
"--port", type=int, required=True, help="Port number to run the server on"
)
parser.add_argument(
"--reload", type=str, default="false", help="Reload the server on code changes"
)
args = parser.parse_args()
reload = args.reload == "true"
uvicorn.run(
"api.main:app",
host="127.0.0.1",
port=args.port,
log_level="info",
reload=reload,
)