Skip to content

Commit d85190d

Browse files
authored
chore: pass files bytes for workspace API instead of base64 encoding (#75)
Signed-off-by: Donnie Adams <donnie@acorn.io>
1 parent d407717 commit d85190d

File tree

1 file changed

+11
-15
lines changed

1 file changed

+11
-15
lines changed

workspace.go

+11-15
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ package gptscript
22

33
import (
44
"context"
5-
"encoding/base64"
65
"strings"
76
)
87

@@ -52,8 +51,7 @@ func (g *GPTScript) ListFilesInWorkspace(ctx context.Context, workspaceID string
5251
return nil, err
5352
}
5453

55-
// The first line of the output is the workspace ID, ignore it.
56-
return strings.Split(strings.TrimSpace(out), "\n")[1:], nil
54+
return strings.Split(strings.TrimSpace(out), "\n"), nil
5755
}
5856

5957
func (g *GPTScript) RemoveAllWithPrefix(ctx context.Context, workspaceID, prefix string) error {
@@ -69,12 +67,11 @@ func (g *GPTScript) RemoveAllWithPrefix(ctx context.Context, workspaceID, prefix
6967

7068
func (g *GPTScript) WriteFileInWorkspace(ctx context.Context, workspaceID, filePath string, contents []byte) error {
7169
_, err := g.runBasicCommand(ctx, "workspaces/write-file", map[string]any{
72-
"id": workspaceID,
73-
"contents": base64.StdEncoding.EncodeToString(contents),
74-
"filePath": filePath,
75-
"workspaceTool": g.globalOpts.WorkspaceTool,
76-
"base64EncodedInput": true,
77-
"env": g.globalOpts.Env,
70+
"id": workspaceID,
71+
"contents": contents,
72+
"filePath": filePath,
73+
"workspaceTool": g.globalOpts.WorkspaceTool,
74+
"env": g.globalOpts.Env,
7875
})
7976

8077
return err
@@ -93,15 +90,14 @@ func (g *GPTScript) DeleteFileInWorkspace(ctx context.Context, workspaceID, file
9390

9491
func (g *GPTScript) ReadFileInWorkspace(ctx context.Context, workspaceID, filePath string) ([]byte, error) {
9592
out, err := g.runBasicCommand(ctx, "workspaces/read-file", map[string]any{
96-
"id": workspaceID,
97-
"filePath": filePath,
98-
"workspaceTool": g.globalOpts.WorkspaceTool,
99-
"base64EncodeOutput": true,
100-
"env": g.globalOpts.Env,
93+
"id": workspaceID,
94+
"filePath": filePath,
95+
"workspaceTool": g.globalOpts.WorkspaceTool,
96+
"env": g.globalOpts.Env,
10197
})
10298
if err != nil {
10399
return nil, err
104100
}
105101

106-
return base64.StdEncoding.DecodeString(out)
102+
return []byte(out), nil
107103
}

0 commit comments

Comments
 (0)