business-assistant/pyproject.toml

[project]
name = "business-assistant"
version = "1.0.0"
requires-python = ">=3.14"
dependencies = [
    "pandas>=3,<4",
    "openpyxl>=3.1,<4",
    "xlrd>=2,<3",
    "python-docx>=1.2,<2",
    "pdfplumber>=0.11,<1",
    "pillow>=12,<13",
    "markdown-it-py>=4,<5",
    "starlette>=0.52,<2",
    "uvicorn>=0.35,<1",
    "langgraph>=1,<2",
    "pymilvus>=2.6,<3",
    "redis>=6,<9",
    "tokenizers>=0.23.2",
    "torch>=2.10",
    "transformers>=4.51,<6",
]

[dependency-groups]
dev = ["httpx>=0.28,<1"]

[[tool.uv.index]]
name = "tuna"
url = "https://pypi.tuna.tsinghua.edu.cn/simple"
default = true

# Windows torch uses the official CUDA wheel; other dependencies keep the default mirror.
[[tool.uv.index]]
name = "pytorch-cu130"
url = "https://download.pytorch.org/whl/cu130"
explicit = true

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[tool.uv.sources]
torch = [
    { index = "pytorch-cu130", marker = "sys_platform == 'win32'" },
    { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'aarch64'" },
]