init claude-code
This commit is contained in:
@@ -0,0 +1,158 @@
|
||||
import {
|
||||
type ReadResourceResult,
|
||||
ReadResourceResultSchema,
|
||||
} from '@modelcontextprotocol/sdk/types.js'
|
||||
import { z } from 'zod/v4'
|
||||
import { ensureConnectedClient } from '../../services/mcp/client.js'
|
||||
import { buildTool, type ToolDef } from '../../Tool.js'
|
||||
import { lazySchema } from '../../utils/lazySchema.js'
|
||||
import {
|
||||
getBinaryBlobSavedMessage,
|
||||
persistBinaryContent,
|
||||
} from '../../utils/mcpOutputStorage.js'
|
||||
import { jsonStringify } from '../../utils/slowOperations.js'
|
||||
import { isOutputLineTruncated } from '../../utils/terminal.js'
|
||||
import { DESCRIPTION, PROMPT } from './prompt.js'
|
||||
import {
|
||||
renderToolResultMessage,
|
||||
renderToolUseMessage,
|
||||
userFacingName,
|
||||
} from './UI.js'
|
||||
|
||||
export const inputSchema = lazySchema(() =>
|
||||
z.object({
|
||||
server: z.string().describe('The MCP server name'),
|
||||
uri: z.string().describe('The resource URI to read'),
|
||||
}),
|
||||
)
|
||||
type InputSchema = ReturnType<typeof inputSchema>
|
||||
|
||||
export const outputSchema = lazySchema(() =>
|
||||
z.object({
|
||||
contents: z.array(
|
||||
z.object({
|
||||
uri: z.string().describe('Resource URI'),
|
||||
mimeType: z.string().optional().describe('MIME type of the content'),
|
||||
text: z.string().optional().describe('Text content of the resource'),
|
||||
blobSavedTo: z
|
||||
.string()
|
||||
.optional()
|
||||
.describe('Path where binary blob content was saved'),
|
||||
}),
|
||||
),
|
||||
}),
|
||||
)
|
||||
type OutputSchema = ReturnType<typeof outputSchema>
|
||||
|
||||
export type Output = z.infer<OutputSchema>
|
||||
|
||||
export const ReadMcpResourceTool = buildTool({
|
||||
isConcurrencySafe() {
|
||||
return true
|
||||
},
|
||||
isReadOnly() {
|
||||
return true
|
||||
},
|
||||
toAutoClassifierInput(input) {
|
||||
return `${input.server} ${input.uri}`
|
||||
},
|
||||
shouldDefer: true,
|
||||
name: 'ReadMcpResourceTool',
|
||||
searchHint: 'read a specific MCP resource by URI',
|
||||
maxResultSizeChars: 100_000,
|
||||
async description() {
|
||||
return DESCRIPTION
|
||||
},
|
||||
async prompt() {
|
||||
return PROMPT
|
||||
},
|
||||
get inputSchema(): InputSchema {
|
||||
return inputSchema()
|
||||
},
|
||||
get outputSchema(): OutputSchema {
|
||||
return outputSchema()
|
||||
},
|
||||
async call(input, { options: { mcpClients } }) {
|
||||
const { server: serverName, uri } = input
|
||||
|
||||
const client = mcpClients.find(client => client.name === serverName)
|
||||
|
||||
if (!client) {
|
||||
throw new Error(
|
||||
`Server "${serverName}" not found. Available servers: ${mcpClients.map(c => c.name).join(', ')}`,
|
||||
)
|
||||
}
|
||||
|
||||
if (client.type !== 'connected') {
|
||||
throw new Error(`Server "${serverName}" is not connected`)
|
||||
}
|
||||
|
||||
if (!client.capabilities?.resources) {
|
||||
throw new Error(`Server "${serverName}" does not support resources`)
|
||||
}
|
||||
|
||||
const connectedClient = await ensureConnectedClient(client)
|
||||
const result = (await connectedClient.client.request(
|
||||
{
|
||||
method: 'resources/read',
|
||||
params: { uri },
|
||||
},
|
||||
ReadResourceResultSchema,
|
||||
)) as ReadResourceResult
|
||||
|
||||
// Intercept any blob fields: decode, write raw bytes to disk with a
|
||||
// mime-derived extension, and replace with a path. Otherwise the base64
|
||||
// would be stringified straight into the context.
|
||||
const contents = await Promise.all(
|
||||
result.contents.map(async (c, i) => {
|
||||
if ('text' in c) {
|
||||
return { uri: c.uri, mimeType: c.mimeType, text: c.text }
|
||||
}
|
||||
if (!('blob' in c) || typeof c.blob !== 'string') {
|
||||
return { uri: c.uri, mimeType: c.mimeType }
|
||||
}
|
||||
const persistId = `mcp-resource-${Date.now()}-${i}-${Math.random().toString(36).slice(2, 8)}`
|
||||
const persisted = await persistBinaryContent(
|
||||
Buffer.from(c.blob, 'base64'),
|
||||
c.mimeType,
|
||||
persistId,
|
||||
)
|
||||
if ('error' in persisted) {
|
||||
return {
|
||||
uri: c.uri,
|
||||
mimeType: c.mimeType,
|
||||
text: `Binary content could not be saved to disk: ${persisted.error}`,
|
||||
}
|
||||
}
|
||||
return {
|
||||
uri: c.uri,
|
||||
mimeType: c.mimeType,
|
||||
blobSavedTo: persisted.filepath,
|
||||
text: getBinaryBlobSavedMessage(
|
||||
persisted.filepath,
|
||||
c.mimeType,
|
||||
persisted.size,
|
||||
`[Resource from ${serverName} at ${c.uri}] `,
|
||||
),
|
||||
}
|
||||
}),
|
||||
)
|
||||
|
||||
return {
|
||||
data: { contents },
|
||||
}
|
||||
},
|
||||
renderToolUseMessage,
|
||||
userFacingName,
|
||||
renderToolResultMessage,
|
||||
isResultTruncated(output: Output): boolean {
|
||||
return isOutputLineTruncated(jsonStringify(output))
|
||||
},
|
||||
mapToolResultToToolResultBlockParam(content, toolUseID) {
|
||||
return {
|
||||
tool_use_id: toolUseID,
|
||||
type: 'tool_result',
|
||||
content: jsonStringify(content),
|
||||
}
|
||||
},
|
||||
} satisfies ToolDef<InputSchema, Output>)
|
||||
@@ -0,0 +1,37 @@
|
||||
import * as React from 'react';
|
||||
import type { z } from 'zod/v4';
|
||||
import { MessageResponse } from '../../components/MessageResponse.js';
|
||||
import { OutputLine } from '../../components/shell/OutputLine.js';
|
||||
import { Box, Text } from '../../ink.js';
|
||||
import type { ToolProgressData } from '../../Tool.js';
|
||||
import type { ProgressMessage } from '../../types/message.js';
|
||||
import { jsonStringify } from '../../utils/slowOperations.js';
|
||||
import type { inputSchema, Output } from './ReadMcpResourceTool.js';
|
||||
export function renderToolUseMessage(input: Partial<z.infer<ReturnType<typeof inputSchema>>>): React.ReactNode {
|
||||
if (!input.uri || !input.server) {
|
||||
return null;
|
||||
}
|
||||
return `Read resource "${input.uri}" from server "${input.server}"`;
|
||||
}
|
||||
export function userFacingName(): string {
|
||||
return 'readMcpResource';
|
||||
}
|
||||
export function renderToolResultMessage(output: Output, _progressMessagesForMessage: ProgressMessage<ToolProgressData>[], {
|
||||
verbose
|
||||
}: {
|
||||
verbose: boolean;
|
||||
}): React.ReactNode {
|
||||
if (!output || !output.contents || output.contents.length === 0) {
|
||||
return <Box justifyContent="space-between" overflowX="hidden" width="100%">
|
||||
<MessageResponse height={1}>
|
||||
<Text dimColor>(No content)</Text>
|
||||
</MessageResponse>
|
||||
</Box>;
|
||||
}
|
||||
|
||||
// Format as JSON for better readability
|
||||
// eslint-disable-next-line no-restricted-syntax -- human-facing UI, not tool_result
|
||||
const formattedOutput = jsonStringify(output, null, 2);
|
||||
return <OutputLine content={formattedOutput} verbose={verbose} />;
|
||||
}
|
||||
//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJuYW1lcyI6WyJSZWFjdCIsInoiLCJNZXNzYWdlUmVzcG9uc2UiLCJPdXRwdXRMaW5lIiwiQm94IiwiVGV4dCIsIlRvb2xQcm9ncmVzc0RhdGEiLCJQcm9ncmVzc01lc3NhZ2UiLCJqc29uU3RyaW5naWZ5IiwiaW5wdXRTY2hlbWEiLCJPdXRwdXQiLCJyZW5kZXJUb29sVXNlTWVzc2FnZSIsImlucHV0IiwiUGFydGlhbCIsImluZmVyIiwiUmV0dXJuVHlwZSIsIlJlYWN0Tm9kZSIsInVyaSIsInNlcnZlciIsInVzZXJGYWNpbmdOYW1lIiwicmVuZGVyVG9vbFJlc3VsdE1lc3NhZ2UiLCJvdXRwdXQiLCJfcHJvZ3Jlc3NNZXNzYWdlc0Zvck1lc3NhZ2UiLCJ2ZXJib3NlIiwiY29udGVudHMiLCJsZW5ndGgiLCJmb3JtYXR0ZWRPdXRwdXQiXSwic291cmNlcyI6WyJVSS50c3giXSwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0ICogYXMgUmVhY3QgZnJvbSAncmVhY3QnXG5pbXBvcnQgdHlwZSB7IHogfSBmcm9tICd6b2QvdjQnXG5pbXBvcnQgeyBNZXNzYWdlUmVzcG9uc2UgfSBmcm9tICcuLi8uLi9jb21wb25lbnRzL01lc3NhZ2VSZXNwb25zZS5qcydcbmltcG9ydCB7IE91dHB1dExpbmUgfSBmcm9tICcuLi8uLi9jb21wb25lbnRzL3NoZWxsL091dHB1dExpbmUuanMnXG5pbXBvcnQgeyBCb3gsIFRleHQgfSBmcm9tICcuLi8uLi9pbmsuanMnXG5pbXBvcnQgdHlwZSB7IFRvb2xQcm9ncmVzc0RhdGEgfSBmcm9tICcuLi8uLi9Ub29sLmpzJ1xuaW1wb3J0IHR5cGUgeyBQcm9ncmVzc01lc3NhZ2UgfSBmcm9tICcuLi8uLi90eXBlcy9tZXNzYWdlLmpzJ1xuaW1wb3J0IHsganNvblN0cmluZ2lmeSB9IGZyb20gJy4uLy4uL3V0aWxzL3Nsb3dPcGVyYXRpb25zLmpzJ1xuaW1wb3J0IHR5cGUgeyBpbnB1dFNjaGVtYSwgT3V0cHV0IH0gZnJvbSAnLi9SZWFkTWNwUmVzb3VyY2VUb29sLmpzJ1xuXG5leHBvcnQgZnVuY3Rpb24gcmVuZGVyVG9vbFVzZU1lc3NhZ2UoXG4gIGlucHV0OiBQYXJ0aWFsPHouaW5mZXI8UmV0dXJuVHlwZTx0eXBlb2YgaW5wdXRTY2hlbWE+Pj4sXG4pOiBSZWFjdC5SZWFjdE5vZGUge1xuICBpZiAoIWlucHV0LnVyaSB8fCAhaW5wdXQuc2VydmVyKSB7XG4gICAgcmV0dXJuIG51bGxcbiAgfVxuICByZXR1cm4gYFJlYWQgcmVzb3VyY2UgXCIke2lucHV0LnVyaX1cIiBmcm9tIHNlcnZlciBcIiR7aW5wdXQuc2VydmVyfVwiYFxufVxuXG5leHBvcnQgZnVuY3Rpb24gdXNlckZhY2luZ05hbWUoKTogc3RyaW5nIHtcbiAgcmV0dXJuICdyZWFkTWNwUmVzb3VyY2UnXG59XG5cbmV4cG9ydCBmdW5jdGlvbiByZW5kZXJUb29sUmVzdWx0TWVzc2FnZShcbiAgb3V0cHV0OiBPdXRwdXQsXG4gIF9wcm9ncmVzc01lc3NhZ2VzRm9yTWVzc2FnZTogUHJvZ3Jlc3NNZXNzYWdlPFRvb2xQcm9ncmVzc0RhdGE+W10sXG4gIHsgdmVyYm9zZSB9OiB7IHZlcmJvc2U6IGJvb2xlYW4gfSxcbik6IFJlYWN0LlJlYWN0Tm9kZSB7XG4gIGlmICghb3V0cHV0IHx8ICFvdXRwdXQuY29udGVudHMgfHwgb3V0cHV0LmNvbnRlbnRzLmxlbmd0aCA9PT0gMCkge1xuICAgIHJldHVybiAoXG4gICAgICA8Qm94IGp1c3RpZnlDb250ZW50PVwic3BhY2UtYmV0d2VlblwiIG92ZXJmbG93WD1cImhpZGRlblwiIHdpZHRoPVwiMTAwJVwiPlxuICAgICAgICA8TWVzc2FnZVJlc3BvbnNlIGhlaWdodD17MX0+XG4gICAgICAgICAgPFRleHQgZGltQ29sb3I+KE5vIGNvbnRlbnQpPC9UZXh0PlxuICAgICAgICA8L01lc3NhZ2VSZXNwb25zZT5cbiAgICAgIDwvQm94PlxuICAgIClcbiAgfVxuXG4gIC8vIEZvcm1hdCBhcyBKU09OIGZvciBiZXR0ZXIgcmVhZGFiaWxpdHlcbiAgLy8gZXNsaW50LWRpc2FibGUtbmV4dC1saW5lIG5vLXJlc3RyaWN0ZWQtc3ludGF4IC0tIGh1bWFuLWZhY2luZyBVSSwgbm90IHRvb2xfcmVzdWx0XG4gIGNvbnN0IGZvcm1hdHRlZE91dHB1dCA9IGpzb25TdHJpbmdpZnkob3V0cHV0LCBudWxsLCAyKVxuXG4gIHJldHVybiA8T3V0cHV0TGluZSBjb250ZW50PXtmb3JtYXR0ZWRPdXRwdXR9IHZlcmJvc2U9e3ZlcmJvc2V9IC8+XG59XG4iXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sS0FBS0EsS0FBSyxNQUFNLE9BQU87QUFDOUIsY0FBY0MsQ0FBQyxRQUFRLFFBQVE7QUFDL0IsU0FBU0MsZUFBZSxRQUFRLHFDQUFxQztBQUNyRSxTQUFTQyxVQUFVLFFBQVEsc0NBQXNDO0FBQ2pFLFNBQVNDLEdBQUcsRUFBRUMsSUFBSSxRQUFRLGNBQWM7QUFDeEMsY0FBY0MsZ0JBQWdCLFFBQVEsZUFBZTtBQUNyRCxjQUFjQyxlQUFlLFFBQVEsd0JBQXdCO0FBQzdELFNBQVNDLGFBQWEsUUFBUSwrQkFBK0I7QUFDN0QsY0FBY0MsV0FBVyxFQUFFQyxNQUFNLFFBQVEsMEJBQTBCO0FBRW5FLE9BQU8sU0FBU0Msb0JBQW9CQSxDQUNsQ0MsS0FBSyxFQUFFQyxPQUFPLENBQUNaLENBQUMsQ0FBQ2EsS0FBSyxDQUFDQyxVQUFVLENBQUMsT0FBT04sV0FBVyxDQUFDLENBQUMsQ0FBQyxDQUN4RCxFQUFFVCxLQUFLLENBQUNnQixTQUFTLENBQUM7RUFDakIsSUFBSSxDQUFDSixLQUFLLENBQUNLLEdBQUcsSUFBSSxDQUFDTCxLQUFLLENBQUNNLE1BQU0sRUFBRTtJQUMvQixPQUFPLElBQUk7RUFDYjtFQUNBLE9BQU8sa0JBQWtCTixLQUFLLENBQUNLLEdBQUcsa0JBQWtCTCxLQUFLLENBQUNNLE1BQU0sR0FBRztBQUNyRTtBQUVBLE9BQU8sU0FBU0MsY0FBY0EsQ0FBQSxDQUFFLEVBQUUsTUFBTSxDQUFDO0VBQ3ZDLE9BQU8saUJBQWlCO0FBQzFCO0FBRUEsT0FBTyxTQUFTQyx1QkFBdUJBLENBQ3JDQyxNQUFNLEVBQUVYLE1BQU0sRUFDZFksMkJBQTJCLEVBQUVmLGVBQWUsQ0FBQ0QsZ0JBQWdCLENBQUMsRUFBRSxFQUNoRTtFQUFFaUI7QUFBOEIsQ0FBckIsRUFBRTtFQUFFQSxPQUFPLEVBQUUsT0FBTztBQUFDLENBQUMsQ0FDbEMsRUFBRXZCLEtBQUssQ0FBQ2dCLFNBQVMsQ0FBQztFQUNqQixJQUFJLENBQUNLLE1BQU0sSUFBSSxDQUFDQSxNQUFNLENBQUNHLFFBQVEsSUFBSUgsTUFBTSxDQUFDRyxRQUFRLENBQUNDLE1BQU0sS0FBSyxDQUFDLEVBQUU7SUFDL0QsT0FDRSxDQUFDLEdBQUcsQ0FBQyxjQUFjLENBQUMsZUFBZSxDQUFDLFNBQVMsQ0FBQyxRQUFRLENBQUMsS0FBSyxDQUFDLE1BQU07QUFDekUsUUFBUSxDQUFDLGVBQWUsQ0FBQyxNQUFNLENBQUMsQ0FBQyxDQUFDLENBQUM7QUFDbkMsVUFBVSxDQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsWUFBWSxFQUFFLElBQUk7QUFDM0MsUUFBUSxFQUFFLGVBQWU7QUFDekIsTUFBTSxFQUFFLEdBQUcsQ0FBQztFQUVWOztFQUVBO0VBQ0E7RUFDQSxNQUFNQyxlQUFlLEdBQUdsQixhQUFhLENBQUNhLE1BQU0sRUFBRSxJQUFJLEVBQUUsQ0FBQyxDQUFDO0VBRXRELE9BQU8sQ0FBQyxVQUFVLENBQUMsT0FBTyxDQUFDLENBQUNLLGVBQWUsQ0FBQyxDQUFDLE9BQU8sQ0FBQyxDQUFDSCxPQUFPLENBQUMsR0FBRztBQUNuRSIsImlnbm9yZUxpc3QiOltdfQ==
|
||||
@@ -0,0 +1,16 @@
|
||||
export const DESCRIPTION = `
|
||||
Reads a specific resource from an MCP server.
|
||||
- server: The name of the MCP server to read from
|
||||
- uri: The URI of the resource to read
|
||||
|
||||
Usage examples:
|
||||
- Read a resource from a server: \`readMcpResource({ server: "myserver", uri: "my-resource-uri" })\`
|
||||
`
|
||||
|
||||
export const PROMPT = `
|
||||
Reads a specific resource from an MCP server, identified by server name and resource URI.
|
||||
|
||||
Parameters:
|
||||
- server (required): The name of the MCP server from which to read the resource
|
||||
- uri (required): The URI of the resource to read
|
||||
`
|
||||
Reference in New Issue
Block a user