feat: add chat-ui option

The chat UI option is sugar to launch the chat-builder tool. If the user
enables the chat-ui option, then the appropriate setup is done to launch
the chat-builder UI tool for the user. If the a file is provided, then
the UI should launch to the "run" UI and the user can chat with the
provided tool. If no file is provided, then the "browse" UI is opened
and the user can decide what to do.

The most important secondary option is the --workspace option. If the
user provides a file, then the workspace will be set to the directory
where that file exists. However, if the user specifies their own
workspace, then it is not overridden.

Signed-off-by: Donnie Adams <[email protected]>

Author: thedadams

pkg/cli/gptscript.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"path/filepath"
 	"sort"
 	"strconv"
 	"strings"
@@ -17,6 +18,7 @@ import (
 	"github.com/gptscript-ai/gptscript/pkg/builtin"
 	"github.com/gptscript-ai/gptscript/pkg/cache"
 	"github.com/gptscript-ai/gptscript/pkg/chat"
+	"github.com/gptscript-ai/gptscript/pkg/env"
 	"github.com/gptscript-ai/gptscript/pkg/gptscript"
 	"github.com/gptscript-ai/gptscript/pkg/input"
 	"github.com/gptscript-ai/gptscript/pkg/loader"
@@ -65,6 +67,7 @@ type GPTScript struct {
 	ForceChat          bool   `usage:"Force an interactive chat session if even the top level tool is not a chat tool"`
 	ForceSequential    bool   `usage:"Force parallel calls to run sequentially"`
 	Workspace          string `usage:"Directory to use for the workspace, if specified it will not be deleted on exit"`
+	ChatUI             bool   `usage:"Launch the chat builder UI" hidden:"true" local:"true" name:"chat-ui"`
 
 	readData []byte
 }
@@ -314,6 +317,28 @@ func (r *GPTScript) PrintOutput(toolInput, toolOutput string) (err error) {
 }
 
 func (r *GPTScript) Run(cmd *cobra.Command, args []string) (retErr error) {
+	// If the user is trying to launch the chat-builder UI, then set up the tool and options here.
+	if r.ChatUI {
+		args = append([]string{env.VarOrDefault("GPTSCRIPT_CHAT_UI_TOOL", "github.com/gptscript-ai/ui@v2")}, args...)
+
+		// If args has more than one element, then the user has provided a file.
+		if len(args) > 1 {
+			if args[1] == "-" {
+				return fmt.Errorf("chat UI only supports files, cannot read from stdin")
+			}
+
+			// Only set the workspace if the user has not already done so.
+			if r.Workspace == "" {
+				r.Workspace = filepath.Dir(args[1])
+			}
+
+			args = append(append([]string{args[0]}, filepath.Base("--file="+args[0])), args[1:]...)
+		}
+
+		// The UI must run in daemon mode.
+		r.Daemon = true
+	}
+
 	gptOpt, err := r.NewGPTScriptOpts()
 	if err != nil {
 		return err

pkg/env/env.go

@@ -12,6 +12,14 @@ func execEquals(bin, check string) bool {
 		bin == check+".exe"
 }
 
+func VarOrDefault(key, defaultValue string) string {
+	if val := os.Getenv(key); val != "" {
+		return val
+	}
+
+	return defaultValue
+}
+
 func ToEnvLike(v string) string {
 	v = strings.ReplaceAll(v, ".", "_")
 	v = strings.ReplaceAll(v, "-", "_")