rig-core 0.42.0

//! This module provides traits for defining and creating provider clients.
//! Clients are used to create models for completion, embeddings, etc.

pub mod audio_generation;
pub mod completion;
pub mod embeddings;
pub mod image_generation;
pub mod model_listing;
pub mod rerank;
pub mod transcription;
pub mod verify;

use bytes::Bytes;
pub use completion::{CompletionClient, ConstructCompletionModel};
pub use embeddings::EmbeddingsClient;
use http::{HeaderMap, HeaderName, HeaderValue};
pub use model_listing::{ModelLister, ModelListingClient};
pub use rerank::RerankingClient;
use std::{env::VarError, fmt::Debug, marker::PhantomData, sync::Arc};
use thiserror::Error;
pub use verify::{VerifyClient, VerifyError};

#[cfg(feature = "image")]
use crate::image_generation::ImageGenerationModel;
#[cfg(feature = "image")]
use image_generation::ImageGenerationClient;

#[cfg(feature = "audio")]
use crate::audio_generation::*;
#[cfg(feature = "audio")]
use audio_generation::*;

use crate::{
    completion::CompletionModel,
    embeddings::EmbeddingModel,
    http_client::{
        self, Builder, HttpClientExt, LazyBody, MultipartForm, Request, Response, make_auth_header,
    },
    markers::Missing,
    prelude::TranscriptionClient,
    rerank::RerankModel,
    transcription::TranscriptionModel,
    wasm_compat::{WasmCompatSend, WasmCompatSync},
};

#[derive(Debug, Error)]
pub enum ClientBuilderError {
    /// The underlying HTTP backend failed during builder construction.
    #[error("reqwest error: {0}")]
    HttpError(
        #[from]
        #[source]
        reqwest::Error,
    ),
    /// A provider-specific builder property was invalid.
    #[error("invalid property: {0}")]
    InvalidProperty(&'static str),
}

/// Errors returned while constructing provider clients from environment variables or explicit input.
///
/// Provider-specific client constructors use this error for configuration problems that can be
/// detected before any model request is sent, such as missing API keys, invalid environment
/// values, or invalid builder configuration.
#[derive(Debug, Error)]
pub enum ProviderClientError {
    /// A required or optional environment variable could not be read as valid Unicode.
    ///
    /// For required variables, this variant is also returned when the variable is not present.
    #[error("environment variable `{name}` is not set or is invalid")]
    EnvironmentVariable {
        /// The environment variable name.
        name: &'static str,
        /// The underlying environment lookup error.
        #[source]
        source: VarError,
    },
    /// The underlying provider client builder failed while constructing HTTP configuration.
    #[error(transparent)]
    Http(#[from] http_client::Error),
    /// The provider received an unsupported or incomplete configuration.
    #[error("{0}")]
    InvalidConfiguration(&'static str),
}

/// Result type returned by provider client construction helpers.
pub type ProviderClientResult<T> = std::result::Result<T, ProviderClientError>;

/// Read a required environment variable for provider client construction.
///
/// Returns [`ProviderClientError::EnvironmentVariable`] when the variable is missing or contains
/// invalid Unicode.
pub fn required_env_var(name: &'static str) -> ProviderClientResult<String> {
    std::env::var(name).map_err(|source| ProviderClientError::EnvironmentVariable { name, source })
}

/// Read an optional environment variable for provider client construction.
///
/// Missing variables return `Ok(None)`. Variables containing invalid Unicode return
/// [`ProviderClientError::EnvironmentVariable`].
pub fn optional_env_var(name: &'static str) -> ProviderClientResult<Option<String>> {
    match std::env::var(name) {
        Ok(value) => Ok(Some(value)),
        Err(VarError::NotPresent) => Ok(None),
        Err(source) => Err(ProviderClientError::EnvironmentVariable { name, source }),
    }
}

/// Abstracts over the ability to instantiate a client, either via environment variables or some
/// `Self::Input`
pub trait ProviderClient {
    /// Input accepted by [`ProviderClient::from_val`].
    type Input;
    /// Error returned when client construction fails.
    type Error;

    /// Create a client from the process's environment.
    fn from_env() -> Result<Self, Self::Error>
    where
        Self: Sized;

    /// Create a client from an explicit provider-specific input value.
    fn from_val(input: Self::Input) -> Result<Self, Self::Error>
    where
        Self: Sized;
}

/// A trait for API key inputs accepted by [`ClientBuilder::api_key`].
///
/// Returning `Some` inserts a header into the generic [`Client`]. Returning `None`
/// lets the provider extension handle credentials itself.
pub trait ApiKey: Sized {
    /// Convert this key into a default request header, if the generic client
    /// should own that authentication header.
    fn into_header(self) -> Option<http_client::Result<(HeaderName, HeaderValue)>> {
        None
    }
}

/// An API key which will be inserted into a `Client`'s default headers as a bearer auth token
pub struct BearerAuth(String);

impl ApiKey for BearerAuth {
    fn into_header(self) -> Option<http_client::Result<(HeaderName, HeaderValue)>> {
        Some(make_auth_header(self.0))
    }
}

impl<S> From<S> for BearerAuth
where
    S: Into<String>,
{
    fn from(value: S) -> Self {
        Self(value.into())
    }
}

/// A type containing nothing at all. For `Option`-like behavior on the type level, i.e. to describe
/// the lack of a capability or field (an API key, for instance)
#[derive(Debug, Default, Clone, Copy)]
pub struct Nothing;

impl ApiKey for Nothing {}

#[derive(Clone)]
/// Generic provider client shared by Rig provider integrations.
///
/// `Ext` stores provider-specific behavior such as URL construction, request
/// customization, and capabilities. `H` is the HTTP backend and defaults to
/// `reqwest::Client`.
pub struct Client<Ext = Nothing, H = reqwest::Client> {
    base_url: Arc<str>,
    headers: Arc<HeaderMap>,
    http_client: H,
    ext: Ext,
}

/// Provider extension hook for redacted [`Debug`] output.
pub trait DebugExt: Debug {
    /// Additional provider-specific fields to include in `Client` debug output.
    fn fields(&self) -> impl Iterator<Item = (&'static str, &dyn Debug)> {
        std::iter::empty()
    }
}

impl<Ext, H> std::fmt::Debug for Client<Ext, H>
where
    Ext: DebugExt,
    H: std::fmt::Debug,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let mut d = &mut f.debug_struct("Client");

        d = d
            .field("base_url", &self.base_url)
            .field(
                "headers",
                &self
                    .headers
                    .iter()
                    .filter_map(|(k, v)| {
                        if k == http::header::AUTHORIZATION || k.as_str().contains("api-key") {
                            None
                        } else {
                            Some((k, v))
                        }
                    })
                    .collect::<Vec<(&HeaderName, &HeaderValue)>>(),
            )
            .field("http_client", &self.http_client);

        self.ext
            .fields()
            .fold(d, |d, (name, field)| d.field(name, field))
            .finish()
    }
}

pub enum Transport {
    /// Regular request/response HTTP transport.
    Http,
    /// Server-sent events streaming transport.
    Sse,
}

/// An API provider extension, this abstracts over extensions which may be used in conjunction with
/// the `Client<Ext, H>` struct to define the behavior of a provider with respect to networking,
/// auth, instantiating models
pub trait Provider: Sized {
    /// The builder type that constructs this provider extension.
    /// This associates extensions with their builders for type inference.
    type Builder: ProviderBuilder;

