feat(core): allow agents to opt into a small model for lightweight turns - #46928
Open
everest-an wants to merge 1 commit into
Open
feat(core): allow agents to opt into a small model for lightweight turns#46928everest-an wants to merge 1 commit into
everest-an wants to merge 1 commit into
Conversation
Contributor
|
This PR doesn't fully meet our contributing guidelines and PR template. What needs to be fixed:
Please edit this PR description to address the above within 2 hours, or it will be automatically closed. If you believe this was flagged incorrectly, please let a maintainer know. |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Allow agents to use a small/fast model for lightweight turns
Problem
The SessionRunner resolves a single model per session (
SessionRunnerModel.resolve) and uses it for every provider turn. TheCatalog.model.small()helper already exists and is used for title generation, but agent turns never take advantage of it — so lightweight steps (status updates, confirmations, plain-text continuations) pay the full cost of the main model, which slows down long-running multi-step tasks.Proposal
Add an opt-in
smallflag to the agent schema (AgentV2.Info). When an agent setssmall: true, the runner resolves a small model via the existingCatalog.model.small(providerID)and prefers it for the turn, falling back to the session model when no small model is available.Catalog.model.small()(keyword + cost/age scoring) rather than inventing new model-selection logic.Changes
packages/schema/src/agent.ts: add optionalsmallboolean toAgentV2.Infopackages/core/src/session/runner/model.ts: addSessionRunnerModel.resolveSmallpackages/core/src/session/runner/llm.ts: prefer the small model whenagent.info?.smallis setpackages/core/test/session-runner-model.test.ts: unit tests forresolveSmallinjection + fallbackNotes