# ===== 查询分割 baseline 依赖（进程内函数版）=====
# ①②③ 直接 import 调用，无需启动任何本地服务。

# 通用
requests

# ① 监督式（本地 Llama+LoRA，transformers+peft 进程内加载）
torch                  # 按你的 CUDA 版本安装对应 wheel
transformers>=4.40
peft                   # LoRA 适配器加载
accelerate             # device_map="auto" 需要

# ② 无监督（本地 XLM seq2seq），需 GPU；服务代码依赖：
#   （XLM 相关模块已内置在 methods/unsupervised/XLM/src，无需 pip）

# ③ ICL（Socratic），默认调外部 gpt-4o-mini：
openai>=1.0            # 也用于 ④ deepseek / 可选的本地 vLLM OpenAI 接口
flask                  # 仅当你想以“服务版”运行（start/*.sh）时才需要

# ④ SearChain（服务版，进阶）：flask + openai + ColBERT 检索栈（colbert-ai、faiss 等）+ 索引
python-dotenv          # ColBERT 服务 load_dotenv