rig_tei/
lib.rs

1//! Category: mod.rs (top-level module and constant exports)
2//! Structure mirrors rig-bedrock:
3//! - client.rs: Client and Builder; implements Provider/Verify/Embedding
4//! - embedding.rs: Text embeddings
5//! - rerank.rs: Text reranking
6//! - predict.rs: Classification/prediction
7
8pub mod client;
9pub mod embedding;
10pub mod predict;
11pub mod rerank;
12
13pub use client::{Client, Endpoints};
14pub use embedding::EmbeddingModel;
15pub use predict::{LabelScore, PredictError, PredictResponse};
16pub use rerank::{RerankError, RerankResult};
17
18use rig::impl_conversion_traits;
19
20// Default local TEI base URL
21pub const TEI_DEFAULT_BASE_URL: &str = "http://127.0.0.1:8080";
22
23// Keep conversion traits consistent with original single-file version
24impl_conversion_traits!(
25    AsCompletion,
26    AsTranscription,
27    AsImageGeneration,
28    AsAudioGeneration for Client<T>
29);