# DualModel — 双模型协作推理框架依赖
# 安装方式: pip install -r requirements.txt

# ── 核心依赖 ─────────────────────────────────────────────────────
# HuggingFace Transformers — 本地模型加载与推理
transformers>=4.36.0

# PyTorch — 深度学习框架（根据 CUDA 版本选择对应的 torch 版本）
torch>=2.1.0

# OpenAI Python SDK — 云端 API 调用（兼容 DeepSeek / OpenAI / Qwen 等）
openai>=1.6.0

# ── 训练相关 ──────────────────────────────────────────────────
# 用于 Gate 的语义编码（训练和推理）
sentence-transformers>=2.2.0

# 科学计算
numpy>=1.24.0
scikit-learn>=1.3.0

# ── API Server ────────────────────────────────────────────────
# FastAPI Web 框架
fastapi>=0.104.0

# ASGI 服务器
uvicorn>=0.24.0

# Pydantic（FastAPI 依赖，明确列出版本）
pydantic>=2.5.0

# CORS 中间件（FastAPI 已自带，列出以确保）
# python-multipart（如需支持表单数据）

# ── 模型下载 ─────────────────────────────────────────────────────
# ModelScope — 国内模型下载（速度快）
modelscope>=1.0.0

# ── 系统托盘（可选，用于 desktop 托盘控制）─────────────────────
# macOS 菜单栏
rumps>=0.4.0; sys_platform == "darwin"
# Windows 系统托盘
pystray>=0.19.0; sys_platform == "win32"
Pillow>=9.0.0
