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.
79 lines
2.6 KiB
TypeScript
79 lines
2.6 KiB
TypeScript
import { afterEach, describe, expect, it } from "vitest";
|
|
import { getModel } from "../src/compat.ts";
|
|
import { findEnvKeys, getEnvApiKey } from "../src/env-api-keys.ts";
|
|
|
|
const originalTogetherApiKey = process.env.TOGETHER_API_KEY;
|
|
|
|
afterEach(() => {
|
|
if (originalTogetherApiKey === undefined) {
|
|
delete process.env.TOGETHER_API_KEY;
|
|
} else {
|
|
process.env.TOGETHER_API_KEY = originalTogetherApiKey;
|
|
}
|
|
});
|
|
|
|
describe("Together models", () => {
|
|
it("registers the default Kimi K2.6 model via OpenAI-compatible Chat Completions API", () => {
|
|
const model = getModel("together", "moonshotai/Kimi-K2.6");
|
|
|
|
expect(model).toBeDefined();
|
|
expect(model.api).toBe("openai-completions");
|
|
expect(model.provider).toBe("together");
|
|
expect(model.baseUrl).toBe("https://api.together.ai/v1");
|
|
expect(model.reasoning).toBe(true);
|
|
expect(model.thinkingLevelMap).toEqual({ minimal: null, low: null, medium: null });
|
|
expect(model.input).toEqual(["text", "image"]);
|
|
expect(model.contextWindow).toBe(262144);
|
|
expect(model.maxTokens).toBe(131000);
|
|
expect(model.cost).toEqual({
|
|
input: 1.2,
|
|
output: 4.5,
|
|
cacheRead: 0.2,
|
|
cacheWrite: 0,
|
|
});
|
|
expect(model.compat).toEqual({
|
|
supportsStore: false,
|
|
supportsDeveloperRole: false,
|
|
supportsReasoningEffort: false,
|
|
maxTokensField: "max_tokens",
|
|
thinkingFormat: "together",
|
|
supportsStrictMode: false,
|
|
supportsLongCacheRetention: false,
|
|
});
|
|
});
|
|
|
|
it("models Together reasoning controls from the Together API surface", () => {
|
|
const gptOss = getModel("together", "openai/gpt-oss-120b");
|
|
expect(gptOss.thinkingLevelMap).toEqual({ off: null, minimal: null });
|
|
expect(gptOss.compat).toMatchObject({
|
|
supportsReasoningEffort: true,
|
|
thinkingFormat: "openai",
|
|
});
|
|
|
|
const deepSeekV4 = getModel("together", "deepseek-ai/DeepSeek-V4-Pro");
|
|
expect(deepSeekV4.thinkingLevelMap).toEqual({
|
|
minimal: null,
|
|
low: null,
|
|
medium: null,
|
|
high: "high",
|
|
xhigh: null,
|
|
});
|
|
expect(deepSeekV4.compat).toMatchObject({
|
|
supportsReasoningEffort: true,
|
|
thinkingFormat: "together",
|
|
});
|
|
|
|
const minimax = getModel("together", "MiniMaxAI/MiniMax-M2.7");
|
|
expect(minimax.thinkingLevelMap).toEqual({ off: null, minimal: null, low: null, medium: null });
|
|
expect(minimax.compat?.thinkingFormat).toBeUndefined();
|
|
expect(minimax.compat?.supportsReasoningEffort).toBe(false);
|
|
});
|
|
|
|
it("resolves TOGETHER_API_KEY from the environment", () => {
|
|
process.env.TOGETHER_API_KEY = "test-together-key";
|
|
|
|
expect(findEnvKeys("together")).toEqual(["TOGETHER_API_KEY"]);
|
|
expect(getEnvApiKey("together")).toBe("test-together-key");
|
|
});
|
|
});
|