feat: add subscription CLI provider catalog

opus, sonnet and haiku aliases, low..max thinking, zero cost, keyless auth. Streaming is injected.
2026-10-24 20:24:57 +04:00
parent 1df95fdfdd
commit 913b29efe5
3 changed files with 98 additions and 0 deletions
@@ -0,0 +1,57 @@
+/**
+ * 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);
+});