Mario Zechner
8a0903ebf2
The root barrel is now core-only and side-effect free: types, createModels/createProvider, auth substrate, lazyStream/lazyApi, faux, utils. Generated catalogs, api-registry, env-api-keys, images, global stream functions, and per-API lazy wrappers leave the root. New @earendil-works/pi-ai/compat preserves the old surface verbatim as a strict superset of the root: api-dispatch stream/complete with env key injection, the builtin registration side effect (skip-if-present so it cannot clobber earlier overrides), deprecated getModel/getModels/ getProviders aliases of the new getBuiltin* reads in providers/all, lazy api wrappers + setBedrockProviderModule, and image generation. Compat dies with the coding-agent ModelManager migration. Packaging: exports map gains ./compat, ./providers/*, ./api/*; sideEffects array lists only the effectful modules. Old-global imports across agent/coding-agent/examples and pi-ai tests switch to /compat (path-only; compat is a superset). The coding-agent extension loader resolves the pi-ai ROOT specifier to compat, so existing user extensions using the old global API keep working at runtime until compat is removed. vitest configs alias /compat to src; browser smoke imports old globals from /compat.
89 lines
2.8 KiB
TypeScript
89 lines
2.8 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { streamSimple } from "../src/compat.ts";
|
|
import type { AssistantMessage, Context, Model } from "../src/types.ts";
|
|
|
|
interface AnthropicPayload {
|
|
messages?: Array<{
|
|
role: string;
|
|
content: Array<{ type: string; text?: string; thinking?: string; signature?: string }>;
|
|
}>;
|
|
}
|
|
|
|
class PayloadCaptured extends Error {
|
|
constructor() {
|
|
super("payload captured");
|
|
this.name = "PayloadCaptured";
|
|
}
|
|
}
|
|
|
|
function makeModel(allowEmptySignature?: boolean): Model<"anthropic-messages"> {
|
|
return {
|
|
id: "mimo-v2.5-pro",
|
|
name: "MiMo-V2.5-Pro",
|
|
api: "anthropic-messages",
|
|
provider: "xiaomi-token-plan-ams",
|
|
baseUrl: "http://127.0.0.1:9/anthropic",
|
|
reasoning: true,
|
|
input: ["text"],
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
|
contextWindow: 1048576,
|
|
maxTokens: 1024,
|
|
...(allowEmptySignature === undefined ? {} : { compat: { allowEmptySignature } }),
|
|
};
|
|
}
|
|
|
|
function makeContext(thinkingSignature: string): Context {
|
|
const assistant: AssistantMessage = {
|
|
role: "assistant",
|
|
content: [{ type: "thinking", thinking: "internal reasoning", thinkingSignature }],
|
|
provider: "xiaomi-token-plan-ams",
|
|
api: "anthropic-messages",
|
|
model: "mimo-v2.5-pro",
|
|
timestamp: Date.now(),
|
|
usage: {
|
|
input: 0,
|
|
output: 0,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 0,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
},
|
|
stopReason: "stop",
|
|
};
|
|
return {
|
|
messages: [
|
|
{ role: "user", content: "first", timestamp: Date.now() },
|
|
assistant,
|
|
{ role: "user", content: "second", timestamp: Date.now() },
|
|
],
|
|
};
|
|
}
|
|
|
|
async function capturePayload(model: Model<"anthropic-messages">, context: Context): Promise<AnthropicPayload> {
|
|
let capturedPayload: AnthropicPayload | undefined;
|
|
const stream = streamSimple(model, context, {
|
|
apiKey: "fake-key",
|
|
onPayload: (payload) => {
|
|
capturedPayload = payload as AnthropicPayload;
|
|
throw new PayloadCaptured();
|
|
},
|
|
});
|
|
await stream.result();
|
|
if (!capturedPayload) throw new Error("Expected payload capture before request");
|
|
return capturedPayload;
|
|
}
|
|
|
|
describe("Anthropic empty thinking signature compat", () => {
|
|
it("converts empty-signature thinking to text by default", async () => {
|
|
const payload = await capturePayload(makeModel(), makeContext(""));
|
|
const assistant = payload.messages?.find((message) => message.role === "assistant");
|
|
expect(assistant?.content).toEqual([{ type: "text", text: "internal reasoning" }]);
|
|
});
|
|
|
|
it("preserves empty-signature thinking when allowEmptySignature is enabled", async () => {
|
|
const payload = await capturePayload(makeModel(true), makeContext(" "));
|
|
const assistant = payload.messages?.find((message) => message.role === "assistant");
|
|
expect(assistant?.content).toEqual([{ type: "thinking", thinking: "internal reasoning", signature: "" }]);
|
|
});
|
|
});
|