[Bug] Rerank 提供商不支持代理配置

[GuJi08233]

问题描述

所有 5 个 Rerank 提供商（vLLM、Xinference、百炼、NVIDIA、TEI）均不支持 HTTP 代理配置。配置了代理后，Rerank 请求完全绕过代理，导致超时失败或仅能直连访问。

对比之下，Chat/LLM、Embedding、TTS、STT 提供商都通过 proxy 配置字段和 create_proxy_client() 工具函数正确支持了代理。

---

复现步骤

1. 在 AstrBot 全局设置中配置 HTTP 代理（http_proxy）
2. 添加一个 Rerank 提供商（如百炼 Rerank），指向外部 API 端点
3. 启用知识库并使用重排序功能
4. Rerank 请求超时或连接失败
5. 代理软件无任何请求记录（流量绕过了代理）

---

预期结果

Rerank 提供商应和 Chat/Embedding/TTS/STT 提供商一样，支持代理配置。

---

根因分析

4 个基于 aiohttp 的 Rerank 提供商（vllm_rerank、bailian_rerank、nvidia_rerank、tei_rerank）在创建 aiohttp.ClientSession 时未传入任何代理参数：

# 当前代码（以 vllm_rerank_source.py 为例）
self.client = aiohttp.ClientSession(
    headers=h,
    timeout=aiohttp.ClientTimeout(total=self.timeout),
)
# 无 proxy 参数，无 trust_env=True

问题链：

• default.py 中 Rerank 提供商配置模板缺少 proxy 字段
• aiohttp.ClientSession 默认 trust_env=False，忽略 core_lifecycle.py 设置的 http_proxy/https_proxy 环境变量
• 请求调用中也未传递 proxy 参数

---

环境信息

• AstrBot 版本：latest (master)
• 部署方式：任意
• 受影响提供商：vLLM Rerank、Xinference Rerank、百炼 Rerank、NVIDIA Rerank、TEI Rerank

[dosubot]

Hi @GuJi08233! I'm Dosu and I'm helping the AstrBotTeam team.

你的根因分析是准确的，我通过代码验证确认了这一点。

问题确认

所有 4 个基于 aiohttp 的 Rerank 提供商（vLLM、百炼、NVIDIA、TEI）在创建 aiohttp.ClientSession 时均只传入了 headers 和 timeout，没有任何代理参数 [1] [2] [3] [4]。同时 default.py 中 Rerank 提供商的配置模板也缺少 proxy 字段 [5]，而紧邻其上方的 Embedding 提供商模板则包含了 proxy 字段 [6]。

对比已有的代理支持模式

Chat/LLM 提供商通过 create_proxy_client() 工具函数创建带代理的 httpx.AsyncClient，并通过 OpenAI SDK 的 http_client= 参数注入 [7]。Embedding 提供商也是类似模式，读取 provider_config["proxy"] 并创建 httpx.AsyncClient(proxy=proxy) [8]。

虽然 core_lifecycle.py 会在初始化时设置 os.environ["http_proxy"] 和 os.environ["https_proxy"] [9]，但 aiohttp.ClientSession 默认 trust_env=False，不会读取这些环境变量。

修复建议

有两种方案：

1. 最小改动方案：在 Rerank 提供商的 aiohttp.ClientSession 中添加 trust_env=True，使其自动使用 core_lifecycle.py 设置的环境变量代理。
2. 与其他提供商对齐方案：
   • 在 default.py 的 Rerank 配置模板中添加 proxy 字段
   • 在各 Rerank 源文件中读取 provider_config.get("proxy")
   • 将 proxy 参数显式传入 aiohttp.ClientSession(proxy=proxy)

方案 2 更符合现有 Chat/Embedding 提供商的做法，支持按提供商单独配置代理。注意 Xinference Rerank 使用的是 xinference_client SDK 而非 aiohttp [5]，可能需要单独处理其代理传递方式。

To reply, just mention @dosu.

---

Share context across your team and agents. Try Dosu.