1 files changed, 4 insertions, 0 deletions
diff --git a/crates/atuin-ai/src/lib.rs b/crates/atuin-ai/src/lib.rs
index 6f431179..febb488e 100644
--- a/crates/atuin-ai/src/lib.rs
+++ b/crates/atuin-ai/src/lib.rs
@@ -1,6 +1,10 @@
 pub mod commands;
 pub(crate) mod context;
+pub(crate) mod context_window;
+pub(crate) mod event_serde;
 pub(crate) mod permissions;
+pub(crate) mod session;
+pub(crate) mod store;
 pub(crate) mod stream;
 pub(crate) mod tools;
 pub(crate) mod tui;