rig_volcengine/
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/Completion/Embedding
4//! - completion.rs: Chat completion model (OpenAI-compatible)
5//! - embedding.rs: Text embeddings (OpenAI-compatible)
6//! - streaming.rs: Streaming (OpenAI-compatible, same flags)
7//! - types/mod.rs: Internal shared types (error response, tool choice mapping)
8
9pub mod client;
10pub mod completion;
11pub mod embedding;
12pub mod streaming;
13pub mod types;
14
15pub use client::Client;
16pub use completion::CompletionModel;
17pub use embedding::{EmbeddingModel, TEXT_DOUBAO_EMBEDDING, TEXT_DOUBAO_EMBEDDING_LARGE};
18
19use rig::impl_conversion_traits;
20
21// Constants (aligned with original single-file version)
22pub const VOLCENGINE_API_BASE_URL: &str = "https://ark.cn-beijing.volces.com/api/v3";
23pub const DOUBAO_SEED: &str = "Doubao-Seed-1.6";
24
25// Keep conversion traits consistent with other providers
26impl_conversion_traits!(
27    AsTranscription,
28    AsImageGeneration,
29    AsAudioGeneration for Client<T>
30);