    /// Provider endpoint used by [`VerifyClient`] to validate credentials.
    const VERIFY_PATH: &'static str;

    /// Build a complete request URI for the given base URL, provider path, and transport.
    fn build_uri(&self, base_url: &str, path: &str, _transport: Transport) -> String {
        // Some providers (like Azure) have a blank base URL to allow users to input their own endpoints.
        let base_url = if base_url.is_empty() || base_url.ends_with('/') {
            base_url.to_string()
        } else {
            // Only add a slash to the base_url when it doesn't already end with a slash
            base_url.to_string() + "/"
        };

        base_url + path.trim_start_matches('/')
    }

    /// Apply provider-specific request customization before sending.
    fn with_custom(&self, req: http_client::Builder) -> http_client::Result<http_client::Builder> {
        Ok(req)
    }
}

/// A wrapper type providing runtime checks on a provider's capabilities via the [Capability] trait
pub struct Capable<M>(PhantomData<M>);

/// Type-level marker for whether a provider supports a capability.
pub trait Capability {
    /// Whether this marker represents a supported capability.
    const CAPABLE: bool;
}

impl<M> Capability for Capable<M> {
    const CAPABLE: bool = true;
}

impl Capability for Nothing {
    const CAPABLE: bool = false;
}

/// The capabilities of a given provider, i.e. embeddings, audio transcriptions, text completion
pub trait Capabilities<H = reqwest::Client> {
    /// Completion model capability marker.
    type Completion: Capability;
    /// Embedding model capability marker.
    type Embeddings: Capability;
    /// Rerank model capability marker.
    type Rerank: Capability;
    /// Audio transcription model capability marker.
    type Transcription: Capability;
    /// Model listing capability marker.
    type ModelListing: Capability;
    #[cfg(feature = "image")]
    /// Image generation model capability marker.
    type ImageGeneration: Capability;
    #[cfg(feature = "audio")]
    /// Audio generation model capability marker.
    type AudioGeneration: Capability;
}

/// An API provider extension *builder*, this abstracts over provider-specific builders which are
/// able to configure and produce a given provider's extension type
///
/// See [Provider]
pub trait ProviderBuilder: Sized + Default + Clone {
    /// Provider extension type built for a concrete HTTP backend.
    type Extension<H>: Provider
    where
        H: HttpClientExt;
    /// API key input type accepted by the provider's client builder.
    type ApiKey: ApiKey;

