update `/tools` to new slash command arch (#4236)
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: matt korwel <matt.korwel@gmail.com>
This commit is contained in:
parent
e4ed1aabac
commit
21eb44b242
|
@ -17,8 +17,9 @@ import { themeCommand } from '../ui/commands/themeCommand.js';
|
||||||
import { statsCommand } from '../ui/commands/statsCommand.js';
|
import { statsCommand } from '../ui/commands/statsCommand.js';
|
||||||
import { privacyCommand } from '../ui/commands/privacyCommand.js';
|
import { privacyCommand } from '../ui/commands/privacyCommand.js';
|
||||||
import { aboutCommand } from '../ui/commands/aboutCommand.js';
|
import { aboutCommand } from '../ui/commands/aboutCommand.js';
|
||||||
import { compressCommand } from '../ui/commands/compressCommand.js';
|
|
||||||
import { extensionsCommand } from '../ui/commands/extensionsCommand.js';
|
import { extensionsCommand } from '../ui/commands/extensionsCommand.js';
|
||||||
|
import { toolsCommand } from '../ui/commands/toolsCommand.js';
|
||||||
|
import { compressCommand } from '../ui/commands/compressCommand.js';
|
||||||
import { mcpCommand } from '../ui/commands/mcpCommand.js';
|
import { mcpCommand } from '../ui/commands/mcpCommand.js';
|
||||||
|
|
||||||
// Mock the command modules to isolate the service from the command implementations.
|
// Mock the command modules to isolate the service from the command implementations.
|
||||||
|
@ -49,18 +50,21 @@ vi.mock('../ui/commands/statsCommand.js', () => ({
|
||||||
vi.mock('../ui/commands/aboutCommand.js', () => ({
|
vi.mock('../ui/commands/aboutCommand.js', () => ({
|
||||||
aboutCommand: { name: 'about', description: 'Mock About' },
|
aboutCommand: { name: 'about', description: 'Mock About' },
|
||||||
}));
|
}));
|
||||||
vi.mock('../ui/commands/compressCommand.js', () => ({
|
|
||||||
compressCommand: { name: 'compress', description: 'Mock Compress' },
|
|
||||||
}));
|
|
||||||
vi.mock('../ui/commands/extensionsCommand.js', () => ({
|
vi.mock('../ui/commands/extensionsCommand.js', () => ({
|
||||||
extensionsCommand: { name: 'extensions', description: 'Mock Extensions' },
|
extensionsCommand: { name: 'extensions', description: 'Mock Extensions' },
|
||||||
}));
|
}));
|
||||||
|
vi.mock('../ui/commands/toolsCommand.js', () => ({
|
||||||
|
toolsCommand: { name: 'tools', description: 'Mock Tools' },
|
||||||
|
}));
|
||||||
|
vi.mock('../ui/commands/compressCommand.js', () => ({
|
||||||
|
compressCommand: { name: 'compress', description: 'Mock Compress' },
|
||||||
|
}));
|
||||||
vi.mock('../ui/commands/mcpCommand.js', () => ({
|
vi.mock('../ui/commands/mcpCommand.js', () => ({
|
||||||
mcpCommand: { name: 'mcp', description: 'Mock MCP' },
|
mcpCommand: { name: 'mcp', description: 'Mock MCP' },
|
||||||
}));
|
}));
|
||||||
|
|
||||||
describe('CommandService', () => {
|
describe('CommandService', () => {
|
||||||
const subCommandLen = 13;
|
const subCommandLen = 14;
|
||||||
|
|
||||||
describe('when using default production loader', () => {
|
describe('when using default production loader', () => {
|
||||||
let commandService: CommandService;
|
let commandService: CommandService;
|
||||||
|
@ -98,8 +102,9 @@ describe('CommandService', () => {
|
||||||
expect(commandNames).toContain('stats');
|
expect(commandNames).toContain('stats');
|
||||||
expect(commandNames).toContain('privacy');
|
expect(commandNames).toContain('privacy');
|
||||||
expect(commandNames).toContain('about');
|
expect(commandNames).toContain('about');
|
||||||
expect(commandNames).toContain('compress');
|
|
||||||
expect(commandNames).toContain('extensions');
|
expect(commandNames).toContain('extensions');
|
||||||
|
expect(commandNames).toContain('tools');
|
||||||
|
expect(commandNames).toContain('compress');
|
||||||
expect(commandNames).toContain('mcp');
|
expect(commandNames).toContain('mcp');
|
||||||
});
|
});
|
||||||
|
|
||||||
|
@ -140,6 +145,7 @@ describe('CommandService', () => {
|
||||||
privacyCommand,
|
privacyCommand,
|
||||||
statsCommand,
|
statsCommand,
|
||||||
themeCommand,
|
themeCommand,
|
||||||
|
toolsCommand,
|
||||||
]);
|
]);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
|
@ -16,8 +16,9 @@ import { chatCommand } from '../ui/commands/chatCommand.js';
|
||||||
import { statsCommand } from '../ui/commands/statsCommand.js';
|
import { statsCommand } from '../ui/commands/statsCommand.js';
|
||||||
import { privacyCommand } from '../ui/commands/privacyCommand.js';
|
import { privacyCommand } from '../ui/commands/privacyCommand.js';
|
||||||
import { aboutCommand } from '../ui/commands/aboutCommand.js';
|
import { aboutCommand } from '../ui/commands/aboutCommand.js';
|
||||||
import { compressCommand } from '../ui/commands/compressCommand.js';
|
|
||||||
import { extensionsCommand } from '../ui/commands/extensionsCommand.js';
|
import { extensionsCommand } from '../ui/commands/extensionsCommand.js';
|
||||||
|
import { toolsCommand } from '../ui/commands/toolsCommand.js';
|
||||||
|
import { compressCommand } from '../ui/commands/compressCommand.js';
|
||||||
|
|
||||||
const loadBuiltInCommands = async (): Promise<SlashCommand[]> => [
|
const loadBuiltInCommands = async (): Promise<SlashCommand[]> => [
|
||||||
aboutCommand,
|
aboutCommand,
|
||||||
|
@ -33,6 +34,7 @@ const loadBuiltInCommands = async (): Promise<SlashCommand[]> => [
|
||||||
privacyCommand,
|
privacyCommand,
|
||||||
statsCommand,
|
statsCommand,
|
||||||
themeCommand,
|
themeCommand,
|
||||||
|
toolsCommand,
|
||||||
];
|
];
|
||||||
|
|
||||||
export class CommandService {
|
export class CommandService {
|
||||||
|
|
|
@ -390,7 +390,6 @@ const App = ({ config, settings, startupWarnings = [], version }: AppProps) => {
|
||||||
openAuthDialog,
|
openAuthDialog,
|
||||||
openEditorDialog,
|
openEditorDialog,
|
||||||
toggleCorgiMode,
|
toggleCorgiMode,
|
||||||
showToolDescriptions,
|
|
||||||
setQuittingMessages,
|
setQuittingMessages,
|
||||||
openPrivacyNotice,
|
openPrivacyNotice,
|
||||||
);
|
);
|
||||||
|
|
|
@ -0,0 +1,108 @@
|
||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect, vi } from 'vitest';
|
||||||
|
import { toolsCommand } from './toolsCommand.js';
|
||||||
|
import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
|
||||||
|
import { MessageType } from '../types.js';
|
||||||
|
import { Tool } from '@google/gemini-cli-core';
|
||||||
|
|
||||||
|
// Mock tools for testing
|
||||||
|
const mockTools = [
|
||||||
|
{
|
||||||
|
name: 'file-reader',
|
||||||
|
displayName: 'File Reader',
|
||||||
|
description: 'Reads files from the local system.',
|
||||||
|
schema: {},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'code-editor',
|
||||||
|
displayName: 'Code Editor',
|
||||||
|
description: 'Edits code files.',
|
||||||
|
schema: {},
|
||||||
|
},
|
||||||
|
] as Tool[];
|
||||||
|
|
||||||
|
describe('toolsCommand', () => {
|
||||||
|
it('should display an error if the tool registry is unavailable', async () => {
|
||||||
|
const mockContext = createMockCommandContext({
|
||||||
|
services: {
|
||||||
|
config: {
|
||||||
|
getToolRegistry: () => Promise.resolve(undefined),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!toolsCommand.action) throw new Error('Action not defined');
|
||||||
|
await toolsCommand.action(mockContext, '');
|
||||||
|
|
||||||
|
expect(mockContext.ui.addItem).toHaveBeenCalledWith(
|
||||||
|
{
|
||||||
|
type: MessageType.ERROR,
|
||||||
|
text: 'Could not retrieve tool registry.',
|
||||||
|
},
|
||||||
|
expect.any(Number),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should display "No tools available" when none are found', async () => {
|
||||||
|
const mockContext = createMockCommandContext({
|
||||||
|
services: {
|
||||||
|
config: {
|
||||||
|
getToolRegistry: () =>
|
||||||
|
Promise.resolve({ getAllTools: () => [] as Tool[] }),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!toolsCommand.action) throw new Error('Action not defined');
|
||||||
|
await toolsCommand.action(mockContext, '');
|
||||||
|
|
||||||
|
expect(mockContext.ui.addItem).toHaveBeenCalledWith(
|
||||||
|
expect.objectContaining({
|
||||||
|
text: expect.stringContaining('No tools available'),
|
||||||
|
}),
|
||||||
|
expect.any(Number),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should list tools without descriptions by default', async () => {
|
||||||
|
const mockContext = createMockCommandContext({
|
||||||
|
services: {
|
||||||
|
config: {
|
||||||
|
getToolRegistry: () =>
|
||||||
|
Promise.resolve({ getAllTools: () => mockTools }),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!toolsCommand.action) throw new Error('Action not defined');
|
||||||
|
await toolsCommand.action(mockContext, '');
|
||||||
|
|
||||||
|
const message = (mockContext.ui.addItem as vi.Mock).mock.calls[0][0].text;
|
||||||
|
expect(message).not.toContain('Reads files from the local system.');
|
||||||
|
expect(message).toContain('File Reader');
|
||||||
|
expect(message).toContain('Code Editor');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should list tools with descriptions when "desc" arg is passed', async () => {
|
||||||
|
const mockContext = createMockCommandContext({
|
||||||
|
services: {
|
||||||
|
config: {
|
||||||
|
getToolRegistry: () =>
|
||||||
|
Promise.resolve({ getAllTools: () => mockTools }),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!toolsCommand.action) throw new Error('Action not defined');
|
||||||
|
await toolsCommand.action(mockContext, 'desc');
|
||||||
|
|
||||||
|
const message = (mockContext.ui.addItem as vi.Mock).mock.calls[0][0].text;
|
||||||
|
expect(message).toContain('Reads files from the local system.');
|
||||||
|
expect(message).toContain('Edits code files.');
|
||||||
|
});
|
||||||
|
});
|
|
@ -0,0 +1,66 @@
|
||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { type CommandContext, type SlashCommand } from './types.js';
|
||||||
|
import { MessageType } from '../types.js';
|
||||||
|
|
||||||
|
export const toolsCommand: SlashCommand = {
|
||||||
|
name: 'tools',
|
||||||
|
description: 'list available Gemini CLI tools',
|
||||||
|
action: async (context: CommandContext, args?: string): Promise<void> => {
|
||||||
|
const subCommand = args?.trim();
|
||||||
|
|
||||||
|
// Default to NOT showing descriptions. The user must opt in with an argument.
|
||||||
|
let useShowDescriptions = false;
|
||||||
|
if (subCommand === 'desc' || subCommand === 'descriptions') {
|
||||||
|
useShowDescriptions = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
const toolRegistry = await context.services.config?.getToolRegistry();
|
||||||
|
if (!toolRegistry) {
|
||||||
|
context.ui.addItem(
|
||||||
|
{
|
||||||
|
type: MessageType.ERROR,
|
||||||
|
text: 'Could not retrieve tool registry.',
|
||||||
|
},
|
||||||
|
Date.now(),
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const tools = toolRegistry.getAllTools();
|
||||||
|
// Filter out MCP tools by checking for the absence of a serverName property
|
||||||
|
const geminiTools = tools.filter((tool) => !('serverName' in tool));
|
||||||
|
|
||||||
|
let message = 'Available Gemini CLI tools:\n\n';
|
||||||
|
|
||||||
|
if (geminiTools.length > 0) {
|
||||||
|
geminiTools.forEach((tool) => {
|
||||||
|
if (useShowDescriptions && tool.description) {
|
||||||
|
message += ` - \u001b[36m${tool.displayName} (${tool.name})\u001b[0m:\n`;
|
||||||
|
|
||||||
|
const greenColor = '\u001b[32m';
|
||||||
|
const resetColor = '\u001b[0m';
|
||||||
|
|
||||||
|
// Handle multi-line descriptions
|
||||||
|
const descLines = tool.description.trim().split('\n');
|
||||||
|
for (const descLine of descLines) {
|
||||||
|
message += ` ${greenColor}${descLine}${resetColor}\n`;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
message += ` - \u001b[36m${tool.displayName}\u001b[0m\n`;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
message += ' No tools available\n';
|
||||||
|
}
|
||||||
|
message += '\n';
|
||||||
|
|
||||||
|
message += '\u001b[0m';
|
||||||
|
|
||||||
|
context.ui.addItem({ type: MessageType.INFO, text: message }, Date.now());
|
||||||
|
},
|
||||||
|
};
|
|
@ -66,7 +66,7 @@ import {
|
||||||
} from 'vitest';
|
} from 'vitest';
|
||||||
import open from 'open';
|
import open from 'open';
|
||||||
import { useSlashCommandProcessor } from './slashCommandProcessor.js';
|
import { useSlashCommandProcessor } from './slashCommandProcessor.js';
|
||||||
import { MessageType, SlashCommandProcessorResult } from '../types.js';
|
import { SlashCommandProcessorResult } from '../types.js';
|
||||||
import { Config, GeminiClient } from '@google/gemini-cli-core';
|
import { Config, GeminiClient } from '@google/gemini-cli-core';
|
||||||
import { useSessionStats } from '../contexts/SessionContext.js';
|
import { useSessionStats } from '../contexts/SessionContext.js';
|
||||||
import { LoadedSettings } from '../../config/settings.js';
|
import { LoadedSettings } from '../../config/settings.js';
|
||||||
|
@ -176,7 +176,7 @@ describe('useSlashCommandProcessor', () => {
|
||||||
process.env = { ...globalThis.process.env };
|
process.env = { ...globalThis.process.env };
|
||||||
});
|
});
|
||||||
|
|
||||||
const getProcessorHook = (showToolDescriptions: boolean = false) => {
|
const getProcessorHook = () => {
|
||||||
const settings = {
|
const settings = {
|
||||||
merged: {
|
merged: {
|
||||||
contextFileName: 'GEMINI.md',
|
contextFileName: 'GEMINI.md',
|
||||||
|
@ -197,15 +197,13 @@ describe('useSlashCommandProcessor', () => {
|
||||||
mockOpenAuthDialog,
|
mockOpenAuthDialog,
|
||||||
mockOpenEditorDialog,
|
mockOpenEditorDialog,
|
||||||
mockCorgiMode,
|
mockCorgiMode,
|
||||||
showToolDescriptions,
|
|
||||||
mockSetQuittingMessages,
|
mockSetQuittingMessages,
|
||||||
vi.fn(), // mockOpenPrivacyNotice
|
vi.fn(), // mockOpenPrivacyNotice
|
||||||
),
|
),
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|
||||||
const getProcessor = (showToolDescriptions: boolean = false) =>
|
const getProcessor = () => getProcessorHook().result.current;
|
||||||
getProcessorHook(showToolDescriptions).result.current;
|
|
||||||
|
|
||||||
describe('Other commands', () => {
|
describe('Other commands', () => {
|
||||||
it('/editor should open editor dialog and return handled', async () => {
|
it('/editor should open editor dialog and return handled', async () => {
|
||||||
|
@ -595,160 +593,4 @@ describe('useSlashCommandProcessor', () => {
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
describe('Unknown command', () => {
|
|
||||||
it('should show an error and return handled for a general unknown command', async () => {
|
|
||||||
const { handleSlashCommand } = getProcessor();
|
|
||||||
let commandResult: SlashCommandProcessorResult | false = false;
|
|
||||||
await act(async () => {
|
|
||||||
commandResult = await handleSlashCommand('/unknowncommand');
|
|
||||||
});
|
|
||||||
expect(mockAddItem).toHaveBeenNthCalledWith(
|
|
||||||
2,
|
|
||||||
expect.objectContaining({
|
|
||||||
type: MessageType.ERROR,
|
|
||||||
text: 'Unknown command: /unknowncommand',
|
|
||||||
}),
|
|
||||||
expect.any(Number),
|
|
||||||
);
|
|
||||||
expect(commandResult).toEqual({ type: 'handled' });
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('/tools command', () => {
|
|
||||||
it('should show an error if tool registry is not available', async () => {
|
|
||||||
mockConfig = {
|
|
||||||
...mockConfig,
|
|
||||||
getToolRegistry: vi.fn().mockResolvedValue(undefined),
|
|
||||||
} as unknown as Config;
|
|
||||||
const { handleSlashCommand } = getProcessor();
|
|
||||||
let commandResult: SlashCommandProcessorResult | false = false;
|
|
||||||
await act(async () => {
|
|
||||||
commandResult = await handleSlashCommand('/tools');
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(mockAddItem).toHaveBeenNthCalledWith(
|
|
||||||
2,
|
|
||||||
expect.objectContaining({
|
|
||||||
type: MessageType.ERROR,
|
|
||||||
text: 'Could not retrieve tools.',
|
|
||||||
}),
|
|
||||||
expect.any(Number),
|
|
||||||
);
|
|
||||||
expect(commandResult).toEqual({ type: 'handled' });
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should show an error if getAllTools returns undefined', async () => {
|
|
||||||
mockConfig = {
|
|
||||||
...mockConfig,
|
|
||||||
getToolRegistry: vi.fn().mockResolvedValue({
|
|
||||||
getAllTools: vi.fn().mockReturnValue(undefined),
|
|
||||||
}),
|
|
||||||
} as unknown as Config;
|
|
||||||
const { handleSlashCommand } = getProcessor();
|
|
||||||
let commandResult: SlashCommandProcessorResult | false = false;
|
|
||||||
await act(async () => {
|
|
||||||
commandResult = await handleSlashCommand('/tools');
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(mockAddItem).toHaveBeenNthCalledWith(
|
|
||||||
2,
|
|
||||||
expect.objectContaining({
|
|
||||||
type: MessageType.ERROR,
|
|
||||||
text: 'Could not retrieve tools.',
|
|
||||||
}),
|
|
||||||
expect.any(Number),
|
|
||||||
);
|
|
||||||
expect(commandResult).toEqual({ type: 'handled' });
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should display only Gemini CLI tools (filtering out MCP tools)', async () => {
|
|
||||||
// Create mock tools - some with serverName property (MCP tools) and some without (Gemini CLI tools)
|
|
||||||
const mockTools = [
|
|
||||||
{ name: 'tool1', displayName: 'Tool1' },
|
|
||||||
{ name: 'tool2', displayName: 'Tool2' },
|
|
||||||
{ name: 'mcp_tool1', serverName: 'mcp-server1' },
|
|
||||||
{ name: 'mcp_tool2', serverName: 'mcp-server1' },
|
|
||||||
];
|
|
||||||
|
|
||||||
mockConfig = {
|
|
||||||
...mockConfig,
|
|
||||||
getToolRegistry: vi.fn().mockResolvedValue({
|
|
||||||
getAllTools: vi.fn().mockReturnValue(mockTools),
|
|
||||||
}),
|
|
||||||
} as unknown as Config;
|
|
||||||
|
|
||||||
const { handleSlashCommand } = getProcessor();
|
|
||||||
let commandResult: SlashCommandProcessorResult | false = false;
|
|
||||||
await act(async () => {
|
|
||||||
commandResult = await handleSlashCommand('/tools');
|
|
||||||
});
|
|
||||||
|
|
||||||
// Should only show tool1 and tool2, not the MCP tools
|
|
||||||
const message = mockAddItem.mock.calls[1][0].text;
|
|
||||||
expect(message).toContain('Tool1');
|
|
||||||
expect(message).toContain('Tool2');
|
|
||||||
expect(commandResult).toEqual({ type: 'handled' });
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should display a message when no Gemini CLI tools are available', async () => {
|
|
||||||
// Only MCP tools available
|
|
||||||
const mockTools = [
|
|
||||||
{ name: 'mcp_tool1', serverName: 'mcp-server1' },
|
|
||||||
{ name: 'mcp_tool2', serverName: 'mcp-server1' },
|
|
||||||
];
|
|
||||||
|
|
||||||
mockConfig = {
|
|
||||||
...mockConfig,
|
|
||||||
getToolRegistry: vi.fn().mockResolvedValue({
|
|
||||||
getAllTools: vi.fn().mockReturnValue(mockTools),
|
|
||||||
}),
|
|
||||||
} as unknown as Config;
|
|
||||||
|
|
||||||
const { handleSlashCommand } = getProcessor();
|
|
||||||
let commandResult: SlashCommandProcessorResult | false = false;
|
|
||||||
await act(async () => {
|
|
||||||
commandResult = await handleSlashCommand('/tools');
|
|
||||||
});
|
|
||||||
|
|
||||||
const message = mockAddItem.mock.calls[1][0].text;
|
|
||||||
expect(message).toContain('No tools available');
|
|
||||||
expect(commandResult).toEqual({ type: 'handled' });
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should display tool descriptions when /tools desc is used', async () => {
|
|
||||||
const mockTools = [
|
|
||||||
{
|
|
||||||
name: 'tool1',
|
|
||||||
displayName: 'Tool1',
|
|
||||||
description: 'Description for Tool1',
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: 'tool2',
|
|
||||||
displayName: 'Tool2',
|
|
||||||
description: 'Description for Tool2',
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
mockConfig = {
|
|
||||||
...mockConfig,
|
|
||||||
getToolRegistry: vi.fn().mockResolvedValue({
|
|
||||||
getAllTools: vi.fn().mockReturnValue(mockTools),
|
|
||||||
}),
|
|
||||||
} as unknown as Config;
|
|
||||||
|
|
||||||
const { handleSlashCommand } = getProcessor();
|
|
||||||
let commandResult: SlashCommandProcessorResult | false = false;
|
|
||||||
await act(async () => {
|
|
||||||
commandResult = await handleSlashCommand('/tools desc');
|
|
||||||
});
|
|
||||||
|
|
||||||
const message = mockAddItem.mock.calls[1][0].text;
|
|
||||||
expect(message).toContain('Tool1');
|
|
||||||
expect(message).toContain('Description for Tool1');
|
|
||||||
expect(message).toContain('Tool2');
|
|
||||||
expect(message).toContain('Description for Tool2');
|
|
||||||
expect(commandResult).toEqual({ type: 'handled' });
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|
|
@ -66,7 +66,6 @@ export const useSlashCommandProcessor = (
|
||||||
openAuthDialog: () => void,
|
openAuthDialog: () => void,
|
||||||
openEditorDialog: () => void,
|
openEditorDialog: () => void,
|
||||||
toggleCorgiMode: () => void,
|
toggleCorgiMode: () => void,
|
||||||
showToolDescriptions: boolean = false,
|
|
||||||
setQuittingMessages: (message: HistoryItem[]) => void,
|
setQuittingMessages: (message: HistoryItem[]) => void,
|
||||||
openPrivacyNotice: () => void,
|
openPrivacyNotice: () => void,
|
||||||
) => {
|
) => {
|
||||||
|
@ -205,80 +204,6 @@ export const useSlashCommandProcessor = (
|
||||||
description: 'set external editor preference',
|
description: 'set external editor preference',
|
||||||
action: (_mainCommand, _subCommand, _args) => openEditorDialog(),
|
action: (_mainCommand, _subCommand, _args) => openEditorDialog(),
|
||||||
},
|
},
|
||||||
{
|
|
||||||
name: 'tools',
|
|
||||||
description: 'list available Gemini CLI tools',
|
|
||||||
action: async (_mainCommand, _subCommand, _args) => {
|
|
||||||
// Check if the _subCommand includes a specific flag to control description visibility
|
|
||||||
let useShowDescriptions = showToolDescriptions;
|
|
||||||
if (_subCommand === 'desc' || _subCommand === 'descriptions') {
|
|
||||||
useShowDescriptions = true;
|
|
||||||
} else if (
|
|
||||||
_subCommand === 'nodesc' ||
|
|
||||||
_subCommand === 'nodescriptions'
|
|
||||||
) {
|
|
||||||
useShowDescriptions = false;
|
|
||||||
} else if (_args === 'desc' || _args === 'descriptions') {
|
|
||||||
useShowDescriptions = true;
|
|
||||||
} else if (_args === 'nodesc' || _args === 'nodescriptions') {
|
|
||||||
useShowDescriptions = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const toolRegistry = await config?.getToolRegistry();
|
|
||||||
const tools = toolRegistry?.getAllTools();
|
|
||||||
if (!tools) {
|
|
||||||
addMessage({
|
|
||||||
type: MessageType.ERROR,
|
|
||||||
content: 'Could not retrieve tools.',
|
|
||||||
timestamp: new Date(),
|
|
||||||
});
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Filter out MCP tools by checking if they have a serverName property
|
|
||||||
const geminiTools = tools.filter((tool) => !('serverName' in tool));
|
|
||||||
|
|
||||||
let message = 'Available Gemini CLI tools:\n\n';
|
|
||||||
|
|
||||||
if (geminiTools.length > 0) {
|
|
||||||
geminiTools.forEach((tool) => {
|
|
||||||
if (useShowDescriptions && tool.description) {
|
|
||||||
// Format tool name in cyan using simple ANSI cyan color
|
|
||||||
message += ` - \u001b[36m${tool.displayName} (${tool.name})\u001b[0m:\n`;
|
|
||||||
|
|
||||||
// Apply green color to the description text
|
|
||||||
const greenColor = '\u001b[32m';
|
|
||||||
const resetColor = '\u001b[0m';
|
|
||||||
|
|
||||||
// Handle multi-line descriptions by properly indenting and preserving formatting
|
|
||||||
const descLines = tool.description.trim().split('\n');
|
|
||||||
|
|
||||||
// If there are multiple lines, add proper indentation for each line
|
|
||||||
if (descLines) {
|
|
||||||
for (const descLine of descLines) {
|
|
||||||
message += ` ${greenColor}${descLine}${resetColor}\n`;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// Use cyan color for the tool name even when not showing descriptions
|
|
||||||
message += ` - \u001b[36m${tool.displayName}\u001b[0m\n`;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
message += ' No tools available\n';
|
|
||||||
}
|
|
||||||
message += '\n';
|
|
||||||
|
|
||||||
// Make sure to reset any ANSI formatting at the end to prevent it from affecting the terminal
|
|
||||||
message += '\u001b[0m';
|
|
||||||
|
|
||||||
addMessage({
|
|
||||||
type: MessageType.INFO,
|
|
||||||
content: message,
|
|
||||||
timestamp: new Date(),
|
|
||||||
});
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
name: 'corgi',
|
name: 'corgi',
|
||||||
action: (_mainCommand, _subCommand, _args) => {
|
action: (_mainCommand, _subCommand, _args) => {
|
||||||
|
@ -503,7 +428,6 @@ export const useSlashCommandProcessor = (
|
||||||
openEditorDialog,
|
openEditorDialog,
|
||||||
toggleCorgiMode,
|
toggleCorgiMode,
|
||||||
config,
|
config,
|
||||||
showToolDescriptions,
|
|
||||||
session,
|
session,
|
||||||
gitService,
|
gitService,
|
||||||
loadHistory,
|
loadHistory,
|
||||||
|
|
Loading…
Reference in New Issue