7c7043947b
- Return unified diff from `Edit` tool so model can see the final state - Format on save if enabled - Provide `Write` tool - Disable `MultiEdit` tool - Better prompting Release Notes: - N/A
59 lines
1.4 KiB
Rust
59 lines
1.4 KiB
Rust
use acp_thread::AcpThread;
|
|
use anyhow::Result;
|
|
use context_server::{
|
|
listener::{McpServerTool, ToolResponse},
|
|
types::ToolAnnotations,
|
|
};
|
|
use gpui::{AsyncApp, WeakEntity};
|
|
|
|
use crate::tools::WriteToolParams;
|
|
|
|
#[derive(Clone)]
|
|
pub struct WriteTool {
|
|
thread_rx: watch::Receiver<WeakEntity<AcpThread>>,
|
|
}
|
|
|
|
impl WriteTool {
|
|
pub fn new(thread_rx: watch::Receiver<WeakEntity<AcpThread>>) -> Self {
|
|
Self { thread_rx }
|
|
}
|
|
}
|
|
|
|
impl McpServerTool for WriteTool {
|
|
type Input = WriteToolParams;
|
|
type Output = ();
|
|
|
|
const NAME: &'static str = "Write";
|
|
|
|
fn annotations(&self) -> ToolAnnotations {
|
|
ToolAnnotations {
|
|
title: Some("Write file".to_string()),
|
|
read_only_hint: Some(false),
|
|
destructive_hint: Some(false),
|
|
open_world_hint: Some(false),
|
|
idempotent_hint: Some(false),
|
|
}
|
|
}
|
|
|
|
async fn run(
|
|
&self,
|
|
input: Self::Input,
|
|
cx: &mut AsyncApp,
|
|
) -> Result<ToolResponse<Self::Output>> {
|
|
let mut thread_rx = self.thread_rx.clone();
|
|
let Some(thread) = thread_rx.recv().await?.upgrade() else {
|
|
anyhow::bail!("Thread closed");
|
|
};
|
|
|
|
thread
|
|
.update(cx, |thread, cx| {
|
|
thread.write_text_file(input.abs_path, input.content, cx)
|
|
})?
|
|
.await?;
|
|
|
|
Ok(ToolResponse {
|
|
content: vec![],
|
|
structured_content: (),
|
|
})
|
|
}
|
|
}
|