summaryrefslogtreecommitdiff
path: root/internal/hexaiaction/run.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/hexaiaction/run.go')
-rw-r--r--internal/hexaiaction/run.go34
1 files changed, 32 insertions, 2 deletions
diff --git a/internal/hexaiaction/run.go b/internal/hexaiaction/run.go
index 84cb9b1..f34a4cd 100644
--- a/internal/hexaiaction/run.go
+++ b/internal/hexaiaction/run.go
@@ -64,10 +64,18 @@ type actionClient interface {
type actionClientFactory func(cfg appconfig.App) (actionClient, error)
+type actionConfigLoader func(context.Context, *log.Logger) appconfig.App
+
+type actionStatusSink interface {
+ SetLLMStart(provider, model string) error
+}
+
// Runner executes action requests with injectable dependencies for testability.
type Runner struct {
chooseAction actionChooser
newClient actionClientFactory
+ loadConfig actionConfigLoader
+ statusSink actionStatusSink
}
// NewRunner builds a Runner with production dependencies.
@@ -75,6 +83,8 @@ func NewRunner() *Runner {
return &Runner{
chooseAction: chooseActionFromConfig,
newClient: defaultActionClientFactory,
+ loadConfig: loadActionConfig,
+ statusSink: tmuxActionStatusSink{},
}
}
@@ -91,6 +101,16 @@ func defaultActionClientFactory(cfg appconfig.App) (actionClient, error) {
return llmutils.NewClientFromApp(cfg)
}
+type tmuxActionStatusSink struct{}
+
+func (tmuxActionStatusSink) SetLLMStart(provider, model string) error {
+ return tmux.SetStatus(tmux.FormatLLMStartStatus(provider, model))
+}
+
+func loadActionConfig(ctx context.Context, logger *log.Logger) appconfig.App {
+ return appconfig.LoadWithOptions(logger, appconfig.LoadOptions{ConfigPath: configPathFromContext(ctx)})
+}
+
type actionPlan struct {
fallback string
run func(context.Context) (string, error)
@@ -127,6 +147,8 @@ func Run(ctx context.Context, stdin io.Reader, stdout, stderr io.Writer) error {
func (r *Runner) Run(ctx context.Context, stdin io.Reader, stdout, stderr io.Writer) error {
chooser := chooseActionFromConfig
newClient := defaultActionClientFactory
+ loadConfig := loadActionConfig
+ statusSink := actionStatusSink(tmuxActionStatusSink{})
if r != nil {
if r.chooseAction != nil {
chooser = r.chooseAction
@@ -134,10 +156,16 @@ func (r *Runner) Run(ctx context.Context, stdin io.Reader, stdout, stderr io.Wri
if r.newClient != nil {
newClient = r.newClient
}
+ if r.loadConfig != nil {
+ loadConfig = r.loadConfig
+ }
+ if r.statusSink != nil {
+ statusSink = r.statusSink
+ }
}
logger := log.New(stderr, "hexai-tmux-action ", log.LstdFlags|log.Lmsgprefix)
- cfg := appconfig.LoadWithOptions(logger, appconfig.LoadOptions{ConfigPath: configPathFromContext(ctx)})
+ cfg := loadConfig(ctx, logger)
if cfg.StatsWindowMinutes > 0 {
stats.SetWindow(time.Duration(cfg.StatsWindowMinutes) * time.Minute)
}
@@ -159,7 +187,9 @@ func (r *Runner) Run(ctx context.Context, stdin io.Reader, stdout, stderr io.Wri
if primaryModel == "" {
primaryModel = cli.DefaultModel()
}
- _ = tmux.SetStatus(tmux.FormatLLMStartStatus(cli.Name(), primaryModel))
+ if statusSink != nil {
+ _ = statusSink.SetLLMStart(cli.Name(), primaryModel)
+ }
var client chatDoer = cli
parts, err := ParseInput(stdin)
if err != nil {