diff options
Diffstat (limited to 'internal/cli/run.go')
| -rw-r--r-- | internal/cli/run.go | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/internal/cli/run.go b/internal/cli/run.go index 49aa28e..2d7c3d7 100644 --- a/internal/cli/run.go +++ b/internal/cli/run.go @@ -84,9 +84,24 @@ func runTasks(file string, parallel int, dryRun bool) error { LogDir: cfg.LogDir, }, } + + localClient := buildLocalLLMClient(cfg.LocalModel, logger) + if localClient != nil { + runners["local"] = &executor.LocalRunner{ + Client: localClient, + Logger: logger, + LogDir: cfg.LogDir, + DefaultTemperature: cfg.LocalModel.DefaultTemperature, + } + } + pool := executor.NewPool(parallel, runners, store, logger) - if cfg.GeminiBinaryPath != "" { - pool.Classifier = &executor.Classifier{GeminiBinaryPath: cfg.GeminiBinaryPath} + pool.Classifier = &executor.Classifier{ + LLM: localClient, + GeminiBinaryPath: cfg.GeminiBinaryPath, + } + if localClient != nil { + pool.LLM = localClient } // Handle graceful shutdown. |
