Skip to content

Commit 663b9b3

Browse files
committed
fix(executor): pass authID to relay instead of provider
1 parent 5dced4c commit 663b9b3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

internal/runtime/executor/aistudio_executor.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ func (e *AIStudioExecutor) Execute(ctx context.Context, auth *cliproxyauth.Auth,
7373
AuthValue: authValue,
7474
})
7575

76-
wsResp, err := e.relay.NonStream(ctx, e.provider, wsReq)
76+
wsResp, err := e.relay.NonStream(ctx, authID, wsReq)
7777
if err != nil {
7878
recordAPIResponseError(ctx, e.cfg, err)
7979
return resp, err
@@ -124,7 +124,7 @@ func (e *AIStudioExecutor) ExecuteStream(ctx context.Context, auth *cliproxyauth
124124
AuthType: authType,
125125
AuthValue: authValue,
126126
})
127-
wsStream, err := e.relay.Stream(ctx, e.provider, wsReq)
127+
wsStream, err := e.relay.Stream(ctx, authID, wsReq)
128128
if err != nil {
129129
recordAPIResponseError(ctx, e.cfg, err)
130130
return nil, err
@@ -221,7 +221,7 @@ func (e *AIStudioExecutor) CountTokens(ctx context.Context, auth *cliproxyauth.A
221221
AuthType: authType,
222222
AuthValue: authValue,
223223
})
224-
resp, err := e.relay.NonStream(ctx, e.provider, wsReq)
224+
resp, err := e.relay.NonStream(ctx, authID, wsReq)
225225
if err != nil {
226226
recordAPIResponseError(ctx, e.cfg, err)
227227
return cliproxyexecutor.Response{}, err

0 commit comments

Comments
 (0)