Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 2 additions & 10 deletions src/everything/prompts/resource.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import {
RESOURCE_TYPE_BLOB,
RESOURCE_TYPE_TEXT,
RESOURCE_TYPES,
validateResourceId,
} from "../resources/templates.js";

/**
Expand Down Expand Up @@ -49,16 +50,7 @@ export const registerEmbeddedResourcePrompt = (server: McpServer) => {
}

// Validate resourceId argument
const resourceId = Number(args?.resourceId);
if (
!Number.isFinite(resourceId) ||
!Number.isInteger(resourceId) ||
resourceId < 1
) {
throw new Error(
`Invalid resourceId: ${args?.resourceId}. Must be a finite positive integer.`
);
}
const resourceId = validateResourceId(args?.resourceId);

// Get resource based on the resource type
const uri =
Expand Down
26 changes: 21 additions & 5 deletions src/everything/resources/templates.ts
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,24 @@ export const resourceIdForResourceTemplateCompleter: CompleteResourceTemplateCal
return Number.isInteger(resourceId) && resourceId > 0 ? [value] : [];
};

/**
* Validate and convert a value to a positive integer resource ID.
* Shared by resource template callbacks and prompt handlers.
*
* @param value - The value to validate (typically a string from user input)
* @returns The validated resource ID as a positive integer
* @throws {Error} If the value is not a finite positive integer
*/
export const validateResourceId = (value: unknown): number => {
const idx = Number(value);
if (Number.isFinite(idx) && Number.isInteger(idx) && idx > 0) {
return idx;
}
throw new Error(
`Invalid resourceId: ${value}. Must be a finite positive integer.`
);
};

const uriBase: string = "demo://resource/dynamic";
const textUriBase: string = `${uriBase}/text`;
const blobUriBase: string = `${uriBase}/blob`;
Expand Down Expand Up @@ -144,11 +162,9 @@ const parseResourceId = (uri: URL, variables: Record<string, unknown>) => {
) {
throw new Error(uriError);
} else {
const idxStr = String((variables as any).resourceId ?? "");
const idx = Number(idxStr);
if (Number.isFinite(idx) && Number.isInteger(idx) && idx > 0) {
return idx;
} else {
try {
return validateResourceId((variables as any).resourceId);
} catch {
throw new Error(uriError);
}
}
Expand Down
4 changes: 3 additions & 1 deletion src/everything/tools/trigger-elicitation-request-async.ts
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,8 @@ export const registerTriggerElicitationRequestAsyncTool = (
}

const taskId = elicitResponse.task.taskId;
const pollInterval =
elicitResponse.task.pollInterval ?? POLL_INTERVAL;
const statusMessages: string[] = [];
statusMessages.push(`Task created: ${taskId}`);

Expand All @@ -146,7 +148,7 @@ export const registerTriggerElicitationRequestAsyncTool = (
attempts < MAX_POLL_ATTEMPTS
) {
// Wait before polling
await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL));
await new Promise((resolve) => setTimeout(resolve, pollInterval));
attempts++;

// Get task status from client
Expand Down
26 changes: 22 additions & 4 deletions src/everything/tools/trigger-elicitation-request.ts
Original file line number Diff line number Diff line change
Expand Up @@ -187,9 +187,10 @@ export const registerTriggerElicitationRequestTool = (server: McpServer) => {
const userData = elicitationResult.content;
const lines = [];
if (userData.name) lines.push(`- Name: ${userData.name}`);
if (userData.check !== undefined)
lines.push(`- Agreed to terms: ${userData.check}`);
if (userData.color) lines.push(`- Favorite Color: ${userData.color}`);
if (userData.agreeToTerms !== undefined)
lines.push(`- Agreed to terms: ${userData.agreeToTerms}`);
if (userData.firstLine)
lines.push(`- First Line: ${userData.firstLine}`);
if (userData.email) lines.push(`- Email: ${userData.email}`);
if (userData.homepage) lines.push(`- Homepage: ${userData.homepage}`);
if (userData.birthdate)
Expand All @@ -198,7 +199,24 @@ export const registerTriggerElicitationRequestTool = (server: McpServer) => {
lines.push(`- Favorite Integer: ${userData.integer}`);
if (userData.number !== undefined)
lines.push(`- Favorite Number: ${userData.number}`);
if (userData.petType) lines.push(`- Pet Type: ${userData.petType}`);
if (userData.untitledSingleSelectEnum)
lines.push(
`- Untitled Single Select: ${userData.untitledSingleSelectEnum}`
);
if (userData.untitledMultipleSelectEnum)
lines.push(
`- Untitled Multiple Select: ${JSON.stringify(userData.untitledMultipleSelectEnum)}`
);
if (userData.titledSingleSelectEnum)
lines.push(
`- Titled Single Select: ${userData.titledSingleSelectEnum}`
);
if (userData.titledMultipleSelectEnum)
lines.push(
`- Titled Multiple Select: ${JSON.stringify(userData.titledMultipleSelectEnum)}`
);
if (userData.legacyTitledEnum)
lines.push(`- Legacy Titled Enum: ${userData.legacyTitledEnum}`);

content.push({
type: "text",
Expand Down
4 changes: 3 additions & 1 deletion src/everything/tools/trigger-sampling-request-async.ts
Original file line number Diff line number Diff line change
Expand Up @@ -135,6 +135,8 @@ export const registerTriggerSamplingRequestAsyncTool = (server: McpServer) => {
}

const taskId = samplingResponse.task.taskId;
const pollInterval =
samplingResponse.task.pollInterval ?? POLL_INTERVAL;
const statusMessages: string[] = [];
statusMessages.push(`Task created: ${taskId}`);

Expand All @@ -150,7 +152,7 @@ export const registerTriggerSamplingRequestAsyncTool = (server: McpServer) => {
attempts < MAX_POLL_ATTEMPTS
) {
// Wait before polling
await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL));
await new Promise((resolve) => setTimeout(resolve, pollInterval));
attempts++;

// Get task status from client
Expand Down
Loading