abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-20 16:57:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: restart MCP servers on /mcp refresh (#5479)

Browse Source 
Co-authored-by: Brian Ray <62354532+emeryray2002@users.noreply.github.com>
Co-authored-by: N. Taylor Mullen <ntaylormullen@google.com>
This commit is contained in:
Ramón Medrano Llamas
2025-08-19 21:03:19 +02:00
committed by GitHub
parent 4828e4daf1
commit b24c5887c4
9 changed files with 447 additions and 467 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
packages/cli/src/ui/commands/mcpCommand.test.ts
View File

@@ -972,6 +972,7 @@ describe('mcpCommand', () => {
it('should refresh the list of tools and display the status', async () => {
const mockToolRegistry = {
discoverMcpTools: vi.fn(),
restartMcpServers: vi.fn(),
getAllTools: vi.fn().mockReturnValue([]),
};
const mockGeminiClient = {
@@ -1004,11 +1005,11 @@ describe('mcpCommand', () => {
expect(context.ui.addItem).toHaveBeenCalledWith(
{
type: 'info',
text: 'Refreshing MCP servers and tools...',
text: 'Restarting MCP servers...',
},
expect.any(Number),
);
expect(mockToolRegistry.discoverMcpTools).toHaveBeenCalled();
expect(mockToolRegistry.restartMcpServers).toHaveBeenCalled();
expect(mockGeminiClient.setTools).toHaveBeenCalled();
expect(context.ui.reloadCommands).toHaveBeenCalledTimes(1);

6
packages/cli/src/ui/commands/mcpCommand.ts
View File

@@ -471,7 +471,7 @@ const listCommand: SlashCommand = {
const refreshCommand: SlashCommand = {
name: 'refresh',
description: 'Refresh the list of MCP servers and tools',
description: 'Restarts MCP servers.',
kind: CommandKind.BUILT_IN,
action: async (
context: CommandContext,
@@ -497,12 +497,12 @@ const refreshCommand: SlashCommand = {
context.ui.addItem(
{
type: 'info',
text: 'Refreshing MCP servers and tools...',
text: 'Restarting MCP servers...',
},
Date.now(),
);
await toolRegistry.discoverMcpTools();
await toolRegistry.restartMcpServers();
// Update the client with the new tools
const geminiClient = config.getGeminiClient();