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

rig_core/test_utils/
mod.rs
