Merge pull request #16 from thedadams/workspace-support

feat: add support for workspace gptscript feature

Author: thedadams

.gitignore

@@ -24,3 +24,5 @@ go.work
 .idea/
 .vscode/
 
+bin/
+workspace/

exec.go

@@ -18,6 +18,7 @@ type Opts struct {
 	Quiet        bool   `json:"quiet"`
 	Chdir        string `json:"chdir"`
 	SubTool      string `json:"subTool"`
+	Workspace    string `json:"workspace"`
 }
 
 func (o Opts) toArgs() []string {
@@ -34,6 +35,9 @@ func (o Opts) toArgs() []string {
 	if o.SubTool != "" {
 		args = append(args, "--sub-tool="+o.SubTool)
 	}
+	if o.Workspace != "" {
+		args = append(args, "--workspace="+o.Workspace)
+	}
 	return append(args, "--quiet="+fmt.Sprint(o.Quiet))
 }
 

exec_test.go

@@ -500,3 +500,19 @@ echo hello there
 		t.Errorf("Unexpected output: %s", out)
 	}
 }
+
+func TestExecWithWorkspace(t *testing.T) {
+	tool := &SimpleTool{
+		Tools:        []string{"sys.workspace.ls", "sys.workspace.write"},
+		Instructions: "Write a file named 'hello.txt' to the workspace with the content 'Hello!' and list the contents of the workspace.",
+	}
+
+	out, err := ExecTool(context.Background(), Opts{Workspace: "./workspace"}, tool)
+	if err != nil {
+		t.Errorf("Error executing tool: %v", err)
+	}
+
+	if !strings.Contains(out, "hello.txt") {
+		t.Errorf("Unexpected output: %s", out)
+	}
+}