Skip to main content

rig_core/test_utils/
mod.rs

1//! Test utilities for deterministic completion-model tests.
2
3mod completion;
4mod embeddings;
5mod http;
6#[cfg(test)]
7pub(crate) mod internal_streaming_profiles;
8mod memory;
9mod model_listing;
10mod streaming;
11mod tracing_isolation;
12
13pub use completion::{MockCompletionModel, MockError, MockTurn};
14pub use embeddings::{MockEmbeddingModel, MockMultiTextDocument, MockTextDocument};
15pub use http::{
16    CapturedHttpRequest, HttpErrorStreamingClient, MockHttpResponse, MockStreamingClient,
17    RecordingHttpClient, SequencedHttpClient, SequencedStreamingHttpClient,
18};
19pub use memory::{AppendFailingMemory, CountingMemory, FailingMemory};
20pub use model_listing::MockModelLister;
21pub use streaming::{MockResponse, MockStreamEvent};
22pub use tracing_isolation::{
23    scoped_tracing_subscriber_guard, scoped_tracing_subscriber_guard_blocking,
24};