    /// Default base URL for the provider.
    const BASE_URL: &'static str;

    /// Build the provider extension from the client builder configuration.
    fn build<H>(
        builder: &ClientBuilder<Self, Self::ApiKey, H>,
    ) -> http_client::Result<Self::Extension<H>>
    where
        H: HttpClientExt;

    /// This method can be used to customize the fields of `builder` before it is used to create
    /// a client. For example, adding default headers
    fn finish<H>(
        &self,
        builder: ClientBuilder<Self, Self::ApiKey, H>,
    ) -> http_client::Result<ClientBuilder<Self, Self::ApiKey, H>> {
        Ok(builder)
    }
}

// These implementations are declarations of associated types and constants,
// so ordinary helper functions cannot express the repeated structure. Keeping
// the variation points in one invocation makes each provider's configuration
// visible without duplicating the generic builder plumbing.
macro_rules! impl_default_provider_builder {
    (
        $builder:ty => $extension:ty,
        api_key = $api_key:ty,
        base_url = $base_url:expr
        $(, finish = $finish:path, state = $state:ident)? $(,)?
    ) => {
        impl $crate::client::ProviderBuilder for $builder {
            type Extension<H>
                = $extension
            where
                H: $crate::http_client::HttpClientExt;
            type ApiKey = $api_key;

            const BASE_URL: &'static str = $base_url;

            fn build<H>(
                _builder: &$crate::client::ClientBuilder<Self, Self::ApiKey, H>,
            ) -> $crate::http_client::Result<Self::Extension<H>>
            where
                H: $crate::http_client::HttpClientExt,
            {
                Ok(<$extension>::default())
            }

            $(
                fn finish<H>(
                    &self,
                    builder: $crate::client::ClientBuilder<Self, Self::ApiKey, H>,
                ) -> $crate::http_client::Result<
                    $crate::client::ClientBuilder<Self, Self::ApiKey, H>,
                > {
                    $finish(&self.$state, builder)
                }
            )?
        }
    };
}
pub(crate) use impl_default_provider_builder;

// A provider's Capabilities impl is a pure associated-type table where every
// slot a provider does not support is `Nothing`. The named optional slots
// keep each provider's invocation down to what it actually supports, and the
// macro owns the feature gating on the image/audio slots.
macro_rules! impl_capabilities {
    (
        $ext:ty
        $(, completion = $completion:ty)?
        $(, embeddings = $embeddings:ty)?
        $(, transcription = $transcription:ty)?
        $(, model_listing = $model_listing:ty)?
        $(, image_generation = $image_generation:ty)?
        $(, audio_generation = $audio_generation:ty)?
        $(, rerank = $rerank:ty)?
        $(,)?
    ) => {
        impl<H> $crate::client::Capabilities<H> for $ext {
            type Completion = $crate::client::impl_capabilities!(@slot $($completion)?);
            type Embeddings = $crate::client::impl_capabilities!(@slot $($embeddings)?);
            type Transcription = $crate::client::impl_capabilities!(@slot $($transcription)?);
            type ModelListing = $crate::client::impl_capabilities!(@slot $($model_listing)?);
            #[cfg(feature = "image")]
            type ImageGeneration = $crate::client::impl_capabilities!(@slot $($image_generation)?);
            #[cfg(feature = "audio")]
            type AudioGeneration = $crate::client::impl_capabilities!(@slot $($audio_generation)?);
            type Rerank = $crate::client::impl_capabilities!(@slot $($rerank)?);
        }
    };
    (@slot $model:ty) => { $crate::client::Capable<$model> };
    (@slot) => { $crate::client::Nothing };
}
pub(crate) use impl_capabilities;

// ProviderClient is implemented for concrete client aliases, which likewise
// cannot be factored into a function. The optional base-URL form captures the
// only common construction variation without hiding provider-specific auth.
macro_rules! impl_provider_client {
    (
        $client:ty,
        input = $input:ty,
        api_key_env = $api_key_env:literal,
        base_url_env_first = $base_url_env:literal $(,)?
    ) => {
        $crate::client::impl_provider_client!(@with_base
            $client,
            input = $input,
            api_key_env = $api_key_env,
            configuration = {
                let base_url = $crate::client::optional_env_var($base_url_env)?;
                let api_key = $crate::client::required_env_var($api_key_env)?;
                (api_key, base_url)
            }
        );
    };
    (
        $client:ty,
        input = $input:ty,
        api_key_env = $api_key_env:literal,
        base_url_env = $base_url_env:literal $(,)?
    ) => {
        $crate::client::impl_provider_client!(@with_base
            $client,
            input = $input,
            api_key_env = $api_key_env,
            configuration = {
                let api_key = $crate::client::required_env_var($api_key_env)?;
                let base_url = $crate::client::optional_env_var($base_url_env)?;
                (api_key, base_url)
            }
        );
    };
    (
        $client:ty,
        input = $input:ty,
        api_key_env = $api_key_env:literal,
        base_url = $base_url:expr $(,)?
    ) => {
        $crate::client::impl_provider_client!(@with_base
            $client,
            input = $input,
            api_key_env = $api_key_env,
            configuration = {
                let api_key = $crate::client::required_env_var($api_key_env)?;
                (api_key, $base_url)
            }
        );
    };
    (@with_base
        $client:ty,
        input = $input:ty,
        api_key_env = $api_key_env:literal,
        configuration = $configuration:block
    ) => {
        impl $crate::client::ProviderClient for $client {
            type Input = $input;
            type Error = $crate::client::ProviderClientError;

            #[doc = concat!("Create this provider client from the `", $api_key_env, "` environment variable.")]
            fn from_env() -> Result<Self, Self::Error> {
                let (api_key, base_url) = $configuration;
                let mut builder = Self::builder().api_key(api_key);
                if let Some(base_url) = base_url {
                    builder = builder.base_url(base_url);
                }
                builder.build().map_err(Into::into)
            }

            fn from_val(input: Self::Input) -> Result<Self, Self::Error> {
                Self::new(input).map_err(Into::into)
            }
        }
    };
    (
        $client:ty,
        input = $input:ty,
        api_key_env = $api_key_env:literal $(,)?
    ) => {
        impl $crate::client::ProviderClient for $client {
            type Input = $input;
            type Error = $crate::client::ProviderClientError;

            #[doc = concat!("Create this provider client from the `", $api_key_env, "` environment variable.")]
            fn from_env() -> Result<Self, Self::Error> {
                let api_key = $crate::client::required_env_var($api_key_env)?;
                Self::new(api_key).map_err(Into::into)
            }

            fn from_val(input: Self::Input) -> Result<Self, Self::Error> {
                Self::new(input).map_err(Into::into)
            }
        }
    };
}
pub(crate) use impl_provider_client;

/// `new` is pinned to `H = reqwest::Client` so the call site infers without an explicit `H`
/// annotation. Callers who want a different backend should go through [`Client::builder`] and
/// chain [`ClientBuilder::http_client`] before [`ClientBuilder::build`].
impl<Ext> Client<Ext, reqwest::Client>
where
    Ext: Provider,
    Ext::Builder: ProviderBuilder<Extension<reqwest::Client> = Ext> + Default,
{
    /// Construct a provider client using the default `reqwest::Client` backend.
    pub fn new(
        api_key: impl Into<<Ext::Builder as ProviderBuilder>::ApiKey>,
    ) -> http_client::Result<Self> {
        Self::builder().api_key(api_key).build()
    }
}

impl<Ext, H> Client<Ext, H> {
    /// Returns the configured provider base URL.
    pub fn base_url(&self) -> &str {
        &self.base_url
    }

