llmopt-server / run.py
Shrot101's picture
Initial commit of LLMOpt framework
3c1db6c
#!/usr/bin/env python3
"""
LLMOpt Server entry point.
Usage:
python run.py
python run.py --host 0.0.0.0 --port 8000
"""
import argparse
import os
from pathlib import Path
# Load .env if present
env_file = Path(__file__).parent / "config" / ".env"
if env_file.exists():
from dotenv import load_dotenv
load_dotenv(env_file)
import uvicorn
def main():
parser = argparse.ArgumentParser(description="Start the LLMOpt server")
parser.add_argument("--host", default=os.getenv("HOST", "0.0.0.0"))
parser.add_argument("--port", type=int, default=int(os.getenv("PORT", "8000")))
parser.add_argument("--reload", action="store_true", help="Enable hot-reload (dev mode)")
parser.add_argument("--log-level", default=os.getenv("LOG_LEVEL", "warning").lower())
args = parser.parse_args()
uvicorn.run(
"llmopt.api.app:app",
host=args.host,
port=args.port,
reload=args.reload,
log_level=args.log_level,
)
if __name__ == "__main__":
main()