assistant_tools.rs

 1mod batch_tool;
 2mod code_action_tool;
 3mod code_symbols_tool;
 4mod copy_path_tool;
 5mod create_directory_tool;
 6mod create_file_tool;
 7mod delete_path_tool;
 8mod diagnostics_tool;
 9mod fetch_tool;
10mod find_replace_file_tool;
11mod list_directory_tool;
12mod move_path_tool;
13mod now_tool;
14mod open_tool;
15mod path_search_tool;
16mod read_file_tool;
17mod regex_search_tool;
18mod rename_tool;
19mod replace;
20mod schema;
21mod symbol_info_tool;
22mod terminal_tool;
23mod thinking_tool;
24
25use std::sync::Arc;
26
27use assistant_tool::ToolRegistry;
28use copy_path_tool::CopyPathTool;
29use gpui::App;
30use http_client::HttpClientWithUrl;
31use move_path_tool::MovePathTool;
32
33use crate::batch_tool::BatchTool;
34use crate::code_action_tool::CodeActionTool;
35use crate::code_symbols_tool::CodeSymbolsTool;
36use crate::create_directory_tool::CreateDirectoryTool;
37use crate::create_file_tool::CreateFileTool;
38use crate::delete_path_tool::DeletePathTool;
39use crate::diagnostics_tool::DiagnosticsTool;
40use crate::fetch_tool::FetchTool;
41use crate::find_replace_file_tool::FindReplaceFileTool;
42use crate::list_directory_tool::ListDirectoryTool;
43use crate::now_tool::NowTool;
44use crate::open_tool::OpenTool;
45use crate::path_search_tool::PathSearchTool;
46use crate::read_file_tool::ReadFileTool;
47use crate::regex_search_tool::RegexSearchTool;
48use crate::rename_tool::RenameTool;
49use crate::symbol_info_tool::SymbolInfoTool;
50use crate::terminal_tool::TerminalTool;
51use crate::thinking_tool::ThinkingTool;
52
53pub fn init(http_client: Arc<HttpClientWithUrl>, cx: &mut App) {
54    assistant_tool::init(cx);
55
56    let registry = ToolRegistry::global(cx);
57    registry.register_tool(TerminalTool);
58    registry.register_tool(BatchTool);
59    registry.register_tool(CreateDirectoryTool);
60    registry.register_tool(CreateFileTool);
61    registry.register_tool(CopyPathTool);
62    registry.register_tool(DeletePathTool);
63    registry.register_tool(FindReplaceFileTool);
64    registry.register_tool(SymbolInfoTool);
65    registry.register_tool(CodeActionTool);
66    registry.register_tool(MovePathTool);
67    registry.register_tool(DiagnosticsTool);
68    registry.register_tool(ListDirectoryTool);
69    registry.register_tool(NowTool);
70    registry.register_tool(OpenTool);
71    registry.register_tool(CodeSymbolsTool);
72    registry.register_tool(PathSearchTool);
73    registry.register_tool(ReadFileTool);
74    registry.register_tool(RegexSearchTool);
75    registry.register_tool(RenameTool);
76    registry.register_tool(ThinkingTool);
77    registry.register_tool(FetchTool::new(http_client));
78}