mirror of
https://github.com/earendil-works/pi.git
synced 2026-06-18 15:54:04 +08:00
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.
54 lines
1.5 KiB
TypeScript
54 lines
1.5 KiB
TypeScript
/**
|
|
* Custom Model Selection
|
|
*
|
|
* Shows how to select a specific model and thinking level.
|
|
*/
|
|
|
|
import { getModel } from "@earendil-works/pi-ai/compat";
|
|
import { AuthStorage, createAgentSession, ModelRegistry } from "@earendil-works/pi-coding-agent";
|
|
|
|
// Set up auth storage and model registry
|
|
const authStorage = AuthStorage.create();
|
|
const modelRegistry = ModelRegistry.create(authStorage);
|
|
|
|
// Option 1: Find a specific built-in model by provider/id
|
|
const opus = getModel("anthropic", "claude-opus-4-5");
|
|
if (opus) {
|
|
console.log(`Found model: ${opus.provider}/${opus.id}`);
|
|
}
|
|
|
|
// Option 2: Find model via registry (includes custom models from models.json)
|
|
const customModel = modelRegistry.find("my-provider", "my-model");
|
|
if (customModel) {
|
|
console.log(`Found custom model: ${customModel.provider}/${customModel.id}`);
|
|
}
|
|
|
|
// Option 3: Pick from available models (have valid API keys)
|
|
const available = await modelRegistry.getAvailable();
|
|
console.log(
|
|
"Available models:",
|
|
available.map((m) => `${m.provider}/${m.id}`),
|
|
);
|
|
|
|
if (available.length > 0) {
|
|
const { session } = await createAgentSession({
|
|
model: available[0],
|
|
thinkingLevel: "medium", // off, low, medium, high
|
|
authStorage,
|
|
modelRegistry,
|
|
});
|
|
|
|
try {
|
|
session.subscribe((event) => {
|
|
if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
|
|
process.stdout.write(event.assistantMessageEvent.delta);
|
|
}
|
|
});
|
|
|
|
await session.prompt("Say hello in one sentence.");
|
|
console.log();
|
|
} finally {
|
|
session.dispose();
|
|
}
|
|
}
|