58 lines
2.1 KiB
TypeScript
58 lines
2.1 KiB
TypeScript
|
/**
|
||
|
|
* Claude Code provider catalog (src/agent/claude-code/provider.ts).
|
||
|
|
*
|
||
|
|
* Registered on a fresh pi model collection with a stub stream function, so
|
||
|
|
* only the catalog and auth surface are under test here.
|
||
|
|
*/
|
||
|
|
import assert from "node:assert/strict";
|
||
|
|
import test from "node:test";
|
||
|
|
|
||
|
|
import { clampThinkingLevel, createAssistantMessageEventStream, createModels, getSupportedThinkingLevels } from "@earendil-works/pi-ai";
|
||
|
|
|
||
|
|
import { CLAUDE_CODE_PROVIDER } from "../src/agent/claude-code/constants.ts";
|
||
|
|
import { createClaudeCodeProvider } from "../src/agent/claude-code/provider.ts";
|
||
|
|
|
||
|
|
function createCollection() {
|
||
|
|
const collection = createModels();
|
||
|
|
collection.setProvider(createClaudeCodeProvider(() => createAssistantMessageEventStream()));
|
||
|
|
return collection;
|
||
|
|
}
|
||
|
|
|
||
|
|
test("lists exactly the opus, sonnet and haiku aliases", () => {
|
||
|
|
const models = createCollection().getModels(CLAUDE_CODE_PROVIDER);
|
||
|
|
assert.deepEqual(
|
||
|
|
models.map((m) => m.id),
|
||
|
|
["opus", "sonnet", "haiku"],
|
||
|
|
);
|
||
|
|
assert.deepEqual(
|
||
|
|
models.map((m) => m.name),
|
||
|
|
["Claude Opus", "Claude Sonnet", "Claude Haiku"],
|
||
|
|
);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("supports low through max thinking and clamps off to low", () => {
|
||
|
|
for (const model of createCollection().getModels(CLAUDE_CODE_PROVIDER)) {
|
||
|
|
assert.deepEqual(getSupportedThinkingLevels(model), ["low", "medium", "high", "xhigh", "max"]);
|
||
|
|
assert.equal(clampThinkingLevel(model, "off"), "low");
|
||
|
|
assert.equal(clampThinkingLevel(model, "minimal"), "low");
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
test("accepts text and image input", () => {
|
||
|
|
for (const model of createCollection().getModels(CLAUDE_CODE_PROVIDER)) {
|
||
|
|
assert.deepEqual(model.input, ["text", "image"]);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
test("every cost field is zero", () => {
|
||
|
|
for (const model of createCollection().getModels(CLAUDE_CODE_PROVIDER)) {
|
||
|
|
assert.deepEqual(model.cost, { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 });
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
test("auth resolves as configured without any key", async () => {
|
||
|
|
const collection = createCollection();
|
||
|
|
assert.ok(await collection.checkAuth(CLAUDE_CODE_PROVIDER));
|
||
|
|
assert.equal((await collection.getAvailable(CLAUDE_CODE_PROVIDER)).length, 3);
|
||
|
|
});
|