33 * Licensed under the MIT License. See License.txt in the project root for license information.
44 *--------------------------------------------------------------------------------------------*/
55
6+ import { Codicon } from '../../../../../base/common/codicons.js' ;
67import { Disposable } from '../../../../../base/common/lifecycle.js' ;
8+ import { ThemeIcon } from '../../../../../base/common/themables.js' ;
9+ import { localize } from '../../../../../nls.js' ;
710import { IConfigurationService } from '../../../../../platform/configuration/common/configuration.js' ;
811import { IInstantiationService } from '../../../../../platform/instantiation/common/instantiation.js' ;
912import { registerWorkbenchContribution2 , WorkbenchPhase , type IWorkbenchContribution } from '../../../../common/contributions.js' ;
10- import { ILanguageModelToolsService } from '../../../chat/common/languageModelToolsService.js' ;
13+ import { ILanguageModelToolsService , ToolDataSource } from '../../../chat/common/languageModelToolsService.js' ;
1114import { GetTerminalOutputTool , GetTerminalOutputToolData } from './getTerminalOutputTool.js' ;
1215import { RunInTerminalTool , RunInTerminalToolData } from './runInTerminalTool.js' ;
1316
@@ -31,6 +34,13 @@ class ChatAgentToolsContribution extends Disposable implements IWorkbenchContrib
3134 const getTerminalOutputTool = instantiationService . createInstance ( GetTerminalOutputTool ) ;
3235 this . _register ( toolsService . registerToolData ( GetTerminalOutputToolData ) ) ;
3336 this . _register ( toolsService . registerToolImplementation ( GetTerminalOutputToolData . id , getTerminalOutputTool ) ) ;
37+
38+ const toolSet = this . _register ( toolsService . createToolSet ( ToolDataSource . Internal , 'runCommands' , 'runCommands' , {
39+ icon : ThemeIcon . fromId ( Codicon . terminal . id ) ,
40+ description : localize ( 'toolset.runCommands' , 'Runs commands in the terminal' )
41+ } ) ) ;
42+ toolSet . addTool ( RunInTerminalToolData ) ;
43+ toolSet . addTool ( GetTerminalOutputToolData ) ;
3444 }
3545}
3646registerWorkbenchContribution2 ( ChatAgentToolsContribution . ID , ChatAgentToolsContribution , WorkbenchPhase . AfterRestored ) ;
0 commit comments