diff --git a/crates/agent2/src/prompts.rs b/crates/agent2/src/prompts.rs index ab30b117a8..015f56f4db 100644 --- a/crates/agent2/src/prompts.rs +++ b/crates/agent2/src/prompts.rs @@ -7,11 +7,11 @@ use gpui::{App, Entity}; use project::Project; #[allow(dead_code)] -struct _BasePrompt { +struct BasePrompt { project: Entity, } -impl Prompt for _BasePrompt { +impl Prompt for BasePrompt { fn render(&self, templates: &Templates, cx: &App) -> Result { BaseTemplate { os: std::env::consts::OS.to_string(), diff --git a/crates/agent2/src/tests/mod.rs b/crates/agent2/src/tests/mod.rs index 42078a3ffa..ac790c8498 100644 --- a/crates/agent2/src/tests/mod.rs +++ b/crates/agent2/src/tests/mod.rs @@ -1,7 +1,6 @@ use super::*; use crate::templates::Templates; use client::{Client, UserStore}; -use fs::FakeFs; use gpui::{AppContext, Entity, TestAppContext}; use language_model::{ LanguageModel, LanguageModelCompletionError, LanguageModelCompletionEvent, @@ -209,9 +208,6 @@ struct AgentTest { async fn setup(cx: &mut TestAppContext) -> AgentTest { cx.executor().allow_parking(); cx.update(settings::init); - let fs = FakeFs::new(cx.executor().clone()); - // let project = Project::test(fs.clone(), [], cx).await; - // let action_log = cx.new(|_| ActionLog::new(project.clone())); let templates = Templates::new(); let agent = cx.new(|_| Thread::new(templates)); @@ -236,7 +232,7 @@ async fn setup(cx: &mut TestAppContext) -> AgentTest { let provider = models.provider(&model.provider_id()).unwrap(); let authenticated = provider.authenticate(cx); - cx.spawn(async move |cx| { + cx.spawn(async move |_cx| { authenticated.await.unwrap(); model }) diff --git a/crates/agent2/src/thread.rs b/crates/agent2/src/thread.rs index 85e2030f8b..bc88cf1d95 100644 --- a/crates/agent2/src/thread.rs +++ b/crates/agent2/src/thread.rs @@ -35,7 +35,7 @@ pub struct Thread { /// we run tools, report their results. running_turn: Option>, system_prompts: Vec>, - tools: BTreeMap>, + tools: BTreeMap>, templates: Arc, // project: Entity, // action_log: Entity, @@ -390,21 +390,21 @@ where /// Runs the tool with the provided input. fn run(self: Arc, input: Self::Input, cx: &mut App) -> Task>; - fn erase(self) -> Arc { + fn erase(self) -> Arc { Arc::new(Erased(Arc::new(self))) } } pub struct Erased(T); -pub trait AgentToolErased { +pub trait AnyAgentTool { fn name(&self) -> SharedString; fn description(&self, cx: &mut App) -> SharedString; fn input_schema(&self, format: LanguageModelToolSchemaFormat) -> Result; fn run(self: Arc, input: serde_json::Value, cx: &mut App) -> Task>; } -impl AgentToolErased for Erased> +impl AnyAgentTool for Erased> where T: AgentTool, {