web_server.py

#!/usr/bin/env python3
"""
Web服务器启动脚本
"""
import sys
import os
import argparse
from pathlib import Path

# 添加src目录到Python路径
sys.path.insert(0, str(Path(__file__).parent / "src"))

# 注册检测器
from core.openai_detector import OpenAICapabilityDetector
from core.anthropic_detector import AnthropicCapabilityDetector
from core.gemini_detector import GeminiCapabilityDetector
from core.capability_detector import CapabilityDetectorFactory

CapabilityDetectorFactory.register("openai", OpenAICapabilityDetector)
CapabilityDetectorFactory.register("anthropic", AnthropicCapabilityDetector)
CapabilityDetectorFactory.register("gemini", GeminiCapabilityDetector)

def main():
    # 导入环境配置
    from src.utils.env_config import env_config

    parser = argparse.ArgumentParser(description="AI API统一转换代理系统Web服务器")
    parser.add_argument("--host", default="0.0.0.0", help="服务器主机地址 (默认: 0.0.0.0)")
    parser.add_argument("--port", type=int, default=env_config.web_port, help=f"服务器端口 (默认: {env_config.web_port})")
    parser.add_argument("--reload", action="store_true", help="开启自动重载 (开发模式)")
    parser.add_argument("--debug", action="store_true", help="开启调试模式")

    args = parser.parse_args()

    # 验证配置
    config_errors = env_config.validate_config()
    if config_errors:
        print("❌ 配置验证失败:")
        for error in config_errors:
            print(f"   - {error}")
        sys.exit(1)
    
    print("🚀 启动AI API统一转换代理系统...")
    print(f"📖 访问 http://localhost:{args.port} 查看Web界面")
    print(f"📚 API文档: http://localhost:{args.port}/docs")
    print(f"🌐 服务器地址: {args.host}:{args.port}")
    
    if args.reload:
        print("⚠️  开发模式：自动重载已启用")
    
    import uvicorn
    
    # 设置日志级别
    log_level = "debug" if args.debug else "info"
    
    # 启动服务器
    uvicorn.run(
        "api.web_api:app",  # 使用import string格式
        host=args.host,
        port=args.port,
        reload=args.reload,
        log_level=log_level
    )

if __name__ == "__main__":
    main()