mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
30a1d1849d
Entries of only +name/-name modify the inherited tool selection, so "defaultTools": ["+codemode"] enables codemode without repeating the defaults. Project entries layer on top of user settings. Document how to enable codemode without MCP and how to use classifier models from it.
61 lines
2 KiB
TypeScript
61 lines
2 KiB
TypeScript
/**
|
|
* Codemode, Tool Search, and MCP
|
|
*
|
|
* The CLI loads `codemode`, `tool_search`, and MCP as built-in extensions. SDK sessions add
|
|
* them to the resource loader's extension factories.
|
|
*
|
|
* `codemode` and `tool_search` are registered inactive. Enable them through the `defaultTools` setting
|
|
* (`tools` would also restrict the session to the named tools, which hides MCP tools), or let the
|
|
* MCP extension activate them: `codemode` for servers with `codemode` or `codemode-deferred` exposure,
|
|
* `tool_search` for servers with `deferred` exposure.
|
|
*/
|
|
|
|
import {
|
|
createAgentSession,
|
|
createCodemodeExtension,
|
|
createMcpExtension,
|
|
createToolSearchExtension,
|
|
DefaultResourceLoader,
|
|
getAgentDir,
|
|
SessionManager,
|
|
SettingsManager,
|
|
} from "@earendil-works/pi-coding-agent";
|
|
|
|
const cwd = process.cwd();
|
|
|
|
const resourceLoader = new DefaultResourceLoader({
|
|
cwd,
|
|
agentDir: getAgentDir(),
|
|
extensionFactories: [
|
|
createCodemodeExtension({ mode: "on" }),
|
|
createToolSearchExtension(),
|
|
// Reads mcp.json from the agent directory and the trusted project, like the CLI.
|
|
createMcpExtension(),
|
|
],
|
|
});
|
|
await resourceLoader.reload();
|
|
|
|
const settingsManager = SettingsManager.create(cwd);
|
|
// `+name` adds to the configured default tools instead of replacing them.
|
|
settingsManager.applyOverrides({ defaultTools: ["+codemode", "+tool_search"] });
|
|
|
|
const { session } = await createAgentSession({
|
|
resourceLoader,
|
|
settingsManager,
|
|
sessionManager: SessionManager.inMemory(),
|
|
});
|
|
|
|
try {
|
|
// Emits session_start, which connects the MCP servers in the background.
|
|
await session.bindExtensions({});
|
|
console.log("Active tools:", session.getActiveToolNames().join(", "));
|
|
session.subscribe((event) => {
|
|
if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
|
|
process.stdout.write(event.assistantMessageEvent.delta);
|
|
}
|
|
});
|
|
await session.prompt("Use codemode to count the TypeScript files in src/ and list the three largest.");
|
|
console.log();
|
|
} finally {
|
|
session.dispose();
|
|
}
|