ai_lib/
lib.rs

1//! AI-lib: A Unified AI SDK for Rust
2//!
3//! This library provides a single, consistent interface for interacting with multiple AI model providers.
4//!
5//! # Quick Start
6//!
7//! ```rust
8//! use ai_lib::{AiClient, Provider, ChatCompletionRequest, Message, Role};
9//! use ai_lib::types::common::Content;
10//!
11//! #[tokio::main]
12//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
13//!     // Switch provider by changing Provider:: value
14//!     let client = AiClient::new(Provider::Groq)?;
15//!
16//!     let request = ChatCompletionRequest::new(
17//!         "llama3-8b-8192".to_string(),
18//!         vec![Message {
19//!             role: Role::User,
20//!             content: Content::Text("Hello, how are you?".to_string()),
21//!             function_call: None,
22//!         }],
23//!     );
24//!
25//!     println!("Client created successfully with provider: {:?}", client.current_provider());
26//!     println!("Request prepared for model: {}", request.model);
27//!
28//!     Ok(())
29//! }
30//! ```
31//!
32//! # Proxy Support
33//!
34//! AI-lib supports proxy configuration via environment variables:
35//!
36//! ```bash
37//! # Set proxy server
38//! export AI_PROXY_URL=http://proxy.example.com:8080
39//!
40//! # Proxy with authentication
41//! export AI_PROXY_URL=http://username:password@proxy.example.com:8080
42//!
43//! # HTTPS proxy
44//! export AI_PROXY_URL=https://proxy.example.com:8080
45//! ```
46//!
47//! All AI provider requests will automatically use the specified proxy server.
48
49pub mod api;
50pub mod client;
51pub mod metrics;
52pub mod provider;
53pub mod transport;
54pub mod types;
55pub mod utils;
56
57// Re-export main types for user convenience
58pub use api::ChatApi;
59pub use client::{AiClient, AiClientBuilder, Provider};
60pub use types::{
61    AiLibError, ChatCompletionRequest, ChatCompletionResponse, Choice, Message, Role, Usage,
62};
63// Convenience re-exports: make the most-used types available from the crate root so
64// users don't need deep imports for common flows.
65pub use api::ChatCompletionChunk;
66pub use client::CancelHandle;
67pub use metrics::{Metrics, MetricsExt, NoopMetrics, NoopTimer, Timer};
68pub use transport::{
69    DynHttpTransport, DynHttpTransportRef, HttpClient, HttpTransport, TransportError,
70};
71pub use types::common::Content;
72
73// Re-export configuration types
74pub use provider::config::{FieldMapping, ProviderConfig};
75pub use provider::configs::ProviderConfigs;
76
77// Re-export model management tools
78pub use provider::models::{
79    CustomModelManager, LoadBalancingStrategy, ModelArray, ModelCapabilities, ModelEndpoint,
80    ModelInfo, ModelSelectionStrategy, PerformanceMetrics, PricingInfo, QualityTier, SpeedTier,
81};
82
83// Re-export batch processing functionality
84pub use api::chat::{batch_utils, BatchResult};
85
86// Re-export enhanced file utilities
87pub use utils::file::{
88    create_temp_dir, get_file_extension, get_file_size, guess_mime_from_path, is_audio_file,
89    is_file_size_acceptable, is_image_file, is_text_file, is_video_file, read_file, remove_file,
90    save_temp_file, validate_file,
91};
ai_lib/lib.rs

ai_lib/
lib.rs
