w-dispatch-ai 1.0.4 → 1.0.6

package/test/unit-callAiWithFallback.test.mjs

@@ -135,6 +135,21 @@ describe('callAiWithFallback', function() {
         assert.strict.deepEqual(r, rr)
     })
 
+    it('minAttemptMs等fallback層設定可經工作流層原樣轉傳(曾因白名單漏鍵而失效)', async function() {
+        //budgetMs=60000(高於預設門檻20000)+minAttemptMs=600000(遠大於budget):
+        //minAttemptMs有被轉送才會立即budget exhausted; 漏鍵時會照樣執行並成功
+        let t = await callAiWithFallback('abc', {
+            providers,
+            spec: { use: 'p-claude' },
+            promptPrefix: '',
+            budgetMs: 60000,
+            minAttemptMs: 600000,
+        })
+        let r = [t.ok, t.error, t.tried.map((x) => x.outcome)]
+        let rr = [false, 'budget exhausted', ['budget-out']]
+        assert.strict.deepEqual(r, rr)
+    })
+
     it('多金鑰條目經由keys輪替且keyId帶索引', async function() {
         let t = await callAiWithFallback('abc', { providers, spec: { use: 'p-oc' }, promptPrefix: '' })
         let auth = JSON.parse(t.json.env.OPENCODE_AUTH_CONTENT)

package/test/unit-dispatchAi.test.mjs

@@ -70,7 +70,7 @@ describe('dispatchAi', function() {
             [true, ['run', '--agent', 'build', '-m', 'mdl']],
             [true, ['-p', '--dangerously-skip-permissions', '--model', 'mdl']],
             [true, ['exec', '--sandbox', 'workspace-write', '--skip-git-repo-check', '-m', 'mdl']],
-            [true, ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'mdl', '--print', 'abc']],
+            [true, ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'mdl', '--add-dir', process.cwd(), '--print', 'abc']],
         ]
         assert.strict.deepEqual(r, rr)
     })

package/test/unit-dispatchAiFallback.test.mjs

@@ -316,6 +316,31 @@ describe('dispatchAiFallback', function() {
         assert.strict.deepEqual(r, rr)
     })
 
