Allow reading script from stdin using "gptscript -"

Author: ibuildthecloud

pkg/cli/gptscript.go

@@ -18,6 +18,7 @@ import (
 	"github.com/gptscript-ai/gptscript/pkg/openai"
 	"github.com/gptscript-ai/gptscript/pkg/runner"
 	"github.com/gptscript-ai/gptscript/pkg/server"
+	"github.com/gptscript-ai/gptscript/pkg/types"
 	"github.com/gptscript-ai/gptscript/pkg/version"
 	"github.com/spf13/cobra"
 	"golang.org/x/term"
@@ -125,9 +126,25 @@ func (r *GPTScript) Run(cmd *cobra.Command, args []string) error {
 		return cmd.Help()
 	}
 
-	prg, err := loader.Program(cmd.Context(), args[0], r.SubTool)
-	if err != nil {
-		return err
+	var (
+		prg types.Program
+		err error
+	)
+
+	if args[0] == "-" {
+		data, err := io.ReadAll(os.Stdin)
+		if err != nil {
+			return err
+		}
+		prg, err = loader.ProgramFromSource(cmd.Context(), string(data), r.SubTool)
+		if err != nil {
+			return err
+		}
+	} else {
+		prg, err = loader.Program(cmd.Context(), args[0], r.SubTool)
+		if err != nil {
+			return err
+		}
 	}
 
 	if r.Assemble {

pkg/loader/loader.go

@@ -342,6 +342,21 @@ func link(ctx context.Context, prg *types.Program, base *source, tool types.Tool
 	return tool, nil
 }
 
+func ProgramFromSource(ctx context.Context, content, subToolName string) (types.Program, error) {
+	prg := types.Program{
+		ToolSet: types.ToolSet{},
+	}
+	tool, err := readTool(ctx, &prg, &source{
+		Content: io.NopCloser(strings.NewReader(content)),
+		File:    "inline",
+	}, subToolName)
+	if err != nil {
+		return types.Program{}, err
+	}
+	prg.EntryToolID = tool.ID
+	return prg, nil
+}
+
 func Program(ctx context.Context, name, subToolName string) (types.Program, error) {
 	prg := types.Program{
 		ToolSet: types.ToolSet{},