Skip to content

Commit 161d7eb

Browse files
committed
wip
Signed-off-by: David Gageot <david.gageot@docker.com>
1 parent cd2ccb2 commit 161d7eb

3 files changed

Lines changed: 8 additions & 6 deletions

File tree

pkg/chatgpt/auth.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import (
99
"crypto/rand"
1010
"encoding/hex"
1111
"encoding/json"
12+
"errors"
1213
"fmt"
1314
"io"
1415
"log/slog"
@@ -104,7 +105,7 @@ func Login(ctx context.Context) (*Token, error) {
104105

105106
// Verify state
106107
if r.URL.Query().Get("state") != state {
107-
resultCh <- callbackResult{err: fmt.Errorf("state mismatch")}
108+
resultCh <- callbackResult{err: errors.New("state mismatch")}
108109
http.Error(w, "State mismatch", http.StatusBadRequest)
109110
return
110111
}
@@ -119,7 +120,7 @@ func Login(ctx context.Context) (*Token, error) {
119120

120121
code := r.URL.Query().Get("code")
121122
if code == "" {
122-
resultCh <- callbackResult{err: fmt.Errorf("no authorization code received")}
123+
resultCh <- callbackResult{err: errors.New("no authorization code received")}
123124
http.Error(w, "No code received", http.StatusBadRequest)
124125
return
125126
}
@@ -206,7 +207,7 @@ func RefreshAccessToken(ctx context.Context, refreshToken string) (*Token, error
206207

207208
// Exchange the new id_token for an API key
208209
if refreshResp.IDToken == "" {
209-
return nil, fmt.Errorf("refresh response did not include an id_token")
210+
return nil, errors.New("refresh response did not include an id_token")
210211
}
211212

212213
apiKey, err := exchangeForAPIKey(ctx, refreshResp.IDToken)

pkg/chatgpt/provider.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package chatgpt
22

33
import (
44
"context"
5+
"errors"
56
"fmt"
67
"log/slog"
78
"sync"
@@ -61,15 +62,15 @@ func (p *Provider) resolveToken(ctx context.Context) (string, error) {
6162
return "", fmt.Errorf("failed to load ChatGPT token: %w", err)
6263
}
6364
if token == nil {
64-
return "", fmt.Errorf("not logged in to ChatGPT - run 'cagent login chatgpt' first")
65+
return "", errors.New("not logged in to ChatGPT - run 'cagent login chatgpt' first")
6566
}
6667
p.token = token
6768
}
6869

6970
// Refresh if expired
7071
if p.token.IsExpired() {
7172
if p.token.RefreshToken == "" {
72-
return "", fmt.Errorf("ChatGPT token expired - run 'cagent login chatgpt' to re-authenticate")
73+
return "", errors.New("ChatGPT token expired - run 'cagent login chatgpt' to re-authenticate")
7374
}
7475

7576
newToken, err := RefreshAccessToken(ctx, p.token.RefreshToken)

pkg/config/examples_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ func TestParseExamples(t *testing.T) {
5656
require.NotEmpty(t, model.Provider)
5757
require.NotEmpty(t, model.Model)
5858
// Skip providers that don't have entries in models.dev
59-
if model.Provider == "dmr" {
59+
if model.Provider == "dmr" || model.Provider == "chatgpt" {
6060
continue
6161
}
6262
// Skip models with routing rules - they use multiple providers

0 commit comments

Comments
 (0)