+    it('失敗事件與tried帶被拒回覆(stdout)與錯誤輸出(stderr)供診斷', async function() {
+        let evs = []
+        let t = await dispatchAiFallback('abc', {
+            providers: [
+                //驗證失敗(skip-group): 被拒的回覆內容應可於事件與tried取得
+                { id: 'g-diag-v', kind: 'claude', exe: fake.exe, validate: 'min:100000' },
+                //exit 1(next-key): stderr之錯誤訊息應可於事件與tried取得
+                { id: 'g-diag-e', kind: 'claude', exe: fake.exe, extraArgs: ['--fake-exit=1', '--fake-stderr=Invalid API key.'] },
+                { id: 'g-diag-ok', kind: 'claude', exe: fake.exe },
+            ],
+            onEvent: (ev) => evs.push(ev),
+        })
+        let evV = evs.find((x) => x.type === 'skip-group')
+        let evE = evs.find((x) => x.type === 'next-key')
+        let r = [
+            t.ok,
+            evV.stdout.includes('"stdin"'), //被拒回覆即假CLI之echo JSON
+            t.tried[0].stdout.includes('"stdin"'),
+            evE.stderr,
+            t.tried[1].stderr,
+        ]
+        let rr = [true, true, true, 'Invalid API key.', 'Invalid API key.']
+        assert.strict.deepEqual(r, rr)
+    })
+
     it('onEvent依序回報事件, 回調拋出例外不影響主流程', async function() {
         let evs = []
         let t = await dispatchAiFallback('abc', {

package/test/unit-dispatchAiWkf.test.mjs

@@ -86,6 +86,14 @@ describe('dispatchAiWkf', function() {
         assert.strict.deepEqual(r, rr)
     })
 
+    it('defaults之minAttemptMs經callAi生效(README第一層教學之回歸測試)', async function() {
+        let wkf = dispatchAiWkf({ providers, defaults: { promptPrefix: '', budgetMs: 60000, minAttemptMs: 600000 } })
+        let t = await wkf.callAi('abc', { spec: { use: 'p-a' } })
+        let r = [t.ok, t.error]
+        let rr = [false, 'budget exhausted']
+        assert.strict.deepEqual(r, rr)
+    })
+
     it('runFanout綁定providers與defaults(defaults併入callOpt)', async function() {
         let wkf = dispatchAiWkf({ providers, defaults: { promptPrefix: '' } })
         let t = await wkf.runFanout({

package/test/unit-dispatchAntigravity.test.mjs

@@ -62,7 +62,7 @@ describe('dispatchAntigravity', function() {
         let t = await dispatchAntigravity('abc', { exe: fake.exe })
         let o = JSON.parse(t.stdout)
         let r = [t.ok, o.args]
-        let rr = [true, ['--dangerously-skip-permissions', '--print-timeout', '270s', '--print', 'abc']]
+        let rr = [true, ['--dangerously-skip-permissions', '--print-timeout', '270s', '--add-dir', process.cwd(), '--print', 'abc']]
         assert.strict.deepEqual(r, rr)
     })
 
@@ -78,7 +78,7 @@ describe('dispatchAntigravity', function() {
         let t = await dispatchAntigravity('abc', { exe: fake.exe, model: 'gemini-3.6-flash-low' })
         let o = JSON.parse(t.stdout)
         let r = o.args
-        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'gemini-3.6-flash-low', '--print', 'abc']
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'gemini-3.6-flash-low', '--add-dir', process.cwd(), '--print', 'abc']
         assert.strict.deepEqual(r, rr)
     })
 
@@ -86,7 +86,7 @@ describe('dispatchAntigravity', function() {
         let t = await dispatchAntigravity('abc', { exe: fake.exe, model: 'gemini-3.1-pro', effort: 'low' })
         let o = JSON.parse(t.stdout)
         let r = o.args
-        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'gemini-3.1-pro', '--effort', 'low', '--print', 'abc']
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--model', 'gemini-3.1-pro', '--effort', 'low', '--add-dir', process.cwd(), '--print', 'abc']
         assert.strict.deepEqual(r, rr)
     })
 
@@ -94,7 +94,7 @@ describe('dispatchAntigravity', function() {
         let t = await dispatchAntigravity('abc', { exe: fake.exe, skipPermissions: false })
         let o = JSON.parse(t.stdout)
         let r = [t.ok, o.args]
-        let rr = [true, ['--print-timeout', '270s', '--print', 'abc']]
+        let rr = [true, ['--print-timeout', '270s', '--add-dir', process.cwd(), '--print', 'abc']]
         assert.strict.deepEqual(r, rr)
     })
 
@@ -123,6 +123,26 @@ describe('dispatchAntigravity', function() {
         assert.strict.deepEqual(r, rr)
     })
 
+    it('未給addDirs時自動納入有效cwd(明給的cwd優先)', async function() {
+        let t1 = await dispatchAntigravity('abc', { exe: fake.exe }) //cwd未給 → process.cwd()
+        let t2 = await dispatchAntigravity('abc', { exe: fake.exe, cwd: fake.fd }) //cwd明給
+        let o1 = JSON.parse(t1.stdout)
+        let o2 = JSON.parse(t2.stdout)
+        let i1 = o1.args.indexOf('--add-dir')
+        let i2 = o2.args.indexOf('--add-dir')
+        let r = [o1.args[i1 + 1], o2.args[i2 + 1]]
+        let rr = [process.cwd(), fake.fd]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('明示addDirs為空陣列時不自動加入(代表不揭露任何目錄)', async function() {
+        let t = await dispatchAntigravity('abc', { exe: fake.exe, addDirs: [] })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.args.includes('--add-dir')]
+        let rr = [true, false]
+        assert.strict.deepEqual(r, rr)
+    })
+
     it('addDirs逐項展開為--add-dir並濾除非有效字串', async function() {
         let t = await dispatchAntigravity('abc', { exe: fake.exe, addDirs: ['c:/a', null, 'c:/b', 123] })
         let o = JSON.parse(t.stdout)
@@ -135,7 +155,7 @@ describe('dispatchAntigravity', function() {
         let t = await dispatchAntigravity('abc', { exe: fake.exe, extraArgs: ['--output-format', 'json'] })
         let o = JSON.parse(t.stdout)
         let r = o.args
-        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--output-format', 'json', '--print', 'abc']
+        let rr = ['--dangerously-skip-permissions', '--print-timeout', '270s', '--add-dir', process.cwd(), '--output-format', 'json', '--print', 'abc']
         assert.strict.deepEqual(r, rr)
     })
 

package/test/unit-providers.test.mjs

@@ -0,0 +1,66 @@
+import assert from 'assert'
+import keys from 'lodash-es/keys.js'
+import providers from '../src/providers.mjs'
+import adapters from '../src/adapters.mjs'
+
+
+describe('providers', function() {
+
+    it('預設匯出為非空條目陣列', function() {
+        let r = [Array.isArray(providers), providers.length > 0]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('各條目皆有非空字串之id、model、kind', function() {
+        let r = providers.map((p) => [typeof p.id === 'string' && p.id !== '', typeof p.model === 'string' && p.model !== '', typeof p.kind === 'string' && p.kind !== ''])
+        let rr = providers.map(() => [true, true, true])
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('各條目之kind皆為有效供應商種類', function() {
+        let ks = keys(adapters)
+        let r = providers.map((p) => [p.id, ks.includes(p.kind)])
+        let rr = providers.map((p) => [p.id, true])
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('id不重複且皆符合「路徑:模型」命名(含冒號)', function() {
+        let ids = providers.map((p) => p.id)
+        let r = [new Set(ids).size === ids.length, ids.every((id) => id.includes(':'))]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('api-openai-compat條目皆有baseURL', function() {
+        let apis = providers.filter((p) => p.kind === 'api-openai-compat')
+        let r = [apis.length > 0, apis.every((p) => typeof p.baseURL === 'string' && p.baseURL.startsWith('http'))]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('opencode條目皆有provider(金鑰注入所需之配對)', function() {
+        let ocs = providers.filter((p) => p.kind === 'opencode')
+        let r = [ocs.length > 0, ocs.every((p) => typeof p.provider === 'string' && p.provider !== '')]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('envVar命名皆為大寫底線格式, 訂閱登入態條目(claude/codex)無envVar', function() {
+        let withVar = providers.filter((p) => p.envVar !== undefined)
+        let subs = providers.filter((p) => ['claude', 'codex'].includes(p.kind))
+        let r = [
+            withVar.every((p) => /^[A-Z][A-Z0-9_]*$/.test(p.envVar)),
+            subs.every((p) => p.envVar === undefined),
+        ]
+        let rr = [true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('條目不自帶timeoutMs(由上層依任務型態統一給定)', function() {
+        let r = providers.every((p) => p.timeoutMs === undefined)
+        let rr = true
+        assert.strict.deepEqual(r, rr)
+    })
+
+})

package/test/unit-resolveProviders.test.mjs

@@ -0,0 +1,85 @@
+import assert from 'assert'
+import resolveProviders from '../src/resolveProviders.mjs'
+import providersAll from '../src/providers.mjs'
+
+
+describe('resolveProviders', function() {
+
+    //測試用定義, 不依賴真實環境變數
+    let defs = [
+        { id: 'api:m1', kind: 'api-openai-compat', baseURL: 'https://x/v1', model: 'm1', envVar: 'T_K1' },
+        { id: 'api:m2', kind: 'api-openai-compat', baseURL: 'https://x/v1', model: 'm2', envVar: 'T_K2' },
+        { id: 'claude:sonnet', kind: 'claude', model: 'sonnet' },
+        { id: 'api:m3', kind: 'api-openai-compat', baseURL: 'https://x/v1', model: 'm3', envVar: 'T_K1', keys: ['sk-own'] },
+    ]
+    let env = { T_K1: 'sk-a, sk-b,', T_K2: '' }
+
+    it('envVar展開為keys(逗號分隔且修剪空白), envVar欄位自輸出移除', function() {
+        let { providers } = resolveProviders(defs, { env })
+        let p = providers.find((x) => x.id === 'api:m1')
+        let r = [p.keys, p.envVar]
+        let rr = [['sk-a', 'sk-b'], undefined]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('缺環境變數(或值為空)之條目停用並列入skipped', function() {
+        let { providers, skipped } = resolveProviders(defs, { env })
+        let r = [providers.map((x) => x.id), skipped]
+        let rr = [['api:m1', 'claude:sonnet', 'api:m3'], [{ id: 'api:m2', envVar: 'T_K2' }]]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('無envVar之條目(訂閱登入態)原樣通過, 自帶keys者以keys為準', function() {
+        let { providers } = resolveProviders(defs, { env })
+        let pc = providers.find((x) => x.id === 'claude:sonnet')
+        let p3 = providers.find((x) => x.id === 'api:m3')
+        let r = [pc.model, p3.keys, p3.envVar]
+        let rr = ['sonnet', ['sk-own'], undefined]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('pick自選子集且依pick順序回傳, 查無之id列入missing', function() {
+        let { providers, missing } = resolveProviders(defs, { env, pick: ['claude:sonnet', 'api:m1', 'api:nope'] })
+        let r = [providers.map((x) => x.id), missing]
+        let rr = [['claude:sonnet', 'api:m1'], ['api:nope']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('table以id為鍵, 可直接作dispatchAiWkf之定義表', function() {
+        let { table } = resolveProviders(defs, { env })
+        let r = [Object.keys(table), table['api:m1'].keys]
+        let rr = [['api:m1', 'claude:sonnet', 'api:m3'], ['sk-a', 'sk-b']]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('輸入陣列與條目皆不被改動', function() {
+        let before = JSON.stringify(defs)
+        resolveProviders(defs, { env })
+        let r = JSON.stringify(defs)
+        assert.strict.deepEqual(r, before)
+    })
+
+    it('providers無效時回傳空結果不throw', function() {
+        let r = []
+        for (let v of [null, undefined, 'abc', {}, [null, 'x']]) {
+            let t = resolveProviders(v, { env })
+            r.push([t.providers.length, t.skipped.length, t.missing.length])
+        }
+        let rr = [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('可展開套件內建providers.mjs(以模擬env驗證, 不依賴真實.env)', function() {
+        let fakeEnv = { OPENCODE_KEYS: 'sk-1,sk-2', AGNES_KEYS: 'sk-3', POOLSIDE_KEYS: 'sk-4' }
+        let { providers, skipped } = resolveProviders(providersAll, { env: fakeEnv })
+        let r = [
+            providers.length === providersAll.length, //三個envVar皆有值 → 無停用
+            skipped.length,
+            providers.every((p) => p.envVar === undefined),
+            providers.filter((p) => Array.isArray(p.keys)).length > 0,
+        ]
+        let rr = [true, 0, true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+})