diff --git a/internal/runtime/executor/aistudio_executor.go b/internal/runtime/executor/aistudio_executor.go index 53de71c8..de90c63a 100644 --- a/internal/runtime/executor/aistudio_executor.go +++ b/internal/runtime/executor/aistudio_executor.go @@ -192,6 +192,10 @@ func (e *AistudioExecutor) CountTokens(ctx context.Context, auth *cliproxyauth.A if err != nil { return cliproxyexecutor.Response{}, err } + + body.payload, _ = sjson.DeleteBytes(body.payload, "generationConfig") + body.payload, _ = sjson.DeleteBytes(body.payload, "tools") + endpoint := e.buildEndpoint(req.Model, "countTokens", "") wsReq := &wsrelay.HTTPRequest{ Method: http.MethodPost,