[project]
name = "cost-prediction"
version = "0.1.0"
description = "装备成本预测系统"
requires-python = ">=3.9,<3.12"
readme = "README.md"
license = {file = "LICENSE"}

dependencies = [
    # Web框架
    "flask>=3.1.0",
    "flask-cors>=5.0.0",
    
    # 数据库
    "sqlalchemy>=2.0.36",
    "pymysql>=1.1.1",
    "cryptography>=43.0.0",
    "mysql-connector-python>=8.0.0",
    
    # 数据处理
    "numpy>=1.26.0,<2.0.0",
    "pandas>=2.2.0",
    
    # 机器学习
    "scikit-learn>=1.5.2",
    "torch==2.5.1",
    "torchvision==0.20.1",
    "torchaudio==2.5.1",
    "xgboost>=2.1.0",      # 添加 XGBoost
    "lightgbm>=4.5.0",     # 添加 LightGBM
    
    # 工具
    "openpyxl>=3.1.5",  # Excel支持
    "python-dotenv>=1.0.0",  # 环境变量
    "requests>=2.31.0",  # API测试
]

[project.optional-dependencies]
dev = [
    # 测试工具
    "pytest>=7.0",
    "black>=22.0",     # 代码格式化
    "mypy>=1.0",       # 类型检查
]

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]

[tool.black]
line-length = 88
target-version = ["py39", "py310", "py311"]

[tool.mypy]
python_version = "3.11"
warn_return_any = true
warn_unused_configs = true