Merge pull request #428 from thedadams/required-extensions

thedadams · web-flow · commit 4b61f72139a5 · 2024-06-04T10:33:05.000-04:00
fix: add ability to set required extensions for certain commands
diff --git a/pkg/engine/cmd.go b/pkg/engine/cmd.go
@@ -20,6 +20,11 @@ import (
 	"github.com/gptscript-ai/gptscript/pkg/version"
 )
 
+var requiredFileExtensions = map[string]string{
+	"powershell.exe": "*.ps1",
+	"powershell":     "*.ps1",
+}
+
 func (e *Engine) runCommand(ctx Context, tool types.Tool, input string, toolCategory ToolCategory) (cmdOut string, cmdErr error) {
 	id := counter.Next()
 
@@ -198,7 +203,7 @@ func (e *Engine) newCommand(ctx context.Context, extraEnv []string, tool types.T
 	)
 
 	if strings.TrimSpace(rest) != "" {
-		f, err := os.CreateTemp("", version.ProgramName)
+		f, err := os.CreateTemp("", version.ProgramName+requiredFileExtensions[args[0]])
 		if err != nil {
 			return nil, nil, err
 		}

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,11 @@ import (`
`20`	`20`	`"github.com/gptscript-ai/gptscript/pkg/version"`
`21`	`21`	`)`
`22`	`22`
	`23`	`+var requiredFileExtensions = map[string]string{`
	`24`	`+ "powershell.exe": "*.ps1",`
	`25`	`+ "powershell": "*.ps1",`
	`26`	`+}`
	`27`	`+`
`23`	`28`	`func (e *Engine) runCommand(ctx Context, tool types.Tool, input string, toolCategory ToolCategory) (cmdOut string, cmdErr error) {`
`24`	`29`	`id := counter.Next()`
`25`	`30`
`@@ -198,7 +203,7 @@ func (e *Engine) newCommand(ctx context.Context, extraEnv []string, tool types.T`
`198`	`203`	`)`
`199`	`204`
`200`	`205`	`if strings.TrimSpace(rest) != "" {`
`201`		`- f, err := os.CreateTemp("", version.ProgramName)`
	`206`	`+ f, err := os.CreateTemp("", version.ProgramName+requiredFileExtensions[args[0]])`
`202`	`207`	`if err != nil {`
`203`	`208`	`return nil, nil, err`
`204`	`209`	`}`