@zimtsui/brainswitch 0.0.54 → 0.0.56

package/package.json

@@ -1,13 +1,13 @@
 {
   "name": "@zimtsui/brainswitch",
-  "version": "0.0.54",
+  "version": "0.0.56",
   "exports": {
     ".": "./build/exports.js",
     "./openai-responses": "./build/native-engines.d/openai-responses/exports.js",
     "./google": "./build/native-engines.d/google/exports.js"
   },
   "imports": {
-    "#config": "./build/config.js"
+    "#@/*.ts": "./build/*.js"
   },
   "description": "",
   "engines": {
@@ -15,7 +15,7 @@
   },
   "type": "module",
   "scripts": {
-    "test": "ava ./build/test.js",
+    "test": "tsc -b ./test/src/tsconfig.json && ava ./test/build/test.js",
     "build": "tsc -b ./src/tsconfig.json",
     "clean": "rm -rf ./build && rm -rf ./docs",
     "docs": "rm -rf ./docs && typedoc --tsconfig ./src/tsconfig.json ./src/exports.ts ./src/native-engines.d/*/exports.ts",
@@ -37,7 +37,7 @@
     "@sinclair/typebox": "^0.34.40",
     "@types/node": "^22.17.2",
     "@zimtsui/coroutine-locks": "^0.0.25",
-    "@zimtsui/typelog": "^0.0.28"
+    "@zimtsui/typelog": "^0.0.29"
   },
   "devDependencies": {
     "ava": "^7.0.0"

package/test/build/test-codecs.js

@@ -0,0 +1,230 @@
+import test from 'ava';
+import { Function } from '#@/function.ts';
+import { ResponseInvalid } from '#@/engine.ts';
+import { ToolCallValidator } from '#@/compatible/tool-call-validator.ts';
+import { GoogleToolCodec } from '#@/api-types/google/tool-codec.ts';
+import { GoogleCompatibleMessageCodec } from '#@/compatible.d/google/message-codec.ts';
+import { GoogleNativeMessageCodec } from '#@/native-engines.d/google/message-codec.ts';
+import { OpenAIResponsesToolCodec } from '#@/api-types/openai-responses/tool-codec.ts';
+import { OpenAIResponsesCompatibleMessageCodec } from '#@/compatible.d/openai-responses/message-codec.ts';
+import { OpenAIResponsesNativeMessageCodec } from '#@/native-engines.d/openai-responses/message-codec.ts';
+import { OpenAIResponsesNativeToolCallValidator } from '#@/native-engines.d/openai-responses/tool-call-validator.ts';
+import { Tool } from '#@/native-engines.d/openai-responses/tool.ts';
+import { GoogleBilling } from '#@/api-types/google/billing.ts';
+import { OpenAIResponsesBilling } from '#@/api-types/openai-responses/billing.ts';
+import { AnthropicBilling } from '#@/api-types/anthropic/billing.ts';
+import { RoleMessage } from '#@/compatible/session.ts';
+import * as Google from '@google/genai';
+import * as OpenAIResponsesNative from '#@/native-engines.d/openai-responses/session.ts';
+import { fdm } from "./test-helpers.js";
+function makeFooCall() {
+    return Function.Call.create({
+        id: 'fc-1',
+        name: 'foo',
+        args: { x: 1 },
+    });
+}
+function makeBarCall() {
+    return Function.Call.create({
+        id: 'fc-2',
+        name: 'bar',
+        args: { y: 'x' },
+    });
+}
+test('ToolCallValidator enforces REQUIRED / allowlist / NONE', t => {
+    const validator = new ToolCallValidator({ toolChoice: Function.ToolChoice.REQUIRED });
+    const fooCall = makeFooCall();
+    const barCall = makeBarCall();
+    t.throws(() => validator.validate([]), { instanceOf: ResponseInvalid });
+    t.notThrows(() => validator.validate([fooCall]));
+    validator['ctx'].toolChoice = ['foo'];
+    t.notThrows(() => validator.validate([fooCall]));
+    t.throws(() => validator.validate([barCall]), { instanceOf: ResponseInvalid });
+    validator['ctx'].toolChoice = Function.ToolChoice.NONE;
+    t.notThrows(() => validator.validate([]));
+    t.throws(() => validator.validate([fooCall]), { instanceOf: ResponseInvalid });
+});
+test('OpenAIResponsesNativeToolCallValidator enforces apply_patch allowlist', t => {
+    const fooCall = makeFooCall();
+    const applyPatchCall = Tool.ApplyPatch.Call.create({
+        id: 'patch-1',
+        call_id: 'call-1',
+        operation: {
+            type: 'update_file',
+            path: 'a.ts',
+            diff: 'diff',
+        },
+        status: 'completed',
+        type: 'apply_patch_call',
+    });
+    const validator = new OpenAIResponsesNativeToolCallValidator({
+        toolChoice: ['foo', Tool.Choice.APPLY_PATCH],
+    });
+    t.notThrows(() => validator.validate([fooCall, applyPatchCall]));
+    const validatorWithoutPatch = new OpenAIResponsesNativeToolCallValidator({
+        toolChoice: ['foo'],
+    });
+    t.throws(() => validatorWithoutPatch.validate([applyPatchCall]), { instanceOf: ResponseInvalid });
+});
+test('GoogleCompatibleMessageCodec converts user and ai messages', t => {
+    const toolCodec = new GoogleToolCodec({ fdm });
+    const codec = new GoogleCompatibleMessageCodec({ toolCodec });
+    const userMessage = RoleMessage.User.create([
+        RoleMessage.Part.Text.create('hello'),
+        Function.Response.create({
+            id: 'fc-1',
+            name: 'foo',
+            text: '{"ok":true}',
+        }),
+    ]);
+    const userContent = codec.convertFromUserMessage(userMessage);
+    t.is(userContent.role, 'user');
+    t.is(userContent.parts?.[0]?.text, 'hello');
+    t.truthy(userContent.parts?.[1]?.functionResponse);
+    const aiContent = {
+        role: 'model',
+        parts: [
+            { text: 'hi' },
+            { functionCall: { id: 'fc-1', name: 'foo', args: { x: 1 } } },
+        ],
+    };
+    const aiMessage = codec.convertToAiMessage(aiContent);
+    t.is(aiMessage.getText(), 'hi');
+    t.is(aiMessage.getFunctionCalls()[0]?.name, 'foo');
+});
+test('GoogleNativeMessageCodec reuses compatible message codec and decodes code execution parts', t => {
+    const toolCodec = new GoogleToolCodec({ fdm });
+    const compatibleMessageCodec = new GoogleCompatibleMessageCodec({ toolCodec });
+    const codec = new GoogleNativeMessageCodec({
+        toolCodec,
+        compatibleMessageCodec,
+        codeExecution: true,
+    });
+    const userContent = codec.convertFromUserMessage(RoleMessage.User.create([
+        RoleMessage.Part.Text.create('hello'),
+    ]));
+    t.is(userContent.role, 'user');
+    const aiMessage = codec.convertToAiMessage({
+        role: 'model',
+        parts: [
+            { text: 'hi' },
+            { functionCall: { id: 'fc-1', name: 'foo', args: { x: 1 } } },
+            { executableCode: { code: 'print(1)', language: Google.Language.PYTHON } },
+            { codeExecutionResult: { outcome: Google.Outcome.OUTCOME_OK, output: '1' } },
+        ],
+    });
+    t.is(aiMessage.getText(), 'hi');
+    t.is(aiMessage.getFunctionCalls()[0]?.name, 'foo');
+    t.is(aiMessage.getParts().length, 4);
+});
+test('OpenAIResponsesCompatibleMessageCodec converts user and ai messages', t => {
+    const toolCodec = new OpenAIResponsesToolCodec({ fdm });
+    const codec = new OpenAIResponsesCompatibleMessageCodec({ toolCodec });
+    const userMessage = RoleMessage.User.create([
+        RoleMessage.Part.Text.create('hello'),
+        Function.Response.create({
+            id: 'fc-1',
+            name: 'foo',
+            text: '{"ok":true}',
+        }),
+    ]);
+    const input = codec.convertFromUserMessage(userMessage);
+    t.is(input[0]?.type, 'message');
+    t.is(input[1]?.type, 'function_call_output');
+    const aiMessage = codec.convertToAiMessage([
+        {
+            id: 'msg_1',
+            type: 'message',
+            role: 'assistant',
+            status: 'completed',
+            content: [{ type: 'output_text', text: 'hi', annotations: [] }],
+        },
+        {
+            id: 'fc_1',
+            type: 'function_call',
+            call_id: 'call_1',
+            name: 'foo',
+            arguments: '{"x":1}',
+            status: 'completed',
+        },
+    ]);
+    t.is(aiMessage.getText(), 'hi');
+    t.is(aiMessage.getFunctionCalls()[0]?.name, 'foo');
+});
+test('OpenAIResponsesNativeMessageCodec converts apply_patch response', t => {
+    const toolCodec = new OpenAIResponsesToolCodec({ fdm });
+    const compatibleMessageCodec = new OpenAIResponsesCompatibleMessageCodec({ toolCodec });
+    const codec = new OpenAIResponsesNativeMessageCodec({
+        toolCodec,
+        compatibleMessageCodec,
+    });
+    const input = codec.convertFromUserMessage(OpenAIResponsesNative.RoleMessage.User.create([
+        OpenAIResponsesNative.RoleMessage.Part.Text.create('hello'),
+        Function.Response.create({
+            id: 'fc-1',
+            name: 'foo',
+            text: '{"ok":true}',
+        }),
+        Tool.ApplyPatch.Response.create({
+            id: 'patch-1',
+            failure: 'failure',
+        }),
+    ]));
+    t.is(input[0]?.type, 'message');
+    t.is(input[1]?.type, 'function_call_output');
+    t.is(input[2]?.type, 'apply_patch_call_output');
+});
+test('GoogleBilling calculates cache hit / miss / thinking cost', t => {
+    const billing = new GoogleBilling({
+        pricing: {
+            inputPrice: 1,
+            cachePrice: 2,
+            outputPrice: 3,
+        },
+    });
+    const charge = billing.charge({
+        promptTokenCount: 100,
+        cachedContentTokenCount: 25,
+        candidatesTokenCount: 10,
+        thoughtsTokenCount: 5,
+    });
+    t.is(charge, (75 + 50 + 30 + 15) / 1e6);
+});
+test('OpenAIResponsesBilling calculates cache hit / miss cost', t => {
+    const billing = new OpenAIResponsesBilling({
+        pricing: {
+            inputPrice: 1,
+            cachePrice: 2,
+            outputPrice: 3,
+        },
+    });
+    const charge = billing.charge({
+        input_tokens: 100,
+        input_tokens_details: { cached_tokens: 20 },
+        output_tokens: 10,
+        output_tokens_details: { reasoning_tokens: 0 },
+        total_tokens: 110,
+    });
+    t.true(Math.abs(charge - (80 + 40 + 30) / 1e6) < 1e-12);
+});
+test('AnthropicBilling calculates cache hit / miss cost', t => {
+    const billing = new AnthropicBilling({
+        pricing: {
+            inputPrice: 1,
+            cachePrice: 2,
+            outputPrice: 3,
+        },
+    });
+    const charge = billing.charge({
+        input_tokens: 100,
+        output_tokens: 10,
+        cache_read_input_tokens: 20,
+        cache_creation_input_tokens: 0,
+        cache_creation: { ephemeral_5m_input_tokens: 0, ephemeral_1h_input_tokens: 0 },
+        inference_geo: 'unknown',
+        server_tool_use: { web_fetch_requests: 0, web_search_requests: 0 },
+        service_tier: 'standard',
+    });
+    t.true(Math.abs(charge - (80 + 40 + 30) / 1e6) < 1e-12);
+});
+//# sourceMappingURL=test-codecs.js.map

package/test/build/test-codecs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-codecs.js","sourceRoot":"","sources":["../src/test-codecs.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,KAAK,CAAC;AACvB,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,4BAA4B,EAAE,MAAM,yCAAyC,CAAC;AACvF,OAAO,EAAE,wBAAwB,EAAE,MAAM,6CAA6C,CAAC;AACvF,OAAO,EAAE,wBAAwB,EAAE,MAAM,6CAA6C,CAAC;AACvF,OAAO,EAAE,qCAAqC,EAAE,MAAM,mDAAmD,CAAC;AAC1G,OAAO,EAAE,iCAAiC,EAAE,MAAM,uDAAuD,CAAC;AAC1G,OAAO,EAAE,sCAAsC,EAAE,MAAM,6DAA6D,CAAC;AACrH,OAAO,EAAE,IAAI,EAAE,MAAM,8CAA8C,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,0CAA0C,CAAC;AAClF,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,qBAAqB,MAAM,iDAAiD,CAAC;AACzF,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AAGxC,SAAS,WAAW;IAChB,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAM;QAC7B,EAAE,EAAE,MAAM;QACV,IAAI,EAAE,KAAK;QACX,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE;KACjB,CAAC,CAAC;AACP,CAAC;AAED,SAAS,WAAW;IAChB,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAM;QAC7B,EAAE,EAAE,MAAM;QACV,IAAI,EAAE,KAAK;QACX,IAAI,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE;KACnB,CAAC,CAAC;AACP,CAAC;AAGD,IAAI,CAAC,wDAAwD,EAAE,CAAC,CAAC,EAAE;IAC/D,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC3F,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;IAC9B,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;IAE9B,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjD,SAAS,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;IAE/E,SAAS,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;IACvD,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACnF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uEAAuE,EAAE,CAAC,CAAC,EAAE;IAC9E,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;IAC9B,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC;QAC/C,EAAE,EAAE,SAAS;QACb,OAAO,EAAE,QAAQ;QACjB,SAAS,EAAE;YACP,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,MAAM;SACf;QACD,MAAM,EAAE,WAAW;QACnB,IAAI,EAAE,kBAAkB;KAC3B,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,sCAAsC,CAAM;QAC9D,UAAU,EAAE,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;KAC/C,CAAC,CAAC;IACH,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC;IAEjE,MAAM,qBAAqB,GAAG,IAAI,sCAAsC,CAAM;QAC1E,UAAU,EAAE,CAAC,KAAK,CAAC;KACtB,CAAC,CAAC;IACH,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACtG,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,4DAA4D,EAAE,CAAC,CAAC,EAAE;IACnE,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,4BAA4B,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;IAE9D,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAM;QAC7C,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QACrC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAM;YAC1B,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,aAAa;SACtB,CAAC;KACL,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,KAAK,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;IAC9D,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC/B,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAEnD,MAAM,SAAS,GAAG;QACd,IAAI,EAAE,OAAO;QACb,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,IAAI,EAAE;YACd,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE;SAChE;KACJ,CAAC;IACF,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;AACvD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2FAA2F,EAAE,CAAC,CAAC,EAAE;IAClG,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,MAAM,sBAAsB,GAAG,IAAI,4BAA4B,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;IAC/E,MAAM,KAAK,GAAG,IAAI,wBAAwB,CAAC;QACvC,SAAS;QACT,sBAAsB;QACtB,aAAa,EAAE,IAAI;KACtB,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,KAAK,CAAC,sBAAsB,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAM;QAC1E,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;KACxC,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAE/B,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAC;QACvC,IAAI,EAAE,OAAO;QACb,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,IAAI,EAAE;YACd,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE;YAC7D,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE;YAC1E,EAAE,mBAAmB,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;SAC/E;KACJ,CAAC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qEAAqE,EAAE,CAAC,CAAC,EAAE;IAC5E,MAAM,SAAS,GAAG,IAAI,wBAAwB,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,IAAI,qCAAqC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;IAEvE,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAM;QAC7C,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QACrC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAM;YAC1B,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,aAAa;SACtB,CAAC;KACL,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,KAAK,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,sBAAsB,CAAC,CAAC;IAE7C,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAC;QACvC;YACI,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,WAAW;YACnB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;SAClE;QACD;YACI,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,QAAQ;YACjB,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,SAAS;YACpB,MAAM,EAAE,WAAW;SACtB;KACJ,CAAC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;AACvD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iEAAiE,EAAE,CAAC,CAAC,EAAE;IACxE,MAAM,SAAS,GAAG,IAAI,wBAAwB,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,MAAM,sBAAsB,GAAG,IAAI,qCAAqC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;IACxF,MAAM,KAAK,GAAG,IAAI,iCAAiC,CAAC;QAChD,SAAS;QACT,sBAAsB;KACzB,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,KAAK,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAEtF;QACE,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QAC3D,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAM;YAC1B,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,aAAa;SACtB,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC5B,EAAE,EAAE,SAAS;YACb,OAAO,EAAE,SAAS;SACrB,CAAC;KACL,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,sBAAsB,CAAC,CAAC;IAC7C,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;AACpD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,CAAC,CAAC,EAAE;IAClE,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC;QAC9B,OAAO,EAAE;YACL,UAAU,EAAE,CAAC;YACb,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;SACjB;KACJ,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC1B,gBAAgB,EAAE,GAAG;QACrB,uBAAuB,EAAE,EAAE;QAC3B,oBAAoB,EAAE,EAAE;QACxB,kBAAkB,EAAE,CAAC;KACxB,CAAC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yDAAyD,EAAE,CAAC,CAAC,EAAE;IAChE,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC;QACvC,OAAO,EAAE;YACL,UAAU,EAAE,CAAC;YACb,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;SACjB;KACJ,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC1B,YAAY,EAAE,GAAG;QACjB,oBAAoB,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;QAC3C,aAAa,EAAE,EAAE;QACjB,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE;QAC9C,YAAY,EAAE,GAAG;KACpB,CAAC,CAAC;IACH,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;AAC5D,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mDAAmD,EAAE,CAAC,CAAC,EAAE;IAC1D,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;QACjC,OAAO,EAAE;YACL,UAAU,EAAE,CAAC;YACb,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;SACjB;KACJ,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC1B,YAAY,EAAE,GAAG;QACjB,aAAa,EAAE,EAAE;QACjB,uBAAuB,EAAE,EAAE;QAC3B,2BAA2B,EAAE,CAAC;QAC9B,cAAc,EAAE,EAAE,yBAAyB,EAAE,CAAC,EAAE,yBAAyB,EAAE,CAAC,EAAE;QAC9E,aAAa,EAAE,SAAS;QACxB,eAAe,EAAE,EAAE,kBAAkB,EAAE,CAAC,EAAE,mBAAmB,EAAE,CAAC,EAAE;QAClE,YAAY,EAAE,UAAU;KAC3B,CAAC,CAAC;IACH,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;AAC5D,CAAC,CAAC,CAAC"}

package/test/build/test-compatible-engine.d.ts

@@ -0,0 +1 @@
+export {};

package/test/build/test-compatible-engine.js

@@ -0,0 +1,27 @@
+import test from 'ava';
+import { CompatibleEngine } from '#@/compatible/engine.ts';
+import { RoleMessage } from '#@/compatible/session.ts';
+import { Function } from '#@/function.ts';
+import {} from '#@/inference-context.ts';
+import { fdm, makeAiMessage, makeBaseOptions, } from "./test-helpers.js";
+class StubCompatibleEngine extends CompatibleEngine {
+    parallelToolCall = false;
+    responder = async () => makeAiMessage();
+    getToolChoice() {
+        return this.toolChoice;
+    }
+    setToolChoice(toolChoice) {
+        this.toolChoice = toolChoice;
+    }
+    infer(wfctx, session, signal) {
+        return this.responder(wfctx, session, signal);
+    }
+}
+test('CompatibleEngine stores default tool choice', t => {
+    const engine = new StubCompatibleEngine({
+        ...makeBaseOptions(),
+        apiType: 'google',
+    });
+    t.is(engine.getToolChoice(), Function.ToolChoice.AUTO);
+});
+//# sourceMappingURL=test-compatible-engine.js.map

package/test/build/test-compatible-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-compatible-engine.js","sourceRoot":"","sources":["../src/test-compatible-engine.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,KAAK,CAAC;AACvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAyB,MAAM,yBAAyB,CAAC;AAChE,OAAO,EACH,GAAG,EACH,aAAa,EACb,eAAe,GAClB,MAAM,mBAAmB,CAAC;AAG3B,MAAM,oBAAqB,SAAQ,gBAAqB;IACjC,gBAAgB,GAAG,KAAK,CAAC;IACrC,SAAS,GAIoB,KAAK,IAAI,EAAE,CAAC,aAAa,EAAE,CAAC;IAEzD,aAAa;QAChB,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEM,aAAa,CAAC,UAAoC;QACrD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAEe,KAAK,CACjB,KAAuB,EACvB,OAAqB,EACrB,MAAoB;QAEpB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;CACJ;AAGD,IAAI,CAAC,6CAA6C,EAAE,CAAC,CAAC,EAAE;IACpD,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;QACpC,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;AAC3D,CAAC,CAAC,CAAC"}

package/test/build/test-compatible-engines.d.ts

@@ -0,0 +1 @@
+export {};

package/test/build/test-compatible-engines.js

@@ -0,0 +1,112 @@
+import test from 'ava';
+import { Adaptor } from '#@/adaptor.ts';
+import { Function } from '#@/function.ts';
+import { OpenAIResponsesCompatibleEngine } from '#@/compatible/engine.d/openai-responses.ts';
+import { GoogleCompatibleEngine } from '#@/compatible/engine.d/google.ts';
+import { AnthropicCompatibleEngine } from '#@/compatible/engine.d/anthropic.ts';
+import { AliyunEngine } from '#@/compatible/engine.d/aliyun.ts';
+import { fdm, makeBaseOptions } from "./test-helpers.js";
+class OpenAIResponsesCompatibleEngineProbe extends OpenAIResponsesCompatibleEngine {
+    getToolChoice() {
+        return this.toolChoice;
+    }
+    getParallelToolCall() {
+        return this.parallelToolCall;
+    }
+}
+class GoogleCompatibleEngineProbe extends GoogleCompatibleEngine {
+    getToolChoice() {
+        return this.toolChoice;
+    }
+    getParallelToolCall() {
+        return this.parallelToolCall;
+    }
+}
+class AnthropicCompatibleEngineProbe extends AnthropicCompatibleEngine {
+    getToolChoice() {
+        return this.toolChoice;
+    }
+    getParallelToolCall() {
+        return this.parallelToolCall;
+    }
+}
+class AliyunEngineProbe extends AliyunEngine {
+    getToolChoice() {
+        return this.toolChoice;
+    }
+    getParallelToolCall() {
+        return this.parallelToolCall;
+    }
+}
+test('compatible engines preserve aggregated defaults', t => {
+    const openaiEngine = new OpenAIResponsesCompatibleEngineProbe({
+        ...makeBaseOptions(),
+        apiType: 'openai-responses',
+    });
+    const googleEngine = new GoogleCompatibleEngineProbe({
+        ...makeBaseOptions(),
+        apiType: 'google',
+    });
+    const anthropicEngine = new AnthropicCompatibleEngineProbe({
+        ...makeBaseOptions(),
+        apiType: 'anthropic',
+    });
+    const aliyunEngine = new AliyunEngineProbe({
+        ...makeBaseOptions(),
+        apiType: 'aliyun',
+    });
+    t.is(openaiEngine.fdm, fdm);
+    t.is(openaiEngine.getToolChoice(), Function.ToolChoice.AUTO);
+    t.false(openaiEngine.getParallelToolCall());
+    t.is(googleEngine.getToolChoice(), Function.ToolChoice.AUTO);
+    t.true(googleEngine.getParallelToolCall());
+    t.is(anthropicEngine.getToolChoice(), Function.ToolChoice.AUTO);
+    t.false(anthropicEngine.getParallelToolCall());
+    t.is(aliyunEngine.getToolChoice(), Function.ToolChoice.AUTO);
+    t.false(aliyunEngine.getParallelToolCall());
+});
+test('Adaptor.makeEngine returns each compatible implementation', t => {
+    const adaptor = Adaptor.create({
+        brainswitch: {
+            endpoints: {
+                oa: {
+                    baseUrl: 'https://example.com/v1',
+                    apiKey: 'k',
+                    model: 'm',
+                    name: 'oa',
+                    apiType: 'openai-responses',
+                },
+                gg: {
+                    baseUrl: 'https://generativelanguage.googleapis.com',
+                    apiKey: 'k',
+                    model: 'gemini-test',
+                    name: 'gg',
+                    apiType: 'google',
+                },
+                an: {
+                    baseUrl: 'https://api.anthropic.com',
+                    apiKey: 'k',
+                    model: 'claude-test',
+                    name: 'an',
+                    apiType: 'anthropic',
+                },
+                al: {
+                    baseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
+                    apiKey: 'k',
+                    model: 'qwen-test',
+                    name: 'al',
+                    apiType: 'aliyun',
+                },
+            },
+        },
+    });
+    const openaiEngine = adaptor.makeEngine('oa', fdm);
+    const googleEngine = adaptor.makeEngine('gg', fdm);
+    const anthropicEngine = adaptor.makeEngine('an', fdm);
+    const aliyunEngine = adaptor.makeEngine('al', fdm);
+    t.true(openaiEngine instanceof OpenAIResponsesCompatibleEngine);
+    t.true(googleEngine instanceof GoogleCompatibleEngine);
+    t.true(anthropicEngine instanceof AnthropicCompatibleEngine);
+    t.true(aliyunEngine instanceof AliyunEngine);
+});
+//# sourceMappingURL=test-compatible-engines.js.map

package/test/build/test-compatible-engines.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-compatible-engines.js","sourceRoot":"","sources":["../src/test-compatible-engines.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,KAAK,CAAC;AACvB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,+BAA+B,EAAE,MAAM,4CAA4C,CAAC;AAC7F,OAAO,EAAE,sBAAsB,EAAE,MAAM,kCAAkC,CAAC;AAC1E,OAAO,EAAE,yBAAyB,EAAE,MAAM,qCAAqC,CAAC;AAChF,OAAO,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAC;AAChE,OAAO,EAAE,GAAG,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGzD,MAAM,oCAAqC,SAAQ,+BAAoC;IAC5E,aAAa;QAChB,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEM,mBAAmB;QACtB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IACjC,CAAC;CACJ;AAED,MAAM,2BAA4B,SAAQ,sBAA2B;IAC1D,aAAa;QAChB,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEM,mBAAmB;QACtB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IACjC,CAAC;CACJ;AAED,MAAM,8BAA+B,SAAQ,yBAA8B;IAChE,aAAa;QAChB,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEM,mBAAmB;QACtB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IACjC,CAAC;CACJ;AAED,MAAM,iBAAkB,SAAQ,YAAiB;IACtC,aAAa;QAChB,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEM,mBAAmB;QACtB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IACjC,CAAC;CACJ;AAGD,IAAI,CAAC,iDAAiD,EAAE,CAAC,CAAC,EAAE;IACxD,MAAM,YAAY,GAAG,IAAI,oCAAoC,CAAC;QAC1D,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,kBAAkB;KAC9B,CAAC,CAAC;IACH,MAAM,YAAY,GAAG,IAAI,2BAA2B,CAAC;QACjD,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IACH,MAAM,eAAe,GAAG,IAAI,8BAA8B,CAAC;QACvD,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,WAAW;KACvB,CAAC,CAAC;IACH,MAAM,YAAY,GAAG,IAAI,iBAAiB,CAAC;QACvC,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IAEH,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC5B,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAE5C,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAE3C,CAAC,CAAC,EAAE,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAChE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAE/C,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC,CAAC;AAChD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,CAAC,CAAC,EAAE;IAClE,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;QAC3B,WAAW,EAAE;YACT,SAAS,EAAE;gBACP,EAAE,EAAE;oBACA,OAAO,EAAE,wBAAwB;oBACjC,MAAM,EAAE,GAAG;oBACX,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,IAAI;oBACV,OAAO,EAAE,kBAAkB;iBAC9B;gBACD,EAAE,EAAE;oBACA,OAAO,EAAE,2CAA2C;oBACpD,MAAM,EAAE,GAAG;oBACX,KAAK,EAAE,aAAa;oBACpB,IAAI,EAAE,IAAI;oBACV,OAAO,EAAE,QAAQ;iBACpB;gBACD,EAAE,EAAE;oBACA,OAAO,EAAE,2BAA2B;oBACpC,MAAM,EAAE,GAAG;oBACX,KAAK,EAAE,aAAa;oBACpB,IAAI,EAAE,IAAI;oBACV,OAAO,EAAE,WAAW;iBACvB;gBACD,EAAE,EAAE;oBACA,OAAO,EAAE,mDAAmD;oBAC5D,MAAM,EAAE,GAAG;oBACX,KAAK,EAAE,WAAW;oBAClB,IAAI,EAAE,IAAI;oBACV,OAAO,EAAE,QAAQ;iBACpB;aACJ;SACJ;KACJ,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACnD,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACnD,MAAM,eAAe,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACtD,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAEnD,CAAC,CAAC,IAAI,CAAC,YAAY,YAAY,+BAA+B,CAAC,CAAC;IAChE,CAAC,CAAC,IAAI,CAAC,YAAY,YAAY,sBAAsB,CAAC,CAAC;IACvD,CAAC,CAAC,IAAI,CAAC,eAAe,YAAY,yBAAyB,CAAC,CAAC;IAC7D,CAAC,CAAC,IAAI,CAAC,YAAY,YAAY,YAAY,CAAC,CAAC;AACjD,CAAC,CAAC,CAAC"}

package/test/build/test-engine.d.ts

@@ -0,0 +1 @@
+export {};

package/test/build/test-engine.js

@@ -0,0 +1,97 @@
+import test from 'ava';
+import { Engine, InferenceTimeout, ResponseInvalid, UserAbortion } from '#@/engine.ts';
+import {} from '#@/session.ts';
+import {} from '#@/inference-context.ts';
+import {} from '#@/compatible/session.ts';
+import { fdm, makeBaseOptions, makeInferenceContext, makeUserMessage, } from "./test-helpers.js";
+class StubEngine extends Engine {
+    parallelToolCall = false;
+    responder = async () => 'ok';
+    infer(wfctx, session, signal) {
+        return this.responder(wfctx, session, signal);
+    }
+    appendUserMessage(session, message) {
+        return {
+            ...session,
+            chatMessages: [...session.chatMessages, message],
+        };
+    }
+    pushUserMessage(session, message) {
+        session.chatMessages.push(message);
+        return session;
+    }
+}
+function makeStringSession() {
+    return { chatMessages: [] };
+}
+test('Engine stores function declaration map', t => {
+    const engine = new StubEngine({
+        ...makeBaseOptions(),
+        apiType: 'google',
+    });
+    t.is(engine.fdm, fdm);
+});
+test('Engine.stateless retries ResponseInvalid and succeeds', async (t) => {
+    const engine = new StubEngine({
+        ...makeBaseOptions(),
+        apiType: 'google',
+    });
+    let attempts = 0;
+    engine.responder = async () => {
+        attempts += 1;
+        if (attempts < 3)
+            throw new ResponseInvalid('temporary');
+        return 'retry-ok';
+    };
+    const response = await engine.stateless(makeInferenceContext(), makeStringSession());
+    t.is(attempts, 3);
+    t.is(response, 'retry-ok');
+});
+test('Engine.stateless converts aborted workflow signal to UserAbortion', async (t) => {
+    const engine = new StubEngine({
+        ...makeBaseOptions(),
+        apiType: 'google',
+    });
+    const controller = new AbortController();
+    engine.responder = async () => {
+        controller.abort();
+        throw new TypeError('aborted');
+    };
+    const e = await engine.stateless(makeInferenceContext(controller.signal), makeStringSession()).catch((e) => e);
+    t.true(e instanceof UserAbortion);
+});
+test('Engine.stateless converts timeout abort to InferenceTimeout', async (t) => {
+    const engine = new StubEngine({
+        ...makeBaseOptions(),
+        apiType: 'google',
+        timeout: 1,
+    });
+    engine.responder = async (_wfctx, _session, signal) => {
+        await new Promise(resolve => setTimeout(resolve, 10));
+        if (signal?.aborted)
+            throw new TypeError('timed out');
+        return 'late';
+    };
+    const e = await t.throwsAsync(engine.stateless(makeInferenceContext(), makeStringSession()));
+    t.true(e instanceof InferenceTimeout);
+});
+test('Engine stateful/append/push message semantics are correct', async (t) => {
+    const engine = new StubEngine({
+        ...makeBaseOptions(),
+        apiType: 'google',
+    });
+    engine.responder = async () => 'stateful-ok';
+    const session = makeStringSession();
+    const response = await engine.stateful(makeInferenceContext(), session);
+    t.is(response, 'stateful-ok');
+    t.deepEqual(session.chatMessages, ['stateful-ok']);
+    const user = makeUserMessage();
+    const appended = engine.appendUserMessage(session, user);
+    t.not(appended, session);
+    t.deepEqual(appended.chatMessages, ['stateful-ok', user]);
+    t.deepEqual(session.chatMessages, ['stateful-ok']);
+    const pushed = engine.pushUserMessage(session, user);
+    t.is(pushed, session);
+    t.deepEqual(session.chatMessages, ['stateful-ok', user]);
+});
+//# sourceMappingURL=test-engine.js.map

package/test/build/test-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-engine.js","sourceRoot":"","sources":["../src/test-engine.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,KAAK,CAAC;AACvB,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACvF,OAAO,EAAuB,MAAM,eAAe,CAAC;AACpD,OAAO,EAAyB,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAoB,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EACH,GAAG,EACH,eAAe,EACf,oBAAoB,EACpB,eAAe,GAClB,MAAM,mBAAmB,CAAC;AAI3B,MAAM,UAAW,SAAQ,MAMxB;IACsB,gBAAgB,GAAG,KAAK,CAAC;IACrC,SAAS,GAIO,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC;IAErB,KAAK,CACpB,KAAuB,EACvB,OAAoB,EACpB,MAAoB;QAEpB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAEe,iBAAiB,CAC7B,OAAoB,EACpB,OAA8B;QAE9B,OAAO;YACH,GAAG,OAAO;YACV,YAAY,EAAE,CAAC,GAAG,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC;SACnD,CAAC;IACN,CAAC;IAEe,eAAe,CAC3B,OAAoB,EACpB,OAA8B;QAE9B,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnC,OAAO,OAAO,CAAC;IACnB,CAAC;CACJ;AAED,SAAS,iBAAiB;IACtB,OAAO,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;AAChC,CAAC;AAGD,IAAI,CAAC,wCAAwC,EAAE,CAAC,CAAC,EAAE;IAC/C,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC;QAC1B,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,EAAC,CAAC,EAAC,EAAE;IACpE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC;QAC1B,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IAEH,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,MAAM,CAAC,SAAS,GAAG,KAAK,IAAI,EAAE;QAC1B,QAAQ,IAAI,CAAC,CAAC;QACd,IAAI,QAAQ,GAAG,CAAC;YAAE,MAAM,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC;QACzD,OAAO,UAAU,CAAC;IACtB,CAAC,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,oBAAoB,EAAE,EAAE,iBAAiB,EAAE,CAAC,CAAC;IAErF,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;AAC/B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mEAAmE,EAAE,KAAK,EAAC,CAAC,EAAC,EAAE;IAChF,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC;QAC1B,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,MAAM,CAAC,SAAS,GAAG,KAAK,IAAI,EAAE;QAC1B,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC,CAAC;IAEF,MAAM,CAAC,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACxH,CAAC,CAAC,IAAI,CAAC,CAAC,YAAY,YAAY,CAAC,CAAC;AACtC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6DAA6D,EAAE,KAAK,EAAC,CAAC,EAAC,EAAE;IAC1E,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC;QAC1B,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;QACjB,OAAO,EAAE,CAAC;KACb,CAAC,CAAC;IAEH,MAAM,CAAC,SAAS,GAAG,KAAK,EAAE,MAAwB,EAAE,QAAqB,EAAE,MAAoB,EAAE,EAAE;QAC/F,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACtD,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,SAAS,CAAC,WAAW,CAAC,CAAC;QACtD,OAAO,MAAM,CAAC;IAClB,CAAC,CAAC;IAEF,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,WAAW,CACzB,MAAM,CAAC,SAAS,CAAC,oBAAoB,EAAE,EAAE,iBAAiB,EAAE,CAAC,CAChE,CAAC;IACF,CAAC,CAAC,IAAI,CAAC,CAAC,YAAY,gBAAgB,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,KAAK,EAAC,CAAC,EAAC,EAAE;IACxE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC;QAC1B,GAAG,eAAe,EAAE;QACpB,OAAO,EAAE,QAAQ;KACpB,CAAC,CAAC;IACH,MAAM,CAAC,SAAS,GAAG,KAAK,IAAI,EAAE,CAAC,aAAa,CAAC;IAE7C,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;IACpC,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,oBAAoB,EAAE,EAAE,OAAO,CAAC,CAAC;IACxE,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC9B,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEnD,MAAM,IAAI,GAAG,eAAe,EAAE,CAAC;IAC/B,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IACzD,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACzB,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEnD,MAAM,MAAM,GAAG,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACtB,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC,CAAC;AAC7D,CAAC,CAAC,CAAC"}

package/test/build/test-helpers.d.ts

@@ -0,0 +1,46 @@
+import { type InferenceContext } from '#@/inference-context.ts';
+import { RoleMessage, type Session } from '#@/compatible/session.ts';
+import { Throttle } from '#@/throttle.ts';
+export declare const fdm: {
+    foo: {
+        description: string;
+        paraschema: import("@sinclair/typebox").TObject<{
+            x: import("@sinclair/typebox").TNumber;
+        }>;
+    };
+    bar: {
+        description: string;
+        paraschema: import("@sinclair/typebox").TObject<{
+            y: import("@sinclair/typebox").TString;
+        }>;
+    };
+};
+export type fdm = typeof fdm;
+export declare function makeBaseOptions(): {
+    baseUrl: string;
+    apiKey: string;
+    model: string;
+    name: string;
+    functionDeclarationMap: {
+        foo: {
+            description: string;
+            paraschema: import("@sinclair/typebox").TObject<{
+                x: import("@sinclair/typebox").TNumber;
+            }>;
+        };
+        bar: {
+            description: string;
+            paraschema: import("@sinclair/typebox").TObject<{
+                y: import("@sinclair/typebox").TString;
+            }>;
+        };
+    };
+    throttle: Throttle;
+    inputPrice: number;
+    outputPrice: number;
+    cachePrice: number;
+};
+export declare function makeSession(): Session<fdm>;
+export declare function makeUserMessage(): RoleMessage.User<fdm>;
+export declare function makeAiMessage(text?: string): RoleMessage.Ai<fdm>;
+export declare function makeInferenceContext(signal?: AbortSignal | null): InferenceContext;

package/test/build/test-helpers.js

@@ -0,0 +1,45 @@
+import { Type } from '@sinclair/typebox';
+import { Function } from '#@/function.ts';
+import {} from '#@/inference-context.ts';
+import { RoleMessage } from '#@/compatible/session.ts';
+import { Throttle } from '#@/throttle.ts';
+export const fdm = {
+    foo: {
+        description: 'foo tool',
+        paraschema: Type.Object({ x: Type.Number() }),
+    },
+    bar: {
+        description: 'bar tool',
+        paraschema: Type.Object({ y: Type.String() }),
+    },
+};
+export function makeBaseOptions() {
+    return {
+        baseUrl: 'https://example.com/v1',
+        apiKey: 'test-key',
+        model: 'test-model',
+        name: 'Test Endpoint',
+        functionDeclarationMap: fdm,
+        throttle: new Throttle(60),
+        inputPrice: 1,
+        outputPrice: 2,
+        cachePrice: 3,
+    };
+}
+export function makeSession() {
+    return { chatMessages: [] };
+}
+export function makeUserMessage() {
+    return RoleMessage.User.create([RoleMessage.Part.Text.create('hello')]);
+}
+export function makeAiMessage(text = 'ok') {
+    return RoleMessage.Ai.create([RoleMessage.Part.Text.create(text)]);
+}
+export function makeInferenceContext(signal = null) {
+    return {
+        busy: null,
+        signal,
+        cost() { },
+    };
+}
+//# sourceMappingURL=test-helpers.js.map

package/test/build/test-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-helpers.js","sourceRoot":"","sources":["../src/test-helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAyB,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG1C,MAAM,CAAC,MAAM,GAAG,GAAG;IACf,GAAG,EAAE;QACD,WAAW,EAAE,UAAU;QACvB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;KAChD;IACD,GAAG,EAAE;QACD,WAAW,EAAE,UAAU;QACvB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;KAChD;CAC+B,CAAC;AAIrC,MAAM,UAAU,eAAe;IAC3B,OAAO;QACH,OAAO,EAAE,wBAAwB;QACjC,MAAM,EAAE,UAAU;QAClB,KAAK,EAAE,YAAY;QACnB,IAAI,EAAE,eAAe;QACrB,sBAAsB,EAAE,GAAG;QAC3B,QAAQ,EAAE,IAAI,QAAQ,CAAC,EAAE,CAAC;QAC1B,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,CAAC;KAChB,CAAC;AACN,CAAC;AAED,MAAM,UAAU,WAAW;IACvB,OAAO,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,eAAe;IAC3B,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM,CAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACjF,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAI,GAAG,IAAI;IACrC,OAAO,WAAW,CAAC,EAAE,CAAC,MAAM,CAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,SAA6B,IAAI;IAClE,OAAO;QACH,IAAI,EAAE,IAAI;QACV,MAAM;QACN,IAAI,KAAI,CAAC;KACZ,CAAC;AACN,CAAC"}

package/test/build/test-native-engines.d.ts

@@ -0,0 +1 @@
+export {};