use super::edit_file_tool::{SensitiveSettingsKind, sensitive_settings_kind};use agent_client_protocol::ToolKind;use agent_settings::AgentSettings;use anyhow::{Context as _, Result, anyhow};use futures::FutureExt as _;use gpui::{App, Entity, SharedString, Task};use project::Project;use schemars::JsonSchema;use serde::{Deserialize, Serialize};use settings::Settings;use std::path::Path;use std::sync::Arc;use util::markdown::MarkdownInlineCode; use crate::{AgentTool, ToolCallEventStream, ToolPermissionDecision, decide_permission_for_path}; /// Creates a new directory at the specified path within the project. Returns confirmation that the directory was created.////// This tool creates a directory and all necessary parent directories. It should be used whenever you need to create new directories within the project.#[derive(Debug, Serialize, Deserialize, JsonSchema)]pub struct CreateDirectoryToolInput { /// The path of the new directory. /// /// <example> /// If the project has the following structure: /// /// - directory1/ /// - directory2/ /// /// You can create a new directory by providing a path of "directory1/new_directory" /// </example> pub path: String,} pub struct CreateDirectoryTool { project: Entity<Project>,} impl CreateDirectoryTool { pub fn new(project: Entity<Project>) -> Self { Self { project } }} impl AgentTool for CreateDirectoryTool { type Input = CreateDirectoryToolInput; type Output = String; const NAME: &'static str = "create_directory"; fn kind() -> ToolKind { ToolKind::Read } fn initial_title( &self, input: Result<Self::Input, serde_json::Value>, _cx: &mut App, ) -> SharedString { if let Ok(input) = input { format!("Create directory {}", MarkdownInlineCode(&input.path)).into() } else { "Create directory".into() } } fn run( self: Arc<Self>, input: Self::Input, event_stream: ToolCallEventStream, cx: &mut App, ) -> Task<Result<Self::Output>> { let settings = AgentSettings::get_global(cx); let mut decision = decide_permission_for_path(Self::NAME, &input.path, settings); let sensitive_kind = sensitive_settings_kind(Path::new(&input.path)); if matches!(decision, ToolPermissionDecision::Allow) && sensitive_kind.is_some() { decision = ToolPermissionDecision::Confirm; } let authorize = match decision { ToolPermissionDecision::Allow => None, ToolPermissionDecision::Deny(reason) => { return Task::ready(Err(anyhow!("{}", reason))); } ToolPermissionDecision::Confirm => { let title = format!("Create directory {}", MarkdownInlineCode(&input.path)); let title = match &sensitive_kind { Some(SensitiveSettingsKind::Local) => format!("{title} (local settings)"), Some(SensitiveSettingsKind::Global) => format!("{title} (settings)"), None => title, }; let context = crate::ToolPermissionContext { tool_name: Self::NAME.to_string(), input_values: vec![input.path.clone()], }; Some(event_stream.authorize(title, context, cx)) } }; let destination_path: Arc<str> = input.path.as_str().into(); let project = self.project.clone(); cx.spawn(async move |cx| { if let Some(authorize) = authorize { authorize.await?; } let create_entry = project.update(cx, |project, cx| { match project.find_project_path(&input.path, cx) { Some(project_path) => Ok(project.create_entry(project_path, true, cx)), None => Err(anyhow!("Path to create was outside the project")), } })?; futures::select! { result = create_entry.fuse() => { result.with_context(|| format!("Creating directory {destination_path}"))?; } _ = event_stream.cancelled_by_user().fuse() => { anyhow::bail!("Create directory cancelled by user"); } } Ok(format!("Created directory {destination_path}")) }) }}
use super::edit_file_tool::{SensitiveSettingsKind, sensitive_settings_kind};
use agent_client_protocol::ToolKind;
use agent_settings::AgentSettings;
use anyhow::{Context as _, Result, anyhow};
use futures::FutureExt as _;
use gpui::{App, Entity, SharedString, Task};
use project::Project;
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
use settings::Settings;
use std::path::Path;
use std::sync::Arc;
use util::markdown::MarkdownInlineCode;
use crate::{AgentTool, ToolCallEventStream, ToolPermissionDecision, decide_permission_for_path};
/// Creates a new directory at the specified path within the project. Returns confirmation that the directory was created.
///
/// This tool creates a directory and all necessary parent directories. It should be used whenever you need to create new directories within the project.
#[derive(Debug, Serialize, Deserialize, JsonSchema)]
pub struct CreateDirectoryToolInput {
/// The path of the new directory.
///
/// <example>
/// If the project has the following structure:
///
/// - directory1/
/// - directory2/
///
/// You can create a new directory by providing a path of "directory1/new_directory"
/// </example>
pub path: String,
}
pub struct CreateDirectoryTool {
project: Entity<Project>,
}
impl CreateDirectoryTool {
pub fn new(project: Entity<Project>) -> Self {
Self { project }
}
}
impl AgentTool for CreateDirectoryTool {
type Input = CreateDirectoryToolInput;
type Output = String;
const NAME: &'static str = "create_directory";
fn kind() -> ToolKind {
ToolKind::Read
}
fn initial_title(
&self,
input: Result<Self::Input, serde_json::Value>,
_cx: &mut App,
) -> SharedString {
if let Ok(input) = input {
format!("Create directory {}", MarkdownInlineCode(&input.path)).into()
} else {
"Create directory".into()
}
}
fn run(
self: Arc<Self>,
input: Self::Input,
event_stream: ToolCallEventStream,
cx: &mut App,
) -> Task<Result<Self::Output>> {
let settings = AgentSettings::get_global(cx);
let mut decision = decide_permission_for_path(Self::NAME, &input.path, settings);
let sensitive_kind = sensitive_settings_kind(Path::new(&input.path));
if matches!(decision, ToolPermissionDecision::Allow) && sensitive_kind.is_some() {
decision = ToolPermissionDecision::Confirm;
}
let authorize = match decision {
ToolPermissionDecision::Allow => None,
ToolPermissionDecision::Deny(reason) => {
return Task::ready(Err(anyhow!("{}", reason)));
}
ToolPermissionDecision::Confirm => {
let title = format!("Create directory {}", MarkdownInlineCode(&input.path));
let title = match &sensitive_kind {
Some(SensitiveSettingsKind::Local) => format!("{title} (local settings)"),
Some(SensitiveSettingsKind::Global) => format!("{title} (settings)"),
None => title,
};
let context = crate::ToolPermissionContext {
tool_name: Self::NAME.to_string(),
input_values: vec![input.path.clone()],
};
Some(event_stream.authorize(title, context, cx))
}
};
let destination_path: Arc<str> = input.path.as_str().into();
let project = self.project.clone();
cx.spawn(async move |cx| {
if let Some(authorize) = authorize {
authorize.await?;
}
let create_entry = project.update(cx, |project, cx| {
match project.find_project_path(&input.path, cx) {
Some(project_path) => Ok(project.create_entry(project_path, true, cx)),
None => Err(anyhow!("Path to create was outside the project")),
}
})?;
futures::select! {
result = create_entry.fuse() => {
result.with_context(|| format!("Creating directory {destination_path}"))?;
}
_ = event_stream.cancelled_by_user().fuse() => {
anyhow::bail!("Create directory cancelled by user");
}
}
Ok(format!("Created directory {destination_path}"))
})
}
}