shumengya/sproutclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(coding-agent): use ModelRegistry in custom-compaction example (fixes #1387)

Browse Source
This commit is contained in:
mom
2026-02-08 12:49:34 +00:00
parent d58d2c2da1
commit e10a831e56
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/coding-agent/examples/extensions/custom-compaction.ts
View File

@@ -13,7 +13,7 @@
* pi --extension examples/extensions/custom-compaction.ts * pi --extension examples/extensions/custom-compaction.ts
*/ */
import { complete, getModel } from "@mariozechner/pi-ai"; import { complete } from "@mariozechner/pi-ai";
import type { ExtensionAPI } from "@mariozechner/pi-coding-agent"; import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
import { convertToLlm, serializeConversation } from "@mariozechner/pi-coding-agent"; import { convertToLlm, serializeConversation } from "@mariozechner/pi-coding-agent";
@@ -25,7 +25,7 @@ export default function (pi: ExtensionAPI) {
const { messagesToSummarize, turnPrefixMessages, tokensBefore, firstKeptEntryId, previousSummary } = preparation; const { messagesToSummarize, turnPrefixMessages, tokensBefore, firstKeptEntryId, previousSummary } = preparation;
// Use Gemini Flash for summarization (cheaper/faster than most conversation models) // Use Gemini Flash for summarization (cheaper/faster than most conversation models)
const model = getModel("google", "gemini-2.5-flash"); const model = ctx.modelRegistry.find("google", "gemini-2.5-flash");
if (!model) { if (!model) {
ctx.ui.notify(`Could not find Gemini Flash model, using default compaction`, "warning"); ctx.ui.notify(`Could not find Gemini Flash model, using default compaction`, "warning");
return; return;