summaryrefslogtreecommitdiff
path: root/internal/cli
diff options
context:
space:
mode:
Diffstat (limited to 'internal/cli')
-rw-r--r--internal/cli/run.go6
-rw-r--r--internal/cli/serve.go6
2 files changed, 8 insertions, 4 deletions
diff --git a/internal/cli/run.go b/internal/cli/run.go
index 771c7b7..1c53b1a 100644
--- a/internal/cli/run.go
+++ b/internal/cli/run.go
@@ -8,6 +8,7 @@ import (
"syscall"
"github.com/thepeterstone/claudomator/internal/executor"
+ "github.com/thepeterstone/claudomator/internal/provider/openaicompat"
"github.com/thepeterstone/claudomator/internal/storage"
"github.com/thepeterstone/claudomator/internal/task"
"github.com/spf13/cobra"
@@ -106,11 +107,12 @@ func runTasks(file string, parallel int, dryRun bool) error {
localClient := buildLocalLLMClient(cfg.LocalModel, logger)
if localClient != nil && cfg.Runners.LocalEnabled() {
- runners["local"] = &executor.LocalRunner{
- Client: localClient,
+ runners["local"] = &executor.NativeRunner{
+ Provider: openaicompat.New(localClient),
Logger: logger,
LogDir: cfg.LogDir,
DefaultTemperature: cfg.LocalModel.DefaultTemperature,
+ DefaultModel: cfg.LocalModel.Model,
}
}
diff --git a/internal/cli/serve.go b/internal/cli/serve.go
index 19db2f8..d8ec6a1 100644
--- a/internal/cli/serve.go
+++ b/internal/cli/serve.go
@@ -15,6 +15,7 @@ import (
"github.com/thepeterstone/claudomator/internal/config"
"github.com/thepeterstone/claudomator/internal/executor"
"github.com/thepeterstone/claudomator/internal/notify"
+ "github.com/thepeterstone/claudomator/internal/provider/openaicompat"
"github.com/thepeterstone/claudomator/internal/storage"
"github.com/thepeterstone/claudomator/internal/version"
"github.com/spf13/cobra"
@@ -139,11 +140,12 @@ func serve(addr, basePath string) error {
localClient := buildLocalLLMClient(cfg.LocalModel, logger)
if localClient != nil && cfg.Runners.LocalEnabled() {
- runners["local"] = &executor.LocalRunner{
- Client: localClient,
+ runners["local"] = &executor.NativeRunner{
+ Provider: openaicompat.New(localClient),
Logger: logger,
LogDir: cfg.LogDir,
DefaultTemperature: cfg.LocalModel.DefaultTemperature,
+ DefaultModel: cfg.LocalModel.Model,
}
logger.Info("local runner registered", "endpoint", cfg.LocalModel.Endpoint, "model", cfg.LocalModel.Model)
}