diff options
Diffstat (limited to 'internal/cli')
| -rw-r--r-- | internal/cli/run.go | 3 | ||||
| -rw-r--r-- | internal/cli/serve.go | 3 |
2 files changed, 6 insertions, 0 deletions
diff --git a/internal/cli/run.go b/internal/cli/run.go index 3624cea..62e1252 100644 --- a/internal/cli/run.go +++ b/internal/cli/run.go @@ -86,6 +86,9 @@ func runTasks(file string, parallel int, dryRun bool) error { }, } pool := executor.NewPool(parallel, runners, store, logger) + if cfg.GeminiBinaryPath != "" { + pool.Classifier = &executor.Classifier{GeminiBinaryPath: cfg.GeminiBinaryPath} + } // Handle graceful shutdown. ctx, cancel := context.WithCancel(context.Background()) diff --git a/internal/cli/serve.go b/internal/cli/serve.go index 2ecb6cd..b679b38 100644 --- a/internal/cli/serve.go +++ b/internal/cli/serve.go @@ -71,6 +71,9 @@ func serve(addr string) error { } pool := executor.NewPool(cfg.MaxConcurrent, runners, store, logger) + if cfg.GeminiBinaryPath != "" { + pool.Classifier = &executor.Classifier{GeminiBinaryPath: cfg.GeminiBinaryPath} + } srv := api.NewServer(store, pool, logger, cfg.ClaudeBinaryPath, cfg.GeminiBinaryPath) srv.StartHub() |