@@ -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