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