1use agent::FindPathToolInput;
 2use agent_settings::AgentProfileId;
 3use anyhow::Result;
 4use async_trait::async_trait;
 5use regex::Regex;
 6
 7use crate::example::{Example, ExampleContext, ExampleMetadata};
 8
 9pub struct FileSearchExample;
10
11#[async_trait(?Send)]
12impl Example for FileSearchExample {
13    fn meta(&self) -> ExampleMetadata {
14        ExampleMetadata {
15            name: "file_search".to_string(),
16            url: "https://github.com/zed-industries/zed.git".to_string(),
17            revision: "03ecb88fe30794873f191ddb728f597935b3101c".to_string(),
18            language_server: None,
19            max_assertions: Some(3),
20            profile_id: AgentProfileId::default(),
21            existing_thread_json: None,
22            max_turns: None,
23        }
24    }
25
26    async fn conversation(&self, cx: &mut ExampleContext) -> Result<()> {
27        const FILENAME: &str = "find_replace_file_tool.rs";
28        cx.push_user_message(format!(
29                r#"
30        Look at the `{FILENAME}`. I want to implement a card for it. The card should implement the `Render` trait.
31
32        The card should show a diff. It should be a beautifully presented diff. The card "box" should look like what we show for
33        markdown codeblocks (look at `MarkdownElement`). I want to see a red background for lines that were deleted and a green
34        background for lines that were added. We should have a div per diff line.
35        "#
36        ));
37
38        let response = cx.run_turn().await?;
39        let tool_use = response.expect_tool("find_path", cx)?;
40        let input = tool_use.parse_input::<FindPathToolInput>()?;
41
42        let glob = input.glob;
43        cx.assert(glob.ends_with(FILENAME), "glob ends with file name")?;
44
45        let without_filename = glob.replace(FILENAME, "");
46        let matches = Regex::new("(\\*\\*|zed)/(\\*\\*?/)?")
47            .unwrap()
48            .is_match(&without_filename);
49
50        cx.assert(matches, "glob starts with `**` or project")?;
51
52        Ok(())
53    }
54}