1mod batch_tool;
  2mod code_action_tool;
  3mod code_symbols_tool;
  4mod contents_tool;
  5mod copy_path_tool;
  6mod create_directory_tool;
  7mod create_file_tool;
  8mod delete_path_tool;
  9mod diagnostics_tool;
 10mod edit_file_tool;
 11mod fetch_tool;
 12mod grep_tool;
 13mod list_directory_tool;
 14mod move_path_tool;
 15mod now_tool;
 16mod open_tool;
 17mod path_search_tool;
 18mod read_file_tool;
 19mod rename_tool;
 20mod replace;
 21mod schema;
 22mod symbol_info_tool;
 23mod terminal_tool;
 24mod thinking_tool;
 25mod ui;
 26mod web_search_tool;
 27
 28use std::sync::Arc;
 29
 30use assistant_tool::ToolRegistry;
 31use copy_path_tool::CopyPathTool;
 32use gpui::App;
 33use http_client::HttpClientWithUrl;
 34use language_model::LanguageModelRegistry;
 35use move_path_tool::MovePathTool;
 36use web_search_tool::WebSearchTool;
 37
 38use crate::batch_tool::BatchTool;
 39use crate::code_action_tool::CodeActionTool;
 40use crate::code_symbols_tool::CodeSymbolsTool;
 41use crate::contents_tool::ContentsTool;
 42use crate::create_directory_tool::CreateDirectoryTool;
 43use crate::create_file_tool::CreateFileTool;
 44use crate::delete_path_tool::DeletePathTool;
 45use crate::diagnostics_tool::DiagnosticsTool;
 46use crate::edit_file_tool::EditFileTool;
 47use crate::fetch_tool::FetchTool;
 48use crate::grep_tool::GrepTool;
 49use crate::list_directory_tool::ListDirectoryTool;
 50use crate::now_tool::NowTool;
 51use crate::open_tool::OpenTool;
 52use crate::path_search_tool::PathSearchTool;
 53use crate::read_file_tool::ReadFileTool;
 54use crate::rename_tool::RenameTool;
 55use crate::symbol_info_tool::SymbolInfoTool;
 56use crate::terminal_tool::TerminalTool;
 57use crate::thinking_tool::ThinkingTool;
 58
 59pub use path_search_tool::PathSearchToolInput;
 60
 61pub fn init(http_client: Arc<HttpClientWithUrl>, cx: &mut App) {
 62    assistant_tool::init(cx);
 63
 64    let registry = ToolRegistry::global(cx);
 65    registry.register_tool(TerminalTool);
 66    registry.register_tool(BatchTool);
 67    registry.register_tool(CreateDirectoryTool);
 68    registry.register_tool(CreateFileTool);
 69    registry.register_tool(CopyPathTool);
 70    registry.register_tool(DeletePathTool);
 71    registry.register_tool(EditFileTool);
 72    registry.register_tool(SymbolInfoTool);
 73    registry.register_tool(CodeActionTool);
 74    registry.register_tool(MovePathTool);
 75    registry.register_tool(DiagnosticsTool);
 76    registry.register_tool(ListDirectoryTool);
 77    registry.register_tool(NowTool);
 78    registry.register_tool(OpenTool);
 79    registry.register_tool(CodeSymbolsTool);
 80    registry.register_tool(ContentsTool);
 81    registry.register_tool(PathSearchTool);
 82    registry.register_tool(ReadFileTool);
 83    registry.register_tool(GrepTool);
 84    registry.register_tool(RenameTool);
 85    registry.register_tool(ThinkingTool);
 86    registry.register_tool(FetchTool::new(http_client));
 87
 88    cx.subscribe(
 89        &LanguageModelRegistry::global(cx),
 90        move |registry, event, cx| match event {
 91            language_model::Event::DefaultModelChanged => {
 92                let using_zed_provider = registry
 93                    .read(cx)
 94                    .default_model()
 95                    .map_or(false, |default| default.is_provided_by_zed());
 96                if using_zed_provider {
 97                    ToolRegistry::global(cx).register_tool(WebSearchTool);
 98                } else {
 99                    ToolRegistry::global(cx).unregister_tool(WebSearchTool);
100                }
101            }
102            _ => {}
103        },
104    )
105    .detach();
106}
107
108#[cfg(test)]
109mod tests {
110    use client::Client;
111    use clock::FakeSystemClock;
112    use http_client::FakeHttpClient;
113
114    use super::*;
115
116    #[gpui::test]
117    fn test_builtin_tool_schema_compatibility(cx: &mut App) {
118        settings::init(cx);
119
120        let client = Client::new(
121            Arc::new(FakeSystemClock::new()),
122            FakeHttpClient::with_200_response(),
123            cx,
124        );
125        language_model::init(client.clone(), cx);
126        crate::init(client.http_client(), cx);
127
128        for tool in ToolRegistry::global(cx).tools() {
129            let actual_schema = tool
130                .input_schema(language_model::LanguageModelToolSchemaFormat::JsonSchemaSubset)
131                .unwrap();
132            let mut expected_schema = actual_schema.clone();
133            assistant_tool::adapt_schema_to_format(
134                &mut expected_schema,
135                language_model::LanguageModelToolSchemaFormat::JsonSchemaSubset,
136            )
137            .unwrap();
138
139            let error_message = format!(
140                "Tool schema for `{}` is not compatible with `language_model::LanguageModelToolSchemaFormat::JsonSchemaSubset` (Gemini Models).\n\
141                Are you using `schema::json_schema_for<T>(format)` to generate the schema?",
142                tool.name(),
143            );
144
145            assert_eq!(actual_schema, expected_schema, "{}", error_message)
146        }
147    }
148}