abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-19 09:33:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bryanmorgan/add mcp description support (#825)

Browse Source
This commit is contained in:
Bryan Morgan
2025-06-07 18:30:56 -04:00
committed by GitHub
parent dd08582f81
commit e95a6086fc
8 changed files with 246 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/core/src/config/config.ts
View File

@@ -49,6 +49,8 @@ export class MCPServerConfig {
// Common
readonly timeout?: number,
readonly trust?: boolean,
// Metadata
readonly description?: string,
) {}
}

10
packages/core/src/tools/mcp-tool.test.ts
View File

@@ -52,7 +52,7 @@ describe('DiscoveredMCPTool', () => {
});
describe('constructor', () => {
it('should set properties correctly and augment description (non-generic server)', () => {
it('should set properties correctly (non-generic server)', () => {
const tool = new DiscoveredMCPTool(
mockCallableToolInstance,
serverName, // serverName is 'mock-mcp-server', not 'mcp'
@@ -64,14 +64,13 @@ describe('DiscoveredMCPTool', () => {
expect(tool.name).toBe(toolNameForModel);
expect(tool.schema.name).toBe(toolNameForModel);
const expectedDescription = `${baseDescription}\n\nThis MCP tool named '${serverToolName}' was discovered from an MCP server.`;
expect(tool.schema.description).toBe(expectedDescription);
expect(tool.schema.description).toBe(baseDescription);
expect(tool.schema.parameters).toEqual(inputSchema);
expect(tool.serverToolName).toBe(serverToolName);
expect(tool.timeout).toBeUndefined();
});
it('should set properties correctly and augment description (generic "mcp" server)', () => {
it('should set properties correctly (generic "mcp" server)', () => {
const genericServerName = 'mcp';
const tool = new DiscoveredMCPTool(
mockCallableToolInstance,
@@ -81,8 +80,7 @@ describe('DiscoveredMCPTool', () => {
inputSchema,
serverToolName,
);
const expectedDescription = `${baseDescription}\n\nThis MCP tool named '${serverToolName}' was discovered from '${genericServerName}' MCP server.`;
expect(tool.schema.description).toBe(expectedDescription);
expect(tool.schema.description).toBe(baseDescription);
});
it('should accept and store a custom timeout', () => {

11
packages/core/src/tools/mcp-tool.ts
View File

@@ -30,17 +30,6 @@ export class DiscoveredMCPTool extends BaseTool<ToolParams, ToolResult> {
readonly timeout?: number,
readonly trust?: boolean,
) {
if (serverName !== 'mcp') {
// Add server name if not the generic 'mcp'
description += `
This MCP tool named '${serverToolName}' was discovered from an MCP server.`;
} else {
description += `
This MCP tool named '${serverToolName}' was discovered from '${serverName}' MCP server.`;
}
super(
name,
name,