alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ssh): allow to customize action name

Browse Source 
Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
mudler
2025-03-20 19:15:12 +01:00
parent c3d3bba32a
commit 0513a327f6
1 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
services/actions/shell.go
View File

@@ -15,12 +15,16 @@ func NewShell(config map[string]string) *ShellAction {
privateKey: config["privateKey"],
user: config["user"],
host: config["host"],
customName: config["customName"],
customDescription: config["customDescription"],
}
}
type ShellAction struct {
privateKey string
user, host string
customName string
customDescription string
}
func (a *ShellAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
@@ -50,10 +54,18 @@ func (a *ShellAction) Run(ctx context.Context, params action.ActionParams) (acti
}
func (a *ShellAction) Definition() action.ActionDefinition {
name := "shell"
description := "Run a shell command on a remote server."
if a.customName != "" {
name = a.customName
}
if a.customDescription != "" {
description = a.customDescription
}
if a.host != "" && a.user != "" {
return action.ActionDefinition{
Name: "shell",
Description: "Run a shell command on a remote server.",
Name: action.ActionDefinitionName(name),
Description: description,
Properties: map[string]jsonschema.Definition{
"command": {
Type: jsonschema.String,
@@ -64,8 +76,8 @@ func (a *ShellAction) Definition() action.ActionDefinition {
}
}
return action.ActionDefinition{
Name: "shell",
Description: "Run a shell command on a remote server.",
Name: action.ActionDefinitionName(name),
Description: description,
Properties: map[string]jsonschema.Definition{
"command": {
Type: jsonschema.String,