    /// Returns default headers applied to outgoing provider requests.
    pub fn headers(&self) -> &HeaderMap {
        &self.headers
    }

    /// Returns the provider extension.
    pub fn ext(&self) -> &Ext {
        &self.ext
    }

    /// Reuse this client's base URL, headers, and HTTP backend with a different extension.
    pub fn with_ext<NewExt>(self, new_ext: NewExt) -> Client<NewExt, H> {
        Client {
            base_url: self.base_url,
            headers: self.headers,
            http_client: self.http_client,
            ext: new_ext,
        }
    }
}

impl<Ext, H> HttpClientExt for Client<Ext, H>
where
    H: HttpClientExt + 'static,
    Ext: WasmCompatSend + WasmCompatSync + 'static,
{
    fn send<T, U>(
        &self,
        mut req: Request<T>,
    ) -> impl Future<Output = http_client::Result<Response<LazyBody<U>>>> + WasmCompatSend + 'static
    where
        T: Into<Bytes> + WasmCompatSend,
        U: From<Bytes>,
        U: WasmCompatSend + 'static,
    {
        req.headers_mut().insert(
            http::header::CONTENT_TYPE,
            http::HeaderValue::from_static("application/json"),
        );

        self.http_client.send(req)
    }

    fn send_multipart<U>(
        &self,
        req: Request<MultipartForm>,
    ) -> impl Future<Output = http_client::Result<Response<LazyBody<U>>>> + WasmCompatSend + 'static
    where
        U: From<Bytes>,
        U: WasmCompatSend + 'static,
    {
        self.http_client.send_multipart(req)
    }

    fn send_streaming<T>(
        &self,
        mut req: Request<T>,
    ) -> impl Future<Output = http_client::Result<http_client::StreamingResponse>> + WasmCompatSend
    where
        T: Into<Bytes> + WasmCompatSend,
    {
        req.headers_mut().insert(
            http::header::CONTENT_TYPE,
            http::HeaderValue::from_static("application/json"),
        );

        self.http_client.send_streaming(req)
    }
}

/// `builder()` is anchored on `Client<Ext, reqwest::Client>` purely as an inference hook so that
/// `provider::Client::builder()` resolves without a `H` annotation. The returned builder itself
/// has `H = Missing`, accurately reflecting that no backend has been chosen yet; the eventual
/// `Client` produced by `build()` may end up with any HTTP backend depending on whether
/// [`ClientBuilder::http_client`] was called.
impl<Ext> Client<Ext, reqwest::Client>
where
    Ext: Provider,
    Ext::Builder: ProviderBuilder + Default,
{
    /// Start constructing a provider client.
    pub fn builder() -> ClientBuilder<Ext::Builder, Missing, Missing> {
        ClientBuilder::default()
    }
}

impl<Ext, H> Client<Ext, H>
where
    Ext: Provider,
{
    fn request(
        &self,
        method: http::Method,
        path: &str,
        transport: Transport,
    ) -> http_client::Result<Builder> {
        let uri = self.ext.build_uri(&self.base_url, path, transport);

        let mut req = Request::builder().method(method).uri(uri);

        if let Some(hs) = req.headers_mut() {
            hs.extend(self.headers.iter().map(|(k, v)| (k.clone(), v.clone())));
        }

        self.ext.with_custom(req)
    }

    /// Build a provider-customized POST request for a regular HTTP endpoint.
    pub fn post<S>(&self, path: S) -> http_client::Result<Builder>
    where
        S: AsRef<str>,
    {
        self.request(http::Method::POST, path.as_ref(), Transport::Http)
    }

    /// Build a provider-customized POST request for an SSE endpoint.
    pub fn post_sse<S>(&self, path: S) -> http_client::Result<Builder>
    where
        S: AsRef<str>,
    {
        self.request(http::Method::POST, path.as_ref(), Transport::Sse)
    }

    /// Build a provider-customized GET request for an SSE endpoint.
    pub fn get_sse<S>(&self, path: S) -> http_client::Result<Builder>
    where
        S: AsRef<str>,
    {
        self.request(http::Method::GET, path.as_ref(), Transport::Sse)
    }

    /// Build a provider-customized GET request for a regular HTTP endpoint.
    pub fn get<S>(&self, path: S) -> http_client::Result<Builder>
    where
        S: AsRef<str>,
    {
        self.request(http::Method::GET, path.as_ref(), Transport::Http)
    }
}

impl<Ext, H> VerifyClient for Client<Ext, H>
where
    H: HttpClientExt,
    Ext: DebugExt + Provider + WasmCompatSync,
{
    async fn verify(&self) -> Result<(), VerifyError> {
        use http::StatusCode;

        let req = self
            .get(Ext::VERIFY_PATH)?
            .body(http_client::NoBody)
            .map_err(http_client::Error::from)?;

        // The reqwest transport reports non-success as an error before this
        // status match can run (found live on rig#2315's error matrix: the
        // 401/403 arms below were dead and every bogus key surfaced as a raw
        // HttpError). Recover the status from the transport error so the
        // documented VerifyError classification actually fires.
        let response = match self.http_client.send(req).await {
            Ok(response) => response,
            Err(error) => {
                return Err(match error.non_success_status() {
                    Some(StatusCode::UNAUTHORIZED) | Some(StatusCode::FORBIDDEN) => {
                        VerifyError::InvalidAuthentication
                    }
                    _ => VerifyError::HttpError(error),
                });
            }
        };

        match response.status() {
            StatusCode::OK => Ok(()),
            StatusCode::UNAUTHORIZED | reqwest::StatusCode::FORBIDDEN => {
                Err(VerifyError::InvalidAuthentication)
            }
            // The failed response's headers are preserved on every branch, so
            // a caller can read rate-limit metadata such as `Retry-After` off
            // a rejected verification (rig#2210).
            StatusCode::INTERNAL_SERVER_ERROR => {
                let headers = Box::new(response.headers().clone());
                let body = http_client::text(response).await?;
                Err(VerifyError::HttpError(
                    http_client::Error::InvalidStatusCodeWithDetails {
                        status: StatusCode::INTERNAL_SERVER_ERROR,
                        body,
                        headers,
                    },
                ))
            }
            status if status.as_u16() == 529 => {
                let headers = Box::new(response.headers().clone());
                let body = http_client::text(response).await?;
                Err(VerifyError::HttpError(
                    http_client::Error::InvalidStatusCodeWithDetails {
                        status,
                        body,
                        headers,
                    },
                ))
            }
            _ => {
                let status = response.status();

                if status.is_success() {
                    Ok(())
                } else {
                    let headers = Box::new(response.headers().clone());
                    let body: String = String::from_utf8_lossy(&response.into_body().await?).into();
                    Err(VerifyError::HttpError(
                        http_client::Error::InvalidStatusCodeWithDetails {
                            status,
                            body,
                            headers,
                        },
                    ))
                }
            }
        }
    }
}

/// Type-state builder for [`Client`].
///
/// Each generic slot encodes a separate "has the user supplied this yet?" question:
///
/// - `ApiKey = Missing` means the caller has not yet called [`Self::api_key`]; transitioning to a
///   concrete `ApiKey` type is required before [`Self::build`] is reachable.
/// - `H = Missing` means the caller has not yet called [`Self::http_client`]; in that state
///   `build()` substitutes the canonical `reqwest::Client` backend at construction time. Once a
///   backend has been supplied, `H` is the concrete HTTP client type and `build()` uses it
///   directly.
///
/// Keeping `Missing` as the *type-level* placeholder (rather than reusing `reqwest::Client`)
/// means the builder's generics describe what the caller has actually provided, instead of
/// pretending a default value is already present. It also avoids carrying an `Option<H>` whose
/// `None` branch existed only to model the same "user hasn't picked a backend" state.
#[derive(Clone)]
pub struct ClientBuilder<Ext, ApiKey = Missing, H = Missing> {
    base_url: String,
    api_key: ApiKey,
    headers: HeaderMap,
    http_client: H,
    ext: Ext,
}

impl<ExtBuilder> Default for ClientBuilder<ExtBuilder, Missing, Missing>
where
    ExtBuilder: ProviderBuilder + Default,
{
    fn default() -> Self {
        Self {
            api_key: Missing,
            headers: Default::default(),
            base_url: ExtBuilder::BASE_URL.into(),
            http_client: Missing,
            ext: Default::default(),
        }
    }
}

impl<Ext, H> ClientBuilder<Ext, Missing, H> {
    /// Set the API key for this client. This *must* be done before the `build` method can be
    /// called
    pub fn api_key<ApiKey>(self, api_key: impl Into<ApiKey>) -> ClientBuilder<Ext, ApiKey, H> {
        ClientBuilder {
            api_key: api_key.into(),
            base_url: self.base_url,
            headers: self.headers,
            http_client: self.http_client,
            ext: self.ext,
        }
    }
}

impl<Ext, ApiKey, H> ClientBuilder<Ext, ApiKey, H>
where
    Ext: Clone,
{
    /// Owned map over the ext field
    pub(crate) fn over_ext<F, NewExt>(self, f: F) -> ClientBuilder<NewExt, ApiKey, H>
    where
        F: FnOnce(Ext) -> NewExt,
    {
        let ClientBuilder {
            base_url,
            api_key,
            headers,
            http_client,
            ext,
        } = self;

        let new_ext = f(ext.clone());

        ClientBuilder {
            base_url,
            api_key,
            headers,
            http_client,
            ext: new_ext,
        }
    }

    /// Set the base URL for this client
    pub fn base_url<S>(self, base_url: S) -> Self
    where
        S: AsRef<str>,
    {
        Self {
            base_url: base_url.as_ref().to_string(),
            ..self
        }
    }

    /// Set the HTTP backend used in this client.
    ///
    /// Calling this advances the builder's `H` slot from whatever it was (typically `Missing`)
    /// to the supplied client's type, which selects the H-generic [`Self::build`] impl below.
    pub fn http_client<U>(self, http_client: U) -> ClientBuilder<Ext, ApiKey, U> {
        ClientBuilder {
            http_client,
            base_url: self.base_url,
            api_key: self.api_key,
            headers: self.headers,
            ext: self.ext,
        }
    }

    /// Set the HTTP headers used in this client
    pub fn http_headers(self, headers: HeaderMap) -> Self {
        Self { headers, ..self }
    }

    pub(crate) fn headers_mut(&mut self) -> &mut HeaderMap {
        &mut self.headers
    }

    pub(crate) fn ext_mut(&mut self) -> &mut Ext {
        &mut self.ext
    }
}

impl<Ext, ApiKey, H> ClientBuilder<Ext, ApiKey, H> {
    pub(crate) fn get_api_key(&self) -> &ApiKey {
        &self.api_key
    }
}

impl<Ext, Key, H> ClientBuilder<Ext, Key, H> {
    /// Returns the provider extension builder state.
    pub fn ext(&self) -> &Ext {
        &self.ext
    }

