| import assert from "node:assert/strict"; |
| import test from "node:test"; |
|
|
| const sdk = await import("../src/index.ts"); |
|
|
| test("host policy refuses unregistered tools and requires exact high-risk consent", () => { |
| const adapter = new sdk.KASIAdapter({ unlock: "high" }); |
| const first = { name: "unlock", arguments: { target: "lab" } }; |
| const second = { name: "unlock", arguments: { target: "vault" } }; |
| assert.equal(adapter.decide({ calls: [second], confidence: 0.9, risk: "low" }).action, "confirm"); |
| assert.equal(adapter.decide({ calls: [second], confidence: 0.9, risk: "low" }, new Set([sdk.callFingerprint(first)])).action, "confirm"); |
| assert.equal(adapter.decide({ calls: [second], confidence: 0.9, risk: "low" }, new Set([sdk.callFingerprint(second)])).action, "call"); |
| assert.equal(new sdk.KASIAdapter({}).decide({ calls: [{ name: "read", arguments: {} }], confidence: 0.9, risk: "low" }).action, "refuse"); |
| }); |
|
|
| test("request and response validators enforce the shared contract", () => { |
| assert.doesNotThrow(() => sdk.validateRequest({ |
| state: "ready", |
| questions: [{ id: "q", type: "choice", prompt: "Pick", options: ["a", "b"] }], |
| })); |
| assert.throws(() => sdk.validateResponse({ |
| answers: [{ question_id: "q", choice: "a", probabilities: [{ option: "a", probability: 0.5 }], confidence: 0.9, status: "ok" }], |
| backend: "direct", |
| model_id: "test", |
| })); |
| assert.throws(() => sdk.validateRequest({ |
| state: "ready", |
| questions: [{ id: "q", type: "choice", prompt: "Pick", options: ["a", "b"], rubric: ["bad"] }], |
| }), /only score question/); |
| assert.doesNotThrow(() => sdk.validateRequest({ |
| state: "x".repeat(2000), |
| questions: [{ id: "q", type: "choice", prompt: "Pick", options: ["a", "b"] }], |
| })); |
| assert.doesNotThrow(() => sdk.validateRequest({ |
| state: "x".repeat(200), |
| questions: Array.from({ length: 8 }, (_, index) => ({ |
| id: `q-${index}`, |
| type: "choice", |
| prompt: "Pick", |
| options: ["a", "b"], |
| })), |
| })); |
| assert.doesNotThrow(() => sdk.validateRequest({ |
| state: "ready", |
| questions: [{ id: "q", type: "choice", prompt: "Pick", options: ["x".repeat(2000), "b"] }], |
| })); |
| assert.throws(() => sdk.validateRequest({ |
| state: "ready", |
| questions: [{ id: "q", type: "choice", prompt: "Pick", options: ["", "b"] }], |
| }), /options must be strings/); |
| assert.doesNotThrow(() => sdk.validateRequest({ |
| state: "ready", |
| questions: [{ id: "q", type: "choice", prompt: "Pick", options: ["a", "b"] }], |
| }, 8, 512)); |
| }); |
|
|
| test("runtime selection is explicit", () => { |
| assert.equal(sdk.selectRuntime("wasm"), "wasm"); |
| assert.throws(() => sdk.selectRuntime("webgpu"), /WebGPU is unavailable/); |
| const previousGpu = globalThis.gpu; |
| globalThis.gpu = {}; |
| assert.equal(sdk.selectRuntime("webgpu"), "webgpu"); |
| if (previousGpu === undefined) delete globalThis.gpu; |
| else globalThis.gpu = previousGpu; |
| }); |
|
|
