Spaces:
Sleeping
Sleeping
| #!/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() | |