update for stacked contexts

Signed-off-by: Grant Linville <[email protected]>

Author: g-linville

credentials.go

@@ -20,8 +20,8 @@ type Credential struct {
 }
 
 type CredentialRequest struct {
-	Content     string `json:"content"`
-	AllContexts bool   `json:"allContexts"`
-	Context     string `json:"context"`
-	Name        string `json:"name"`
+	Content     string   `json:"content"`
+	AllContexts bool     `json:"allContexts"`
+	Context     []string `json:"context"`
+	Name        string   `json:"name"`
 }

gptscript.go

@@ -307,12 +307,12 @@ func (g *GPTScript) PromptResponse(ctx context.Context, resp PromptResponse) err
 	return err
 }
 
-func (g *GPTScript) ListCredentials(ctx context.Context, credCtx string, allContexts bool) ([]Credential, error) {
+func (g *GPTScript) ListCredentials(ctx context.Context, credCtxs []string, allContexts bool) ([]Credential, error) {
 	req := CredentialRequest{}
 	if allContexts {
 		req.AllContexts = true
 	} else {
-		req.Context = credCtx
+		req.Context = credCtxs
 	}
 
 	out, err := g.runBasicCommand(ctx, "credentials", req)
@@ -337,9 +337,9 @@ func (g *GPTScript) CreateCredential(ctx context.Context, cred Credential) error
 	return err
 }
 
-func (g *GPTScript) RevealCredential(ctx context.Context, credCtx, name string) (Credential, error) {
+func (g *GPTScript) RevealCredential(ctx context.Context, credCtxs []string, name string) (Credential, error) {
 	out, err := g.runBasicCommand(ctx, "credentials/reveal", CredentialRequest{
-		Context: credCtx,
+		Context: credCtxs,
 		Name:    name,
 	})
 	if err != nil {
@@ -355,7 +355,7 @@ func (g *GPTScript) RevealCredential(ctx context.Context, credCtx, name string)
 
 func (g *GPTScript) DeleteCredential(ctx context.Context, credCtx, name string) error {
 	_, err := g.runBasicCommand(ctx, "credentials/delete", CredentialRequest{
-		Context: credCtx,
+		Context: []string{credCtx}, // Only one context can be specified for delete operations
 		Name:    name,
 	})
 	return err

gptscript_test.go

@@ -12,7 +12,7 @@ import (
 	"testing"
 
 	"github.com/getkin/kin-openapi/openapi3"
-	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 var g *GPTScript
@@ -1467,21 +1467,21 @@ func TestCredentials(t *testing.T) {
 		Env:          map[string]string{"ENV": "testing"},
 		RefreshToken: "my-refresh-token",
 	})
-	assert.NoError(t, err)
+	require.NoError(t, err)
 
 	// List
-	creds, err := g.ListCredentials(context.Background(), "testing", false)
-	assert.NoError(t, err)
-	assert.GreaterOrEqual(t, len(creds), 1)
+	creds, err := g.ListCredentials(context.Background(), []string{"testing"}, false)
+	require.NoError(t, err)
+	require.GreaterOrEqual(t, len(creds), 1)
 
 	// Reveal
-	cred, err := g.RevealCredential(context.Background(), "testing", name)
-	assert.NoError(t, err)
-	assert.Contains(t, cred.Env, "ENV")
-	assert.Equal(t, cred.Env["ENV"], "testing")
-	assert.Equal(t, cred.RefreshToken, "my-refresh-token")
+	cred, err := g.RevealCredential(context.Background(), []string{"testing"}, name)
+	require.NoError(t, err)
+	require.Contains(t, cred.Env, "ENV")
+	require.Equal(t, cred.Env["ENV"], "testing")
+	require.Equal(t, cred.RefreshToken, "my-refresh-token")
 
 	// Delete
 	err = g.DeleteCredential(context.Background(), "testing", name)
-	assert.NoError(t, err)
+	require.NoError(t, err)
 }