w-dispatch-ai 1.0.2 → 1.0.3

package/test/unit-dispatchApiOpenaiCompat.test.mjs

@@ -0,0 +1,233 @@
+import assert from 'assert'
+import dispatchApiOpenaiCompat from '../src/dispatchApiOpenaiCompat.mjs'
+import dispatchAi from '../src/dispatchAi.mjs'
+import dispatchAiFallback from '../src/dispatchAiFallback.mjs'
+import dispatchAiWkf from '../src/dispatchAiWkf.mjs'
+import fakeServerForApiTest from './tools/fakeServerForApiTest.mjs'
+
+
+describe('dispatchApiOpenaiCompat', function() {
+
+    let svr = null
+
+    before(async function() {
+        svr = await fakeServerForApiTest()
+    })
+
+    after(async function() {
+        if (svr) {
+            await svr.close()
+        }
+    })
+
+    it('prompt非有效字串時回傳錯誤結果物件且不reject', async function() {
+        let r = []
+        for (let prompt of [null, undefined, '', 123, {}, []]) {
+            let t = await dispatchApiOpenaiCompat(prompt)
+            r.push([t.ok, t.error, t.attempts])
+        }
+        let rr = [
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+            [false, 'prompt must be a non-empty string', 0],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('baseURL與model必填, 缺一即回傳錯誤結果物件', async function() {
+        let t1 = await dispatchApiOpenaiCompat('abc', { model: 'echo' })
+        let t2 = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url })
+        let r = [[t1.ok, t1.error], [t2.ok, t2.error]]
+        let rr = [
+            [false, 'baseURL must be a non-empty string'],
+            [false, 'model must be a non-empty string'],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('成功時stdout為回覆內容, 請求含Bearer金鑰與model與user訊息', async function() {
+        let t = await dispatchApiOpenaiCompat('中文prompt\n第二行', { baseURL: svr.url, key: 'sk-good-1', model: 'echo' })
+        let o = JSON.parse(t.stdout)
+        let r = [
+            t.ok,
+            t.code,
+            t.error,
+            t.attempts,
+            o.auth,
+            o.body.model,
+            o.body.messages,
+        ]
+        let rr = [
+            true,
+            200,
+            '',
+            1,
+            'Bearer sk-good-1',
+            'echo',
+            [{ role: 'user', content: '中文prompt\n第二行' }],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('未給key時不帶Authorization標頭', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, model: 'echo' })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.auth]
+        let rr = [true, '']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('baseURL尾端斜線自動正規化', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url + '///', key: 'sk-good-1', model: 'echo' })
+        let r = [t.ok, t.code]
+        let rr = [true, 200]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('system置於messages首位, body額外鍵可併入', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', {
+            baseURL: svr.url,
+            key: 'sk-good-1',
+            model: 'echo',
+            system: '你是審計者',
+            body: { temperature: 0.2, max_tokens: 100 },
+        })
+        let o = JSON.parse(t.stdout)
+        let r = [
+            o.body.messages,
+            o.body.temperature,
+            o.body.max_tokens,
+        ]
+        let rr = [
+            [{ role: 'system', content: '你是審計者' }, { role: 'user', content: 'abc' }],
+            0.2,
+            100,
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('無效金鑰回401且不重試(4xx為客戶端錯誤)', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-bad-x', model: 'echo', maxRetries: 3, retryDelayMs: 100 })
+        let r = [t.ok, t.code, t.error, t.attempts, t.stderr.includes('Invalid API key')]
+        let rr = [false, 401, 'HTTP 401', 1, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('429依maxRetries重試後成功', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-flaky1', model: 'flaky-429', maxRetries: 1, retryDelayMs: 100 })
+        let r = [t.ok, t.code, t.attempts, JSON.parse(t.stdout).attempt]
+        let rr = [true, 200, 2, 2]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('429且maxRetries為0時直接失敗', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-flaky2', model: 'flaky-429' })
+        let r = [t.ok, t.code, t.error, t.attempts]
+        let rr = [false, 429, 'HTTP 429', 1]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('HTTP 500回傳失敗且stderr含回應本體', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-1', model: 'err-500' })
+        let r = [t.ok, t.code, t.error, t.stderr.includes('internal error')]
+        let rr = [false, 500, 'HTTP 500', true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('逾時時中止請求且error以TIMEOUT開頭', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-1', model: 'slow', timeoutMs: 1000 })
+        let r = [t.ok, t.code, t.error, t.durationMs < 9000]
+        let rr = [false, null, 'TIMEOUT after 1s', true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('網路層錯誤回傳FETCH_ERROR', async function() {
+        let t = await dispatchApiOpenaiCompat('abc', { baseURL: 'http://127.0.0.1:1/v1', key: 'sk-good-1', model: 'echo', timeoutMs: 3000 })
+        let r = [t.ok, t.code, t.error.indexOf('FETCH_ERROR') === 0]
+        let rr = [false, null, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('回應缺choices時回傳INVALID_RESPONSE', async function() {
+        let r = []
+        for (let model of ['no-choices', 'not-json']) {
+            let t = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-1', model })
+            r.push([t.ok, t.code, t.error.indexOf('INVALID_RESPONSE') === 0])
+        }
+        let rr = [[false, 200, true], [false, 200, true]]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('validate字串規則與自訂函數皆可用, 失敗回傳OUTPUT_VALIDATION_FAILED', async function() {
+        let t1 = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-1', model: 'empty-content', validate: 'nonempty' })
+        let t2 = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-1', model: 'echo', validate: 'nonempty,json' })
+        let t3 = await dispatchApiOpenaiCompat('abc', { baseURL: svr.url, key: 'sk-good-1', model: 'echo', validate: (s) => s.includes('絕不存在的字串') })
+        let r = [
+            [t1.ok, t1.code, t1.error],
+            [t2.ok, t2.error],
+            [t3.ok, t3.error],
+        ]
+        let rr = [
+            [false, 200, 'OUTPUT_VALIDATION_FAILED'],
+            [true, ''],
+            [false, 'OUTPUT_VALIDATION_FAILED'],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('可經dispatchAi以kind api-openai-compat分派', async function() {
+        let t = await dispatchAi('api-openai-compat', 'abc', { baseURL: svr.url, key: 'sk-good-1', model: 'echo' })
+        let o = JSON.parse(t.stdout)
+        let r = [t.ok, o.body.model]
+        let rr = [true, 'echo']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('可作為dispatchAiFallback條目, 多金鑰輪替(401換下一把)', async function() {
+        let t = await dispatchAiFallback('abc', {
+            providers: [{
+                id: 'g-api',
+                kind: 'api-openai-compat',
+                baseURL: svr.url,
+                model: 'echo',
+                keys: ['sk-bad-r0', 'sk-good-r1'],
+            }],
+        })
+        let o = JSON.parse(t.stdout)
+        let r = [
+            t.ok,
+            t.providerId,
+            t.keyIndex,
+            o.auth,
+            t.tried.map((x) => [x.keyId, x.outcome, x.error || null]),
+        ]
+        let rr = [
+            true,
+            'g-api',
+            1,
+            'Bearer sk-good-r1',
+            [['g-api#0', 'next-key', 'HTTP 401'], ['g-api#1', 'ok', null]],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('可作為工作流provider, 回覆JSON經解析與檢核', async function() {
+        let wkf = dispatchAiWkf({
+            providers: {
+                'api-echo': { kind: 'api-openai-compat', baseURL: svr.url, model: 'echo', keys: ['sk-good-w0'] },
+            },
+            defaults: { promptPrefix: '' },
+        })
+        let t = await wkf.callAi('abc', {
+            spec: { use: 'api-echo' },
+            check: (j) => j.body.model === 'echo',
+        })
+        let r = [t.ok, t.providerId, t.keyId, t.json.auth, t.json.body.messages[0].content]
+        let rr = [true, 'api-echo', 'api-echo#0', 'Bearer sk-good-w0', 'abc']
+        assert.strict.deepEqual(r, rr)
+    })
+
+})

package/test/unit-extractJsonLoose.test.mjs

@@ -0,0 +1,78 @@
+import assert from 'assert'
+import extractJsonLoose from '../src/wkf/extractJsonLoose.mjs'
+
+
+describe('extractJsonLoose', function() {
+
+    it('整段即為合法JSON時直接解析', function() {
+        let r = [
+            extractJsonLoose('{"a":1}'),
+            extractJsonLoose('[1,2,3]'),
+            extractJsonLoose('  {"a":{"b":[1]}}  '),
+        ]
+        let rr = [{ a: 1 }, [1, 2, 3], { a: { b: [1] } }]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('去除code fence與前後說明文字後抽取第一個完整物件', function() {
+        let r = [
+            extractJsonLoose('說明文字\n```json\n{"a":1}\n```\n後記'),
+            extractJsonLoose('```\n{"a":1}\n```'),
+            extractJsonLoose('前綴 {"a":1} 後綴 {"b":2}'),
+        ]
+        let rr = [{ a: 1 }, { a: 1 }, { a: 1 }]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('去除ANSI色碼', function() {
+        let esc = String.fromCharCode(27)
+        let r = extractJsonLoose(`${esc}[0m{"a":1}${esc}[91m`)
+        let rr = { a: 1 }
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('字串值內之括號與跳脫不干擾配對', function() {
+        let r = extractJsonLoose('{"a":"x{y}[z]","b":"q\\"w"}')
+        let rr = { a: 'x{y}[z]', b: 'q"w' }
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('截斷(括號未閉合)回傳null', function() {
+        let r = [
+            extractJsonLoose('{"a":1'),
+            extractJsonLoose('[1,2'),
+            extractJsonLoose('{"a":"未閉合字串'),
+        ]
+        let rr = [null, null, null]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('非物件與陣列之JSON回傳null', function() {
+        let r = [
+            extractJsonLoose('"hello"'),
+            extractJsonLoose('123'),
+            extractJsonLoose('true'),
+            extractJsonLoose('null'),
+        ]
+        let rr = [null, null, null, null]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('無JSON內容回傳null', function() {
+        let r = [
+            extractJsonLoose('純文字回覆'),
+            extractJsonLoose(''),
+            extractJsonLoose(null),
+            extractJsonLoose(undefined),
+        ]
+        let rr = [null, null, null, null]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('第一個片段非法時回傳null(不繼續嘗試後續片段)', function() {
+        let r = extractJsonLoose('{"a":1,} {"b":2}')
+        let rr = null
+        assert.strict.deepEqual(r, rr)
+    })
+
+})

package/test/unit-runFanout.test.mjs

@@ -0,0 +1,166 @@
+import assert from 'assert'
+import runFanout, { defaultIntegratePrompt } from '../src/wkf/runFanout.mjs'
+import createFakeCli from './tools/fakeCliForTest.mjs'
+
+
+describe('runFanout', function() {
+
+    let fake = null
+    let providers = null
+    let callOpt = { promptPrefix: '' }
+
+    before(function() {
+        fake = createFakeCli('fake-wkf-fanout')
+        providers = {
+            'p-a': { kind: 'claude', exe: fake.exe, model: 'sonnet' },
+            'p-b': { kind: 'codex', exe: fake.exe, model: 'gpt-5.6-luna' },
+            'p-int': { kind: 'opencode', exe: fake.exe, model: 'opencode/free' },
+            'p-dead': { kind: 'claude', exe: fake.exe, extraArgs: ['--fake-exit=1'] },
+        }
+    })
+
+    after(function() {
+        if (fake) {
+            fake.clean()
+        }
+    })
+
+    it('task或agents無效時回傳錯誤結果物件', async function() {
+        let t1 = await runFanout({ providers, task: '', agents: [{ use: 'p-a' }], callOpt })
+        let t2 = await runFanout({ providers, task: 'abc', agents: null, callOpt })
+        let r = [[t1.ok, t1.error], [t2.ok, t2.error]]
+        let rr = [
+            [false, 'task must be a non-empty string'],
+            [false, 'agents must be a non-empty array'],
+        ]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('候選達門檻時進整合, 整合提示詞含各候選', async function() {
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            integrate: { use: 'p-int' },
+            callOpt,
+        })
+        let r = [
+            t.ok,
+            t.integrated,
+            t.candidates.length,
+            t.agents.map((x) => x.providerId),
+            t.result.args[0], //整合者為opencode轉接器
+            t.result.stdin.includes('【候選 1】') && t.result.stdin.includes('【候選 2】'),
+            t.integrateDetail.providerId,
+        ]
+        let rr = [true, true, 2, ['p-a', 'p-b'], 'run', true, 'p-int']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('個別名額失敗不炸整輪, 恰1份成功候選直接放行(integrated:false)', async function() {
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-dead' }, { use: 'p-a' }],
+            integrate: { use: 'p-int' },
+            callOpt,
+        })
+        let r = [t.ok, t.integrated, t.candidates.length, t.result.args, t.agents[0].ok, t.agents[1].ok]
+        let rr = [true, false, 1, ['-p', '--dangerously-skip-permissions', '--model', 'sonnet'], false, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('候選未達minCandidates時不硬整合, 以首位候選為成果', async function() {
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            integrate: { use: 'p-int' },
+            minCandidates: 3,
+            callOpt,
+        })
+        let r = [t.ok, t.integrated, t.candidates.length, t.result === t.candidates[0], t.error]
+        let rr = [true, false, 2, true, '']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('全數名額失敗時回傳all agents failed', async function() {
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-dead' }, { use: 'p-dead' }],
+            integrate: { use: 'p-int' },
+            callOpt,
+        })
+        let r = [t.ok, t.result, t.candidates.length, t.error]
+        let rr = [false, null, 0, 'all agents failed']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('達門檻但缺integrate規格時回報必要性', async function() {
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            callOpt,
+        })
+        let r = [t.ok, t.error, t.candidates.length]
+        let rr = [false, 'integrate spec (with use) is required', 2]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('整合失敗時成功候選仍完整回傳供接續重試', async function() {
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            integrate: { use: 'p-dead' },
+            callOpt,
+        })
+        let r = [t.ok, t.integrated, t.candidates.length, t.error.indexOf('integrate failed:') === 0]
+        let rr = [false, false, 2, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('integrate.prompt可自訂整合提示詞', async function() {
+        let seen = null
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            integrate: {
+                use: 'p-int',
+                prompt: (candidates) => {
+                    seen = candidates.length
+                    return 'CUSTOM-INTEGRATE'
+                },
+            },
+            callOpt,
+        })
+        let r = [t.ok, seen, t.result.stdin]
+        let rr = [true, 2, 'CUSTOM-INTEGRATE']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('名額規格之額外鍵覆寫該名額呼叫設定', async function() {
+        //p-a名額帶rawText覆寫 → 該名額json為文字, 候選為字串
+        let t = await runFanout({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a', rawText: true }],
+            integrate: { use: 'p-int' },
+            callOpt,
+        })
+        let r = [t.ok, t.integrated, typeof t.result]
+        let rr = [true, false, 'string']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('defaultIntegratePrompt嵌入候選數與schema', function() {
+        let s = defaultIntegratePrompt([{ a: 1 }, { b: 2 }], { schema: '{"x":"..."}' })
+        let r = [s.includes('2 個獨立執行'), s.includes('{"x":"..."}'), s.includes('【候選 2】')]
+        let rr = [true, true, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+})

package/test/unit-runFanoutPipeline.test.mjs

@@ -0,0 +1,86 @@
+import assert from 'assert'
+import runFanoutPipeline from '../src/wkf/runFanoutPipeline.mjs'
+import createFakeCli from './tools/fakeCliForTest.mjs'
+
+
+describe('runFanoutPipeline', function() {
+
+    let fake = null
+    let providers = null
+    let callOpt = { promptPrefix: '' }
+
+    before(function() {
+        fake = createFakeCli('fake-wkf-fp')
+        providers = {
+            'p-a': { kind: 'claude', exe: fake.exe, model: 'sonnet' },
+            'p-b': { kind: 'codex', exe: fake.exe, model: 'gpt-5.6-luna' },
+            'p-int': { kind: 'opencode', exe: fake.exe, model: 'opencode/free' },
+            'p-dead': { kind: 'claude', exe: fake.exe, extraArgs: ['--fake-exit=1'] },
+        }
+    })
+
+    after(function() {
+        if (fake) {
+            fake.clean()
+        }
+    })
+
+    it('前段多開整合, 後段以前段成果為input執行角色鏈', async function() {
+        let t = await runFanoutPipeline({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            integrate: { use: 'p-int' },
+            stages: [
+                { id: 'audit', use: 'p-b', prompt: (ctx) => `audit ${ctx.input.args[0]}` }, //input為整合者echo物件
+            ],
+            callOpt,
+        })
+        let r = [
+            t.ok,
+            t.A.ok,
+            t.A.integrated,
+            t.B.ok,
+            t.B.order,
+            t.result.stdin, //後段最末階段之echo, 其stdin含前段成果之args[0](整合者為opencode轉接器故為run)
+        ]
+        let rr = [true, true, true, true, ['audit'], 'audit run']
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('前段失敗即回, 後段不執行', async function() {
+        let t = await runFanoutPipeline({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-dead' }, { use: 'p-dead' }],
+            integrate: { use: 'p-int' },
+            stages: [{ id: 's1', use: 'p-a', prompt: () => 'x' }],
+            callOpt,
+        })
+        let r = [t.ok, t.A.ok, t.B, t.error.indexOf('A failed:') === 0]
+        let rr = [false, false, null, true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+    it('後段失敗時前段成果仍完整回傳, 可只重跑後段', async function() {
+        let t = await runFanoutPipeline({
+            providers,
+            task: 'abc',
+            agents: [{ use: 'p-a' }, { use: 'p-b' }],
+            integrate: { use: 'p-int' },
+            stages: [{ id: 's1', use: 'p-dead', prompt: () => 'x' }],
+            callOpt,
+        })
+        let r = [
+            t.ok,
+            t.A.ok,
+            t.A.result !== null,
+            t.B.ok,
+            t.B.failedStage,
+            t.error.indexOf('B failed:') === 0,
+        ]
+        let rr = [false, true, true, false, 's1', true]
+        assert.strict.deepEqual(r, rr)
+    })
+
+})