Set working directory of tool calls to the directory of the file

Author: ibuildthecloud

pkg/cli/gptscript.go

@@ -37,7 +37,7 @@ type GPTScript struct {
 	OpenAIOptions
 	DisplayOptions
 	Debug         bool   `usage:"Enable debug logging"`
-	Quiet         *bool  `usage:"No output logging" short:"q"`
+	Quiet         *bool  `usage:"No output logging (set --quiet=false to force on even when there is no TTY)" short:"q"`
 	Output        string `usage:"Save output to a file, or - for stdout" short:"o"`
 	Input         string `usage:"Read input from a file (\"-\" for stdin)" short:"f"`
 	SubTool       string `usage:"Use tool of this name, not the first tool in file"`

pkg/engine/cmd.go

@@ -59,6 +59,9 @@ func (e *Engine) runCommand(ctx context.Context, tool types.Tool, input string)
 	cmd.Stdin = strings.NewReader(input)
 	cmd.Stderr = io.MultiWriter(all, os.Stderr)
 	cmd.Stdout = io.MultiWriter(all, output)
+	if tool.WorkingDir != "" {
+		cmd.Dir = tool.WorkingDir
+	}
 
 	if err := cmd.Run(); err != nil {
 		_, _ = os.Stderr.Write(output.Bytes())

pkg/engine/engine.go

@@ -230,8 +230,7 @@ func (e *Engine) complete(ctx context.Context, state *State) (*Return, error) {
 				ToolName: content.ToolCall.Function.Name,
 				Input:    content.ToolCall.Function.Arguments,
 			}
-		}
-		if content.Text != "" {
+		} else {
 			cp := content.Text
 			ret.Result = &cp
 		}

pkg/engine/http.go

@@ -7,6 +7,7 @@ import (
 	"io"
 	"net/http"
 	"net/url"
+	"os"
 	"strings"
 
 	"github.com/gptscript-ai/gptscript/pkg/types"
@@ -15,7 +16,19 @@ import (
 const DaemonURLSuffix = ".daemon.gpt.local"
 
 func (e *Engine) runHTTP(ctx context.Context, prg *types.Program, tool types.Tool, input string) (cmdRet *Return, cmdErr error) {
+	envMap := map[string]string{}
+	for _, env := range e.Env {
+		v, ok := strings.CutPrefix(env, "GPTSCRIPT_VAR_")
+		if ok {
+			k, v, _ := strings.Cut(v, "=")
+			envMap[k] = v
+		}
+	}
+
 	toolURL := strings.Split(tool.Instructions, "\n")[0][2:]
+	toolURL = os.Expand(toolURL, func(s string) string {
+		return envMap[s]
+	})
 
 	parsed, err := url.Parse(toolURL)
 	if err != nil {
@@ -51,6 +64,10 @@ func (e *Engine) runHTTP(ctx context.Context, prg *types.Program, tool types.Too
 
 	req.Header.Set("X-GPTScript-Tool-Name", tool.Parameters.Name)
 
+	for k, v := range envMap {
+		req.Header.Set("X-GPTScript-Var-"+k, v)
+	}
+
 	if err := json.Unmarshal([]byte(input), &map[string]any{}); err == nil {
 		req.Header.Set("Content-Type", "application/json")
 	} else {

pkg/loader/loader.go

@@ -200,6 +200,7 @@ func readTool(ctx context.Context, prg *types.Program, base *source, targetToolN
 	)
 
 	for i, tool := range tools {
+		tool.WorkingDir = base.Path
 		tool.Source.File = base.File
 
 		// Probably a better way to come up with an ID
@@ -234,7 +235,13 @@ var (
 	invalidChars  = regexp.MustCompile("[^a-zA-Z0-9_-]+")
 )
 
-func toolNormalizer(tool string) string {
+func ToolNormalizer(tool string) string {
+	parts := strings.Split(tool, "/")
+	tool = parts[len(parts)-1]
+	if strings.HasSuffix(tool, ".gpt") {
+		tool = strings.TrimSuffix(tool, filepath.Ext(tool))
+	}
+
 	if validToolName.MatchString(tool) {
 		return tool
 	}
@@ -251,22 +258,17 @@ func toolNormalizer(tool string) string {
 }
 
 func pickToolName(toolName string, existing map[string]struct{}) string {
-	newName, suffix, ok := strings.Cut(toolName, "/")
-	if ok {
-		newName = suffix
-	}
-	newName = strings.TrimSuffix(newName, filepath.Ext(newName))
-	if newName == "" {
-		newName = "external"
+	if toolName == "" {
+		toolName = "external"
 	}
 
 	for {
-		testName := toolNormalizer(newName)
+		testName := ToolNormalizer(toolName)
 		if _, ok := existing[testName]; !ok {
 			existing[testName] = struct{}{}
 			return testName
 		}
-		newName += "0"
+		toolName += "0"
 	}
 }
 

pkg/mvl/log.go

@@ -48,7 +48,10 @@ func Package() Logger {
 	_, p, _, _ := runtime.Caller(1)
 	_, suffix, _ := strings.Cut(p, "gptscript/")
 	i := strings.LastIndex(suffix, "/")
-	return New(suffix[:i])
+	if i > 0 {
+		return New(suffix[:i])
+	}
+	return New(p)
 }
 
 func New(name string) Logger {

pkg/runner/log.go

@@ -0,0 +1,5 @@
+package runner
+
+import "github.com/gptscript-ai/gptscript/pkg/mvl"
+
+var log = mvl.Package()

pkg/runner/runner.go

@@ -2,6 +2,7 @@ package runner
 
 import (
 	"context"
+	"os"
 	"sync"
 	"time"
 
@@ -20,30 +21,41 @@ type Monitor interface {
 }
 
 type Options struct {
+	WorkingDir     string
 	MonitorFactory MonitorFactory `usage:"-"`
 }
 
 func complete(opts ...Options) (result Options) {
 	for _, opt := range opts {
+		result.WorkingDir = types.FirstSet(opt.WorkingDir, result.WorkingDir)
 		result.MonitorFactory = types.FirstSet(opt.MonitorFactory, result.MonitorFactory)
 	}
 	if result.MonitorFactory == nil {
 		result.MonitorFactory = noopFactory{}
 	}
+	if result.WorkingDir == "" {
+		var err error
+		result.WorkingDir, err = os.Getwd()
+		if err != nil {
+			log.Fatalf("failed to determine current working directory: %v", err)
+		}
+	}
 	return
 }
 
 type Runner struct {
-	c       engine.Model
-	factory MonitorFactory
+	c          engine.Model
+	factory    MonitorFactory
+	workingDir string
 }
 
 func New(client engine.Model, opts ...Options) (*Runner, error) {
 	opt := complete(opts...)
 
 	return &Runner{
-		c:       client,
-		factory: opt.MonitorFactory,
+		c:          client,
+		factory:    opt.MonitorFactory,
+		workingDir: opt.WorkingDir,
 	}, nil
 }
 
@@ -94,6 +106,10 @@ func (r *Runner) call(callCtx engine.Context, monitor Monitor, env []string, inp
 		Env:      env,
 	}
 
+	if r.workingDir != "" {
+		e.Env = append(e.Env, "GPTSCRIPT_VAR_WORKDIR="+r.workingDir)
+	}
+
 	monitor.Event(Event{
 		Time:        time.Now(),
 		CallContext: &callCtx,