assistant_tools.rs

 1mod bash_tool;
 2mod copy_path_tool;
 3mod create_file_tool;
 4mod delete_path_tool;
 5mod diagnostics_tool;
 6mod edit_files_tool;
 7mod fetch_tool;
 8mod find_replace_file_tool;
 9mod list_directory_tool;
10mod move_path_tool;
11mod now_tool;
12mod path_search_tool;
13mod read_file_tool;
14mod regex_search_tool;
15mod replace;
16mod thinking_tool;
17
18use std::sync::Arc;
19
20use assistant_tool::ToolRegistry;
21use copy_path_tool::CopyPathTool;
22use gpui::App;
23use http_client::HttpClientWithUrl;
24use move_path_tool::MovePathTool;
25
26use crate::bash_tool::BashTool;
27use crate::create_file_tool::CreateFileTool;
28use crate::delete_path_tool::DeletePathTool;
29use crate::diagnostics_tool::DiagnosticsTool;
30use crate::edit_files_tool::EditFilesTool;
31use crate::fetch_tool::FetchTool;
32use crate::find_replace_file_tool::FindReplaceFileTool;
33use crate::list_directory_tool::ListDirectoryTool;
34use crate::now_tool::NowTool;
35use crate::path_search_tool::PathSearchTool;
36use crate::read_file_tool::ReadFileTool;
37use crate::regex_search_tool::RegexSearchTool;
38use crate::thinking_tool::ThinkingTool;
39
40pub fn init(http_client: Arc<HttpClientWithUrl>, cx: &mut App) {
41    assistant_tool::init(cx);
42    crate::edit_files_tool::log::init(cx);
43
44    let registry = ToolRegistry::global(cx);
45    registry.register_tool(BashTool);
46    registry.register_tool(CreateFileTool);
47    registry.register_tool(CopyPathTool);
48    registry.register_tool(DeletePathTool);
49    registry.register_tool(FindReplaceFileTool);
50    registry.register_tool(MovePathTool);
51    registry.register_tool(DiagnosticsTool);
52    registry.register_tool(EditFilesTool);
53    registry.register_tool(ListDirectoryTool);
54    registry.register_tool(NowTool);
55    registry.register_tool(PathSearchTool);
56    registry.register_tool(ReadFileTool);
57    registry.register_tool(RegexSearchTool);
58    registry.register_tool(ThinkingTool);
59    registry.register_tool(FetchTool::new(http_client));
60}