53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
from gevent import monkey
|
|
|
|
monkey.patch_all()
|
|
|
|
import urllib3
|
|
|
|
from src.application import Application
|
|
|
|
# 禁用 HTTPS 证书告警
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="LLM Proxy Service")
|
|
parser.add_argument(
|
|
"--config", type=str, default=None, help="Configuration file path"
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def app() -> Application:
|
|
args = parse_args()
|
|
config_path = args.config
|
|
project_root = Path(__file__).resolve().parent
|
|
if config_path is None:
|
|
resolved_config_path = project_root / "config.yaml"
|
|
else:
|
|
config_candidate = Path(config_path)
|
|
resolved_config_path = (
|
|
config_candidate
|
|
if config_candidate.is_absolute()
|
|
else (Path.cwd() / config_candidate).resolve()
|
|
)
|
|
|
|
if not resolved_config_path.exists():
|
|
raise FileNotFoundError(f"Configuration file not found: {resolved_config_path}")
|
|
|
|
return Application(resolved_config_path)
|
|
|
|
|
|
def main() -> None:
|
|
app().run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|