@@ -46,6 +46,7 @@ export const resolveAgentModelFallbacksOverrideMock = createMock();
4646export const resolveAgentSkillsFilterMock = createMock ( ) ;
4747export const getModelRefStatusMock = createMock ( ) ;
4848export const isCliProviderMock = createMock ( ) ;
49+ export const resolveCliRuntimeExecutionProviderMock = createMock ( ) ;
4950export const resolveAllowedModelRefMock = createMock ( ) ;
5051export const resolveConfiguredModelRefMock = createMock ( ) ;
5152export const resolveHooksGmailModelMock = createMock ( ) ;
@@ -131,6 +132,13 @@ vi.mock("./run.runtime.js", () => ({
131132 getRemoteSkillEligibility : getRemoteSkillEligibilityMock ,
132133} ) ) ;
133134
135+ vi . mock ( "../../agents/model-runtime-aliases.js" , async ( ) => ( {
136+ ...( await vi . importActual < typeof import ( "../../agents/model-runtime-aliases.js" ) > (
137+ "../../agents/model-runtime-aliases.js" ,
138+ ) ) ,
139+ resolveCliRuntimeExecutionProvider : resolveCliRuntimeExecutionProviderMock ,
140+ } ) ) ;
141+
134142vi . mock ( "./run-external-content.runtime.js" , ( ) => ( {
135143 buildSafeExternalPrompt : buildSafeExternalPromptMock ,
136144 detectSuspiciousPatterns : detectSuspiciousPatternsMock ,
@@ -296,7 +304,7 @@ function makeDefaultModelFallbackResult() {
296304 return {
297305 result : {
298306 payloads : [ { text : "test output" } ] ,
299- meta : { agentMeta : { usage : { input : 10 , output : 20 } } } ,
307+ meta : { agentMeta : { } } ,
300308 } ,
301309 provider : "openai" ,
302310 model : "gpt-5.4" ,
@@ -306,7 +314,7 @@ function makeDefaultModelFallbackResult() {
306314function makeDefaultEmbeddedResult ( ) {
307315 return {
308316 payloads : [ { text : "test output" } ] ,
309- meta : { agentMeta : { usage : { input : 10 , output : 20 } } } ,
317+ meta : { agentMeta : { } } ,
310318 } ;
311319}
312320
@@ -379,6 +387,7 @@ function resetRunConfigMocks(): void {
379387 resolveAgentModelFallbacksOverrideMock . mockReturnValue ( undefined ) ;
380388 resolveAgentSkillsFilterMock . mockReturnValue ( undefined ) ;
381389 resolveConfiguredModelRefMock . mockReturnValue ( { provider : "openai" , model : "gpt-5.4" } ) ;
390+ resolveCliRuntimeExecutionProviderMock . mockReturnValue ( undefined ) ;
382391 resolveAllowedModelRefMock . mockReturnValue ( { ref : { provider : "openai" , model : "gpt-5.4" } } ) ;
383392 resolveHooksGmailModelMock . mockReturnValue ( null ) ;
384393 resolveThinkingDefaultMock . mockReturnValue ( "off" ) ;
@@ -389,7 +398,11 @@ function resetRunConfigMocks(): void {
389398 } ) ;
390399 resolveAgentTimeoutMsMock . mockReturnValue ( 60_000 ) ;
391400 deriveSessionTotalTokensMock . mockReturnValue ( 30 ) ;
392- hasNonzeroUsageMock . mockReturnValue ( true ) ;
401+ hasNonzeroUsageMock . mockImplementation (
402+ ( usage : { input ?: unknown ; output ?: unknown } | undefined ) =>
403+ ( typeof usage ?. input === "number" && usage . input > 0 ) ||
404+ ( typeof usage ?. output === "number" && usage . output > 0 ) ,
405+ ) ;
393406 ensureAgentWorkspaceMock . mockResolvedValue ( { dir : "/tmp/workspace" } ) ;
394407 normalizeThinkLevelMock . mockImplementation ( ( value : unknown ) => value ) ;
395408 isThinkingLevelSupportedMock . mockReturnValue ( true ) ;
0 commit comments