    /// Returns the configured base URL.
    pub fn get_base_url(&self) -> &str {
        &self.base_url
    }
}

/// Default-backend `build`: when the caller never called [`ClientBuilder::http_client`], the
/// builder's `H` slot is still `Missing`, and we substitute the canonical `reqwest::Client` at
/// build time. This is the only place in the crate that knows about that default, and it is
/// disjoint by trait bound from the H-generic `build` below (`Missing` does not implement
/// [`HttpClientExt`]).
impl<ExtBuilder, Key> ClientBuilder<ExtBuilder, Key, Missing>
where
    ExtBuilder: ProviderBuilder<ApiKey = Key>,
    Key: ApiKey,
{
    /// Build a client using the default `reqwest::Client` backend.
    pub fn build(
        self,
    ) -> http_client::Result<Client<ExtBuilder::Extension<reqwest::Client>, reqwest::Client>> {
        self.http_client(reqwest::Client::default()).build()
    }
}

/// Concrete-backend `build`: the caller supplied an HTTP client via
/// [`ClientBuilder::http_client`], so `H` is a real `HttpClientExt` type and we use it directly.
impl<ExtBuilder, Key, H> ClientBuilder<ExtBuilder, Key, H>
where
    ExtBuilder: ProviderBuilder<ApiKey = Key>,
    Key: ApiKey,
    H: HttpClientExt,
{
    /// Build a client using the HTTP backend supplied with [`ClientBuilder::http_client`].
    pub fn build(mut self) -> http_client::Result<Client<ExtBuilder::Extension<H>, H>> {
        let ext_builder = self.ext.clone();

        self = ext_builder.finish(self)?;
        let ext = ExtBuilder::build(&self)?;

        let ClientBuilder {
            http_client,
            base_url,
            mut headers,
            api_key,
            ..
        } = self;

        if let Some((k, v)) = api_key.into_header().transpose()?
            && !headers.contains_key(&k)
        {
            headers.insert(k, v);
        }

        Ok(Client {
            http_client,
            base_url: Arc::from(base_url.as_str()),
            headers: Arc::new(headers),
            ext,
        })
    }
}

// Every single-model capability client impl on `Client<Ext, H>` shares the
// same shape: gate on the matching `Capabilities` slot, name the model type,
// and construct it with `M::make`. The macro keeps the per-capability
// variation (trait, slot, associated type, method, extra model bounds, and
// feature gate) in one invocation each. `CompletionClient` (different
// constructor protocol) and `EmbeddingsClient` (extra `_with_ndims` method)
// stay hand-written below.
macro_rules! impl_capability_client {
    (
        $(#[cfg(feature = $feature:literal)])?
        $client_trait:ident { $slot:ident, $assoc:ident, $method:ident, $model_trait:ident $(+ $extra:path)* }
    ) => {
        $(#[cfg(feature = $feature)])?
        impl<M, Ext, H> $client_trait for Client<Ext, H>
        where
            Ext: Capabilities<H, $slot = Capable<M>>,
            M: $model_trait<Client = Self> $(+ $extra)*,
        {
            type $assoc = M;

            fn $method(&self, model: impl Into<String>) -> Self::$assoc {
                M::make(self, model)
            }
        }
    };
}

impl<M, Ext, H> CompletionClient for Client<Ext, H>
where
    Ext: Capabilities<H, Completion = Capable<M>>,
    M: CompletionModel + ConstructCompletionModel<Self>,
{
    type CompletionModel = M;

    fn completion_model(&self, model: impl Into<String>) -> Self::CompletionModel {
        M::construct(self, model.into())
    }
}

impl<M, Ext, H> EmbeddingsClient for Client<Ext, H>
where
    Ext: Capabilities<H, Embeddings = Capable<M>>,
    M: EmbeddingModel<Client = Self>,
{
    type EmbeddingModel = M;

    fn embedding_model(&self, model: impl Into<String>) -> Self::EmbeddingModel {
        M::make(self, model, None)
    }

    fn embedding_model_with_ndims(
        &self,
        model: impl Into<String>,
        ndims: usize,
    ) -> Self::EmbeddingModel {
        M::make(self, model, Some(ndims))
    }
}

impl_capability_client!(RerankingClient {
    Rerank,
    RerankModel,
    rerank_model,
    RerankModel
});

impl_capability_client!(TranscriptionClient {
    Transcription,
    TranscriptionModel,
    transcription_model,
    TranscriptionModel + WasmCompatSend
});

impl_capability_client!(
    #[cfg(feature = "image")]
    ImageGenerationClient {
        ImageGeneration,
        ImageGenerationModel,
        image_generation_model,
        ImageGenerationModel
    }
);

impl_capability_client!(
    #[cfg(feature = "audio")]
    AudioGenerationClient {
        AudioGeneration,
        AudioGenerationModel,
        audio_generation_model,
        AudioGenerationModel
    }
);

impl<M, Ext, H> ModelListingClient for Client<Ext, H>
where
    Ext: Capabilities<H, ModelListing = Capable<M>> + Clone,
    M: ModelLister<H, Client = Self> + WasmCompatSend + WasmCompatSync + Clone + 'static,
    H: WasmCompatSend + WasmCompatSync + Clone,
{
    fn list_models(
        &self,
    ) -> impl std::future::Future<
        Output = Result<crate::model::ModelList, crate::model::ModelListingError>,
    > + WasmCompatSend {
        let lister = M::new(self.clone());
        async move { lister.list_all().await }
    }
}

#[cfg(all(target_arch = "wasm32", target_os = "unknown"))]
mod wasm_model_listing_compile_checks {
    use super::{ModelListingClient, Nothing};
    use crate::{
        http_client::{self, HttpClientExt, LazyBody, MultipartForm, Request, Response},
        providers::{anthropic, deepseek, mistral, ollama, openai, openrouter},
        wasm_compat::WasmCompatSend,
    };
    use bytes::Bytes;
    use std::{
        future::{self, Future},
        marker::PhantomData,
        rc::Rc,
    };

    #[derive(Clone, Default)]
    struct WasmOnlyHttpClient {
        _not_send_sync: PhantomData<Rc<()>>,
    }

    impl HttpClientExt for WasmOnlyHttpClient {
        fn send<T, U>(
            &self,
            _req: Request<T>,
        ) -> impl Future<Output = http_client::Result<Response<LazyBody<U>>>> + WasmCompatSend + 'static
        where
            T: Into<Bytes> + WasmCompatSend,
            U: From<Bytes> + WasmCompatSend + 'static,
        {
            future::ready(Err(http_client::Error::StreamEnded))
        }

        fn send_multipart<U>(
            &self,
            _req: Request<MultipartForm>,
        ) -> impl Future<Output = http_client::Result<Response<LazyBody<U>>>> + WasmCompatSend + 'static
        where
            U: From<Bytes> + WasmCompatSend + 'static,
        {
            future::ready(Err(http_client::Error::StreamEnded))
        }

        fn send_streaming<T>(
            &self,
            _req: Request<T>,
        ) -> impl Future<Output = http_client::Result<http_client::StreamingResponse>> + WasmCompatSend
        where
            T: Into<Bytes> + WasmCompatSend,
        {
            future::ready(Err(http_client::Error::StreamEnded))
        }
    }

    fn assert_model_listing_client<C>(client: C)
    where
        C: ModelListingClient,
    {
        let _ = client.list_models();
    }

    fn assert_simple_model_listers_accept_wasm_only_http_clients() {
        let _ = openrouter::Client::builder()
            .api_key("dummy-key")
            .http_client(WasmOnlyHttpClient::default())
            .build()
            .map(assert_model_listing_client);

        let _ = openai::Client::builder()
            .api_key("dummy-key")
            .http_client(WasmOnlyHttpClient::default())
            .build()
            .map(assert_model_listing_client);

        let _ = mistral::Client::builder()
            .api_key("dummy-key")
            .http_client(WasmOnlyHttpClient::default())
            .build()
            .map(assert_model_listing_client);

        let _ = anthropic::Client::builder()
            .api_key("dummy-key")
            .http_client(WasmOnlyHttpClient::default())
            .build()
            .map(assert_model_listing_client);

        let _ = ollama::Client::builder()
            .api_key(Nothing)
            .http_client(WasmOnlyHttpClient::default())
            .build()
            .map(assert_model_listing_client);

        let _ = deepseek::Client::builder()
            .api_key("dummy-key")
            .http_client(WasmOnlyHttpClient::default())
            .build()
            .map(assert_model_listing_client);
    }

    #[allow(dead_code)]
    fn compile_assertions() {
        assert_simple_model_listers_accept_wasm_only_http_clients();
    }
}

#[cfg(test)]
mod tests {
    use crate::providers::anthropic;

    /// Type-level test that `Client::builder()` methods do not require annotation to determine
    /// backig HTTP client
    #[test]
    fn ensures_client_builder_no_annotation() {
        let http_client = reqwest::Client::default();
        let _ = anthropic::Client::builder()
            .http_client(http_client)
            .api_key("Foo")
            .build()
            .unwrap();
    }
}