dsh-router-core 0.4.4 → 0.4.5

package/src/router/supplier-switch.test.ts

@@ -218,3 +218,94 @@ test('指定不存在的连接：明确报「连接不存在」，而不是静
   assert.equal(a.calls, 0, '一个号都不该被试')
   assert.match(r.error ?? '', /不存在/)
 })
+
+test('指定连接失败：报上游真实原因，不许说成「连接不存在」', async () => {
+  // 2026-10-14 实测：0 积分号（uid 在池里、面板上看得见）测出来是
+  // 「连接 "4135677434275728" 不存在（可能已被删除）」——真话是额度用完。
+  // 机制：失败后 `continue`，下一轮 onlyUid 被 `tried` 挡掉 → uid undefined，
+  // 旧代码在那里无条件覆盖 lastError，把上游原因冲掉。假诊断比没有诊断更坏：
+  // 用户会去删链接、重新登录，而正解是充值。
+  const { router } = rig()
+  const a = spy('supA', 'a-model')
+  a.s.accounts = () => [
+    { uid: 'broke', credits: 0, state: 'ok' },
+    { uid: 'rich', credits: 100, state: 'ok' },
+  ]
+  a.s.status = () => ({ id: 'supA', name: 'supA', accounts: a.s.accounts() })
+  const seen: string[] = []
+  a.s.chatOnce = async (uid) => {
+    seen.push(uid)
+    return { ok: false, state: 'quota', message: 'Your requests have exceeded the quota.' }
+  }
+  add(router, a.s)
+
+  const r = await router.testModel('supA', 'a-model', 'broke')
+  assert.equal(r.ok, false)
+  assert.deepEqual(seen, ['broke'], '钉死单连接，失败也不换号')
+  assert.match(r.error ?? '', /quota|exceeded/, `要报上游真因，实际: ${r.error}`)
+  assert.doesNotMatch(r.error ?? '', /不存在/, 'uid 在池里，绝不是「不存在」')
+})
+
+// ---- 积分用完的号必须退出路由（插件报 quota → 核心不选它） ----
+// 端到端串起来：traework 的 pool.list() 在 credits===0 时报 quota，核心 pick()
+// 据此跳过该号。这条走真实入口 chatCompletions，钉住「报出的事实真的改变了路由」，
+// 而不只是「状态字符串好看」。
+
+test('插件报 quota 的号：请求直接落到还有额度的号，不去撞死号', async () => {
+  const { router } = rig()
+  const a = spy('supA', 'a-model')
+  const tried: string[] = []
+  a.s.accounts = () => [
+    { uid: 'broke', credits: 0, state: 'quota' },
+    { uid: 'rich', credits: 500, state: 'ok' },
+  ]
+  a.s.status = () => ({ id: 'supA', name: 'supA', accounts: a.s.accounts() })
+  a.s.chatOnce = async (uid) => {
+    tried.push(uid)
+    return { ok: true, status: 200, body: '{"ok":1}' }
+  }
+  add(router, a.s)
+
+  await router.chatCompletions(reqWith('supA/a-model'), fakeRes())
+
+  assert.deepEqual(tried, ['rich'], '一次都不该撞到积分用完的号')
+})
+
+test('积分用完的号恢复（充值后报 ok）：立刻重新参与路由，无需重启', async () => {
+  const { router } = rig()
+  const a = spy('supA', 'a-model')
+  let broke = true
+  const tried: string[] = []
+  a.s.accounts = () => (broke
+    ? [{ uid: 'was-broke', credits: 0, state: 'quota' }, { uid: 'other', credits: 10, state: 'ok' }]
+    : [{ uid: 'was-broke', credits: 900, state: 'ok' }])
+  a.s.status = () => ({ id: 'supA', name: 'supA', accounts: a.s.accounts() })
+  a.s.chatOnce = async (uid) => {
+    tried.push(uid)
+    return { ok: true, status: 200, body: '{"ok":1}' }
+  }
+  add(router, a.s)
+
+  await router.chatCompletions(reqWith('supA/a-model'), fakeRes())
+  assert.deepEqual(tried, ['other'], '没额度时不选它')
+
+  broke = false
+  await router.chatCompletions(reqWith('supA/a-model'), fakeRes())
+  assert.equal(tried[1], 'was-broke', '充值后立刻回到候选')
+})
+
+test('全池都没额度：如实失败，不假装成功也不无限转圈', async () => {
+  const { router } = rig()
+  const a = spy('supA', 'a-model')
+  a.s.accounts = () => [
+    { uid: 'b1', credits: 0, state: 'quota' },
+    { uid: 'b2', credits: 0, state: 'quota' },
+  ]
+  a.s.status = () => ({ id: 'supA', name: 'supA', accounts: a.s.accounts() })
+  a.s.chatOnce = async () => ({ ok: true, status: 200, body: '{"ok":1}' })
+  add(router, a.s)
+
+  const r = await router.testModel('supA', 'a-model')
+  assert.equal(r.ok, false, '没号可选就是失败')
+  assert.equal(a.calls, 0, '不该去撞任何一个已知没额度的号')
+})

