@@ -2,7 +2,6 @@ package gptscript
2
2
3
3
import (
4
4
"context"
5
- "encoding/base64"
6
5
"strings"
7
6
)
8
7
@@ -52,8 +51,7 @@ func (g *GPTScript) ListFilesInWorkspace(ctx context.Context, workspaceID string
52
51
return nil , err
53
52
}
54
53
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
57
55
}
58
56
59
57
func (g * GPTScript ) RemoveAllWithPrefix (ctx context.Context , workspaceID , prefix string ) error {
@@ -69,12 +67,11 @@ func (g *GPTScript) RemoveAllWithPrefix(ctx context.Context, workspaceID, prefix
69
67
70
68
func (g * GPTScript ) WriteFileInWorkspace (ctx context.Context , workspaceID , filePath string , contents []byte ) error {
71
69
_ , 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 ,
78
75
})
79
76
80
77
return err
@@ -93,15 +90,14 @@ func (g *GPTScript) DeleteFileInWorkspace(ctx context.Context, workspaceID, file
93
90
94
91
func (g * GPTScript ) ReadFileInWorkspace (ctx context.Context , workspaceID , filePath string ) ([]byte , error ) {
95
92
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 ,
101
97
})
102
98
if err != nil {
103
99
return nil , err
104
100
}
105
101
106
- return base64 . StdEncoding . DecodeString (out )
102
+ return [] byte (out ), nil
107
103
}
0 commit comments