From c3d3bba32abdc77eb2c6a3d04a6be612a3c92223 Mon Sep 17 00:00:00 2001
From: mudler <mudler@localai.io>
Date: Thu, 20 Mar 2025 19:13:37 +0100
Subject: [PATCH] feat(ssh): allow to specify a fixed host/user to run commands

Signed-off-by: mudler <mudler@localai.io>
---
 services/actions/shell.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/services/actions/shell.go b/services/actions/shell.go
index aa24a83..3460460 100644
--- a/services/actions/shell.go
+++ b/services/actions/shell.go
@@ -13,11 +13,14 @@ import (
 func NewShell(config map[string]string) *ShellAction {
 	return &ShellAction{
 		privateKey: config["privateKey"],
+		user:       config["user"],
+		host:       config["host"],
 	}
 }
 
 type ShellAction struct {
 	privateKey string
+	user, host string
 }
 
 func (a *ShellAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
@@ -33,6 +36,11 @@ func (a *ShellAction) Run(ctx context.Context, params action.ActionParams) (acti
 		return action.ActionResult{}, err
 	}
 
+	if a.host != "" && a.user != "" {
+		result.Host = a.host
+		result.User = a.user
+	}
+
 	output, err := sshCommand(a.privateKey, result.Command, result.User, result.Host)
 	if err != nil {
 		return action.ActionResult{}, err
@@ -42,6 +50,19 @@ func (a *ShellAction) Run(ctx context.Context, params action.ActionParams) (acti
 }
 
 func (a *ShellAction) Definition() action.ActionDefinition {
+	if a.host != "" && a.user != "" {
+		return action.ActionDefinition{
+			Name:        "shell",
+			Description: "Run a shell command on a remote server.",
+			Properties: map[string]jsonschema.Definition{
+				"command": {
+					Type:        jsonschema.String,
+					Description: "The command to run on the remote server.",
+				},
+			},
+			Required: []string{"command"},
+		}
+	}
 	return action.ActionDefinition{
 		Name:        "shell",
 		Description: "Run a shell command on a remote server.",