stakai 0.3.66

//! # AI SDK
//!
//! A provider-agnostic Rust SDK for AI completions with streaming support.
//!
//! Built by [Stakpak](https://stakpak.dev) 🚀
//!
//! ## Features
//!
//! - **Provider-agnostic**: Unified interface for multiple AI providers (OpenAI, Anthropic, etc.)
//! - **Streaming support**: Real-time streaming responses with unified event types
//! - **Type-safe**: Strong typing with compile-time guarantees
//! - **Zero-cost abstractions**: Static dispatch for optimal performance
//! - **Ergonomic API**: Builder patterns and intuitive interfaces
//!
//! ## Quick Start
//!
//! ```rust,no_run
//! use stakai::{Inference, GenerateRequest, Message, Model, Role};
//!
//! #[tokio::main]
//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
//!     let client = Inference::new();
//!
//!     let request = GenerateRequest::new(
//!         Model::custom("gpt-4", "openai"),
//!         vec![Message::new(Role::User, "What is Rust?")]
//!     );
//!
//!     let response = client.generate(&request).await?;
//!     println!("Response: {}", response.text());
//!
//!     Ok(())
//! }
//! ```

pub mod client;
pub mod error;
pub mod provider;
pub mod providers;
pub mod registry;
pub mod types;

#[cfg(feature = "tracing")]
pub mod tracing;

// Re-export commonly used types
pub use client::{Inference, InferenceConfig};
pub use error::{Error, Result};
pub use registry::{
    ProviderRegistry,
    models_dev::{
        DEFAULT_CACHE_PATH, MODELS_DEV_URL, ProviderInfo, fetch_models_dev,
        filter_configured_providers, get_available_models, load_available_models,
        load_models_for_provider, load_models_for_provider_from_path, parse_models_dev,
    },
};
pub use types::{
    // Cache control types
    AnthropicCacheConfig,
    // Message types
    AnthropicContentPartOptions,
    AnthropicMessageOptions,
    // Request types
    AnthropicOptions,
    // Options types
    AnthropicToolOptions,
    CacheContext,
    CacheControl,
    CacheControlValidator,
    CacheStrategy,
    CacheWarning,
    CacheWarningType,
    CompletionsConfig,
    ContentPart,
    ContentPartProviderOptions,
    // Response types
    FinishReason,
    FinishReasonKind,
    GenerateOptions,
    GenerateRequest,
    GenerateResponse,
    GenerateStream,
    GoogleOptions,
    Headers,
    ImageDetail,
    InputTokenDetails,
    Message,
    MessageContent,
    MessageProviderOptions,
    // Model types
    Model,
    ModelCost,
    ModelLimit,
    OpenAIApiConfig,
    OpenAIOptions,
    OutputTokenDetails,
    PromptCacheRetention,
    ProviderOptions,
    ReasoningEffort,
    ReasoningSummary,
    ResponseContent,
    ResponseWarning,
    ResponsesConfig,
    Role,
    StreamEvent,
    SystemMessageMode,
    ThinkingOptions,
    Tool,
    ToolCall,
    ToolChoice,
    ToolFunction,
    ToolProviderOptions,
    Usage,
};

/// Prelude module for convenient imports
pub mod prelude {
    pub use crate::client::Inference;
    pub use crate::error::{Error, Result};
    pub use crate::provider::Provider;
    pub use crate::types::*;
}