package/src/router/types.ts

@@ -3,7 +3,7 @@
  * 每个供应商提供 OpenAI 兼容的 chat/models，并暴露面板状态。
  * 仿 9router：供应商 = 一种"免费/付费"通道；dsh-router 在其上做路由。
  */
-import type { ChatOnceResult, SupplierAccountNow } from '../suppliers/contract.ts'
+import type { AccountState, ChatOnceResult, SupplierAccountNow } from '../suppliers/contract.ts'
 import type { AccountPool } from './account-pool.ts'
 
 /** 面板展示的账号状态（脱敏）。 */
@@ -11,6 +11,14 @@ export interface SupplierAccount {
   uid: string
   nickname?: string
   credits: number
+  /**
+   * 插件解读上游信号后的语义状态（`quota` = 额度/积分用完，`session_dead` = 凭证失效）。
+   *
+   * `decorate()` 一直原样带出插件报的状态，核心选号也读它（`quota`/`session_dead`
+   * 的号不进候选）——面板类型里缺了这一格就会自相矛盾：号明明已被排除在路由之外，
+   * 界面上却只看得到「0 积分」，看不出它为什么不参与。
+   */
+  state?: AccountState
   cooling: boolean
   until?: string
   reason?: string

package/src/shared.ts

@@ -25,6 +25,12 @@ export interface RouterAccount {
   uid: string
   nickname?: string
   credits: number
+  /**
+   * 插件解读上游信号后的语义状态（`quota` = 额度/积分用完，`session_dead` = 凭证失效）。
+   * 核心选号会跳过这些号，所以面板必须显示出来——否则界面写着「健康」，而路由
+   * 永远不选它，用户无从知道为什么。
+   */
+  state?: string
   cooling: boolean
   until?: string
   reason?: string

package/src/suppliers/loader.test.ts

@@ -215,3 +215,56 @@ test('用户目录只加载 *.js，且同 id 后者覆盖前者（按文件名
   assert.equal(suppliers[0]?.supplier.name, 'Second')
   assert.equal(suppliers[0]?.source, 'user')
 })
+
+// ---------------------------------------------------------------------------
+// 插件报的「额度用完」要真的进路由（end-to-end 的那一段线）
+// ---------------------------------------------------------------------------
+
+/**
+ * 造一个报 quota 的插件（模拟 traework 的 pool.list()：credits===0 → state quota）。
+ * 这条钉住 loader 的 `accounts()` 是**原样透传**插件状态（不带 decorate/hydrate）：
+ * 核心 pick() 读的就是它。哪天有人给它加一层加工，把 quota 抹成 ok，路由又会
+ * 去撞没额度的号，而面板还显示正常——所以这里必须红。
+ */
+function quotaPlugin(id: string): SupplierModule {
+  const accounts: SupplierAccountNow[] = [
+    { uid: 'u-broke', credits: 0, state: 'quota' },
+    { uid: 'u-rich', credits: 500, state: 'ok' },
+  ]
+  return {
+    id,
+    name: id,
+    status: () => ({ id, name: id, accounts }),
+    listModels: () => [],
+    getAlias: () => id,
+    chatOnce: async (_uid: string, _req: ChatRequest): Promise<ChatOnceResult> => ({ ok: true, status: 200, body: '{}' }),
+    dispose: () => {},
+  } as unknown as SupplierModule
+}
+
+test('插件报的 quota 原样到达核心 → 没额度的号不进选号', () => {
+  const { env: e } = env()
+  const { supplier } = wrapModule(quotaPlugin('traework'), e, 'test')
+  assert.equal(
+    supplier.accounts().find((a) => a.uid === 'u-broke')?.state,
+    'quota',
+    'accounts() 必须原样带出插件状态（核心 pick 读的就是它）',
+  )
+  assert.equal(supplier.pool.pick(supplier.accounts(), [], 'm1'), 'u-rich', '额度用完的号被跳过')
+})
+
+test('额度用完的号在面板上也能看出来（decorate 不吞掉插件报的 state）', () => {
+  const { env: e } = env()
+  const { supplier } = wrapModule(quotaPlugin('traework'), e, 'test')
+  const shown = supplier.status().accounts.find((a) => a.uid === 'u-broke')
+  assert.equal(shown?.state, 'quota', '面板要能区分「没额度」与「正常」')
+  assert.equal(shown?.cooling, false, '还没发生过失败，就不该标成冷却')
+})
+
+test('额度用完 ≠ -1：water 后积分真值 0 照旧落进核心缓存（不冲值、不误判）', () => {
+  const { env: e, store } = env()
+  const { supplier } = wrapModule(quotaPlugin('traework'), e, 'test')
+  supplier.status() // 触发 hydrateCredits
+  assert.equal(store.get('traework').credits['u-broke'], 0, '0 是真值，要落盘')
+  assert.equal(store.get('traework').credits['u-rich'], 500)
+})