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 pipeline;
11mod streaming;
12mod tools;
13mod tracing_isolation;
14
15pub use completion::{MockCompletionModel, MockError, MockTurn};
16pub use embeddings::{MockEmbeddingModel, MockMultiTextDocument, MockTextDocument};
17pub use http::{
18    CapturedHttpRequest, MockHttpResponse, MockStreamingClient, RecordingHttpClient,
19    SequencedStreamingHttpClient,
20};
21pub use memory::{AppendFailingMemory, CountingMemory, FailingMemory};
22pub use model_listing::MockModelLister;
23pub use pipeline::{Foo, MockPromptModel, MockVectorStoreIndex};
24pub use streaming::{MockResponse, MockStreamEvent};
25pub use tools::{
26    BarrierMockToolIndex, MockAddTool, MockBarrierTool, MockControlledTool, MockExampleTool,
27    MockImageGeneratorTool, MockImageOutputTool, MockObjectOutputTool, MockOperationArgs,
28    MockStringOutputTool, MockSubtractTool, MockToolError, MockToolIndex, mock_math_toolset,
29};
30pub use tracing_isolation::{
31    scoped_tracing_subscriber_guard, scoped_tracing_subscriber_guard_blocking,
32};
rig_core/test_utils/mod.rs

rig_core/test_utils/
mod.rs
