diff --git a/app/src/ai/mod.rs b/app/src/ai/mod.rs
index c91d8b11..f39b6be3 100644
--- a/app/src/ai/mod.rs
+++ b/app/src/ai/mod.rs
@@ -17,6 +17,7 @@ pub(crate) mod attachment_utils;
 pub mod aws_credentials;
 #[cfg(not(target_family = "wasm"))]
 pub mod bedrock;
+#[allow(dead_code)]
 pub mod prompt_builder;
 pub(crate) mod block_context;
 pub(crate) mod blocklist;
diff --git a/app/src/ai/prompt_builder/mod.rs b/app/src/ai/prompt_builder/mod.rs
index f97219b7..bb664fb4 100644
--- a/app/src/ai/prompt_builder/mod.rs
+++ b/app/src/ai/prompt_builder/mod.rs
@@ -23,6 +23,7 @@ mod tools;
 pub use context::PromptContext;
 pub use mode::Mode;
 pub use prompts::provider::Provider;
+#[allow(unused_imports)]
 pub use tools::tools_for_mode;
 
 use crate::ai::bedrock::convert::ToolDefinition;