modelcontextprotocol/typescript-sdk
Watch
1
0
Fork
You've already forked typescript-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

add custom headers on initial _startOrAuth call (#318)

Browse source 
* add custom headers on initial _startOrAuth call

* update client/sse.ts: align commonHeaders w/ streamableHttp version

---------

Co-authored-by: Olivier Chafik <ochafik@anthropic.com>
This commit is contained in:
anthonjn 2025-07-07 11:54:44 -07:00 • committed by GitHub
commit 6ae4a5727c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 33 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/client/sse.ts
View file

@ -106,10 +106,8 @@ export class SSEClientTransport implements Transport {
return await this._startOrAuth();
}
private async _commonHeaders(): Promise<HeadersInit> {
const headers = {
...this._requestInit?.headers,
} as HeadersInit & Record<string, string>;
private async _commonHeaders(): Promise<Headers> {
const headers: HeadersInit = {};
if (this._authProvider) {
const tokens = await this._authProvider.tokens();
if (tokens) {
@ -120,24 +118,24 @@ export class SSEClientTransport implements Transport {
headers["mcp-protocol-version"] = this._protocolVersion;
}
return headers;
return new Headers(
{ ...headers, ...this._requestInit?.headers }
);
}
private _startOrAuth(): Promise<void> {
const fetchImpl = (this?._eventSourceInit?.fetch ?? this._fetch ?? fetch) as typeof fetch
const fetchImpl = (this?._eventSourceInit?.fetch ?? this._fetch ?? fetch) as typeof fetch
return new Promise((resolve, reject) => {
this._eventSource = new EventSource(
this._url.href,
{
...this._eventSourceInit,
fetch: async (url, init) => {
const headers = await this._commonHeaders()
const headers = await this._commonHeaders();
headers.set("Accept", "text/event-stream");
const response = await fetchImpl(url, {
...init,
headers: new Headers({
...headers,
Accept: "text/event-stream"
})
headers,
})
if (response.status === 401 && response.headers.has('www-authenticate')) {
@ -238,8 +236,7 @@ const fetchImpl = (this?._eventSourceInit?.fetch ?? this._fetch ?? fetch) as typ
}
try {
const commonHeaders = await this._commonHeaders();
const headers = new Headers(commonHeaders);
const headers = await this._commonHeaders();
headers.set("content-type", "application/json");
const init = {
...this._requestInit,