diff --git a/src/services/api/streaming.ts b/src/services/api/streaming.ts index 1b78247bf6..08dc2ba636 100644 --- a/src/services/api/streaming.ts +++ b/src/services/api/streaming.ts @@ -28,6 +28,12 @@ export interface StreamQueryOptions { } export class StreamingAPI extends BaseYdbAPI { + private csrfToken?: string; + + setCSRFToken = (token: string) => { + this.csrfToken = token; + }; + async streamQuery( params: StreamQueryParams, options: StreamQueryOptions, @@ -48,6 +54,10 @@ export class StreamingAPI extends BaseYdbAPI { 'Content-Type': 'application/json', }); + if (this.csrfToken) { + headers.set('X-CSRF-Token', this.csrfToken); + } + if (params.tracingLevel) { headers.set('X-Trace-Verbosity', String(params.tracingLevel)); }