diff --git a/pkg/cli/compile_compiler_setup_test.go b/pkg/cli/compile_compiler_setup_test.go index bdbafbd64ff..804b16b138c 100644 --- a/pkg/cli/compile_compiler_setup_test.go +++ b/pkg/cli/compile_compiler_setup_test.go @@ -3,21 +3,28 @@ package cli import ( + "reflect" "testing" "github.com/github/gh-aw/pkg/workflow" ) +func hasModelPricingResolver(compiler *workflow.Compiler) bool { + // Keep this field name in sync with workflow.Compiler; this helper intentionally + // inspects private state to preserve behavioral coverage without re-exporting API. + return !reflect.ValueOf(compiler).Elem().FieldByName("modelPricingResolver").IsNil() +} + func TestCreateAndConfigureCompiler_RegistersModelPricingResolverByDefault(t *testing.T) { compiler := createAndConfigureCompiler(CompileConfig{}) - if !compiler.HasModelPricingResolver() { + if !hasModelPricingResolver(compiler) { t.Fatal("expected model pricing resolver to be registered by default") } } func TestCreateAndConfigureCompiler_SkipsModelPricingResolverWhenDisabled(t *testing.T) { compiler := createAndConfigureCompiler(CompileConfig{DisableModelsDevLookup: true}) - if compiler.HasModelPricingResolver() { + if hasModelPricingResolver(compiler) { t.Fatal("expected model pricing resolver to be nil when models.dev lookup is disabled") } } diff --git a/pkg/workflow/compiler_types.go b/pkg/workflow/compiler_types.go index 2b56bb9352f..8a57eb3f565 100644 --- a/pkg/workflow/compiler_types.go +++ b/pkg/workflow/compiler_types.go @@ -178,11 +178,6 @@ func (c *Compiler) SetModelPricingResolver(fn func(ctx context.Context, provider c.modelPricingResolver = fn } -// HasModelPricingResolver reports whether a model pricing resolver callback is configured. -func (c *Compiler) HasModelPricingResolver() bool { - return c.modelPricingResolver != nil -} - // SetRequireDocker configures whether Docker must be available for container image validation. // When true, validation fails with an error if Docker is not installed or the daemon is not running. // When false (default), validation is silently skipped when Docker is unavailable.