Skip to content

Commit e59beb9

Browse files
committed
chore: make body size limit of payloads a default param
1 parent 2196262 commit e59beb9

File tree

2 files changed

+7
-14
lines changed

2 files changed

+7
-14
lines changed

src/McpResponse.ts

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ import {
1616
getFormattedRequestBody,
1717
getShortDescriptionForRequest,
1818
getStatusFromRequest,
19-
BODY_CONTEXT_SIZE_LIMIT,
2019
} from './formatters/networkFormatter.js';
2120
import {formatA11ySnapshot} from './formatters/snapshotFormatter.js';
2221
import type {McpContext} from './McpContext.js';
@@ -172,10 +171,7 @@ export class McpResponse implements Response {
172171
async processResponseBody(
173172
httpResponse: HTTPResponse,
174173
): Promise<string | null> {
175-
const formattedResponseData = await getFormattedResponseBody(
176-
httpResponse,
177-
BODY_CONTEXT_SIZE_LIMIT,
178-
);
174+
const formattedResponseData = await getFormattedResponseBody(httpResponse);
179175
if (formattedResponseData.length > 0) {
180176
return formattedResponseData;
181177
}
@@ -184,12 +180,9 @@ export class McpResponse implements Response {
184180
}
185181

186182
async processRequestBody(httpRequest: HTTPRequest): Promise<string | null> {
187-
const formattedResponseData = await getFormattedRequestBody(
188-
httpRequest,
189-
BODY_CONTEXT_SIZE_LIMIT,
190-
);
191-
if (formattedResponseData.length > 0) {
192-
return formattedResponseData;
183+
const formattedRequestData = await getFormattedRequestBody(httpRequest);
184+
if (formattedRequestData.length > 0) {
185+
return formattedRequestData;
193186
}
194187

195188
return null;

src/formatters/networkFormatter.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ import {isUtf8} from 'node:buffer';
88

99
import type {HTTPRequest, HTTPResponse} from 'puppeteer-core';
1010

11-
export const BODY_CONTEXT_SIZE_LIMIT = 10000;
11+
const BODY_CONTEXT_SIZE_LIMIT = 10000;
1212

1313
export function getShortDescriptionForRequest(request: HTTPRequest): string {
1414
return `${request.url()} ${request.method()} ${getStatusFromRequest(request)}`;
@@ -44,7 +44,7 @@ export function getFormattedHeaderValue(
4444

4545
export async function getFormattedResponseBody(
4646
httpResponse: HTTPResponse,
47-
sizeLimit: number,
47+
sizeLimit: number = BODY_CONTEXT_SIZE_LIMIT,
4848
): Promise<string> {
4949
try {
5050
const responseBuffer = await httpResponse.buffer();
@@ -68,7 +68,7 @@ export async function getFormattedResponseBody(
6868

6969
export async function getFormattedRequestBody(
7070
httpRequest: HTTPRequest,
71-
sizeLimit: number,
71+
sizeLimit: number = BODY_CONTEXT_SIZE_LIMIT,
7272
): Promise<string> {
7373
if (httpRequest.hasPostData()) {
7474
const data = httpRequest.postData();

0 commit comments

Comments
 (0)