assistant tools: Add Tool::icon method instead of matching on name (#27444)

Release Notes:

- N/A

crates/assistant_tools/src/bash_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::sync::Arc;
+use ui::IconName;
 use util::command::new_smol_command;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
@@ -31,6 +32,10 @@ impl Tool for BashTool {
         include_str!("./bash_tool/description.md").to_string()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Terminal
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(BashToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/copy_path_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::sync::Arc;
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct CopyPathToolInput {
@@ -47,6 +48,10 @@ impl Tool for CopyPathTool {
         include_str!("./copy_path_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Clipboard
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(CopyPathToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/create_file_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::sync::Arc;
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct CreateFileToolInput {
@@ -44,6 +45,10 @@ impl Tool for CreateFileTool {
         include_str!("./create_file_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::File
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(CreateFileToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/delete_path_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::sync::Arc;
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct DeletePathToolInput {
@@ -38,6 +39,10 @@ impl Tool for DeletePathTool {
         include_str!("./delete_path_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Trash
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(DeletePathToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/diagnostics_tool.rs

@@ -11,6 +11,7 @@ use std::{
     path::{Path, PathBuf},
     sync::Arc,
 };
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct DiagnosticsToolInput {
@@ -45,6 +46,10 @@ impl Tool for DiagnosticsTool {
         include_str!("./diagnostics_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Warning
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(DiagnosticsToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/edit_files_tool.rs

@@ -17,6 +17,7 @@ use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::fmt::Write;
 use std::sync::Arc;
+use ui::IconName;
 use util::ResultExt;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
@@ -86,6 +87,10 @@ impl Tool for EditFilesTool {
         include_str!("./edit_files_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Pencil
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(EditFilesToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/fetch_tool.rs

@@ -12,6 +12,7 @@ use language_model::LanguageModelRequestMessage;
 use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
+use ui::IconName;
 
 #[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Copy)]
 enum ContentType {
@@ -121,6 +122,10 @@ impl Tool for FetchTool {
         include_str!("./fetch_tool/description.md").to_string()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Globe
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(FetchToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/find_replace_file_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::{collections::HashSet, path::PathBuf, sync::Arc};
+use ui::IconName;
 
 use crate::replace::replace_exact;
 
@@ -135,6 +136,10 @@ impl Tool for FindReplaceFileTool {
         include_str!("find_replace_tool/description.md").to_string()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Pencil
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(FindReplaceFileToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/list_directory_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::{fmt::Write, path::Path, sync::Arc};
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct ListDirectoryToolInput {
@@ -49,6 +50,10 @@ impl Tool for ListDirectoryTool {
         include_str!("./list_directory_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Folder
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(ListDirectoryToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/move_path_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::{path::Path, sync::Arc};
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct MovePathToolInput {
@@ -47,6 +48,10 @@ impl Tool for MovePathTool {
         include_str!("./move_path_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::ArrowRightLeft
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(MovePathToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/now_tool.rs

@@ -8,6 +8,7 @@ use language_model::LanguageModelRequestMessage;
 use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 #[serde(rename_all = "snake_case")]
@@ -39,6 +40,10 @@ impl Tool for NowTool {
         "Returns the current datetime in RFC 3339 format. Only use this tool when the user specifically asks for it or the current task would benefit from knowing the current datetime.".into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Info
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(NowToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/path_search_tool.rs

@@ -6,6 +6,7 @@ use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::{path::PathBuf, sync::Arc};
+use ui::IconName;
 use util::paths::PathMatcher;
 use worktree::Snapshot;
 
@@ -47,6 +48,10 @@ impl Tool for PathSearchTool {
         include_str!("./path_search_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::SearchCode
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(PathSearchToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/read_file_tool.rs

@@ -9,6 +9,7 @@ use language_model::LanguageModelRequestMessage;
 use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct ReadFileToolInput {
@@ -52,6 +53,10 @@ impl Tool for ReadFileTool {
         include_str!("./read_file_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Eye
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(ReadFileToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/regex_search_tool.rs

@@ -11,6 +11,7 @@ use project::{
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 use std::{cmp, fmt::Write, sync::Arc};
+use ui::IconName;
 use util::paths::PathMatcher;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
@@ -49,6 +50,10 @@ impl Tool for RegexSearchTool {
         include_str!("./regex_search_tool/description.md").into()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Regex
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(RegexSearchToolInput);
         serde_json::to_value(&schema).unwrap()

crates/assistant_tools/src/thinking_tool.rs

@@ -7,6 +7,7 @@ use language_model::LanguageModelRequestMessage;
 use project::Project;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
+use ui::IconName;
 
 #[derive(Debug, Serialize, Deserialize, JsonSchema)]
 pub struct ThinkingToolInput {
@@ -30,6 +31,10 @@ impl Tool for ThinkingTool {
         include_str!("./thinking_tool/description.md").to_string()
     }
 
+    fn icon(&self) -> IconName {
+        IconName::Brain
+    }
+
     fn input_schema(&self) -> serde_json::Value {
         let schema = schemars::schema_for!(ThinkingToolInput);
         serde_json::to_value(&schema).unwrap()