diff --git a/src/agent/mod.rs b/src/agent/mod.rs index 93d1222..89406ef 100644 --- a/src/agent/mod.rs +++ b/src/agent/mod.rs @@ -7,7 +7,7 @@ pub mod prompt; #[allow(unused_imports)] pub use agent::{Agent, AgentBuilder}; -pub use loop_::run; +pub use loop_::{process_message, run}; #[cfg(test)] mod tests { @@ -18,6 +18,7 @@ mod tests { #[test] fn run_function_is_reexported() { assert_reexport_exists(run); + assert_reexport_exists(process_message); assert_reexport_exists(loop_::run); assert_reexport_exists(loop_::process_message); } diff --git a/src/observability/mod.rs b/src/observability/mod.rs index 89284c1..1093a4e 100644 --- a/src/observability/mod.rs +++ b/src/observability/mod.rs @@ -6,9 +6,11 @@ pub mod traits; pub mod verbose; pub use self::log::LogObserver; +pub use self::multi::MultiObserver; pub use noop::NoopObserver; pub use otel::OtelObserver; pub use traits::{Observer, ObserverEvent}; +pub use verbose::VerboseObserver; use crate::config::ObservabilityConfig;