adk-model 2.3.0

//! Configuration types for Azure AI Inference provider.

use serde::{Deserialize, Serialize};

/// Configuration for Azure AI Inference endpoints.
///
/// Azure AI Inference uses a dedicated endpoint URL and API key for
/// authentication, rather than a shared base URL. Each endpoint hosts
/// a specific model deployment (e.g., Cohere, Llama, Mistral).
///
/// # Example
///
/// ```rust,ignore
/// use adk_model::azure_ai::AzureAIConfig;
///
/// let config = AzureAIConfig::new(
///     "https://my-endpoint.eastus.inference.ai.azure.com",
///     "my-api-key",
///     "meta-llama-3.1-8b-instruct",
/// );
/// ```
///
/// The API key is redacted from `Debug` output and omitted when serializing.
#[derive(Clone, Serialize, Deserialize)]
pub struct AzureAIConfig {
    /// Azure AI Inference endpoint URL (e.g.,
    /// `"https://my-endpoint.eastus.inference.ai.azure.com"`).
    pub endpoint: String,
    /// Azure API key for the endpoint. Never serialized; deserializing a config
    /// without it yields an empty key.
    #[serde(skip_serializing, default)]
    pub api_key: String,
    /// Model name deployed at the endpoint.
    pub model: String,
}

impl std::fmt::Debug for AzureAIConfig {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("AzureAIConfig")
            .field("endpoint", &self.endpoint)
            .field("api_key", &"[REDACTED]")
            .field("model", &self.model)
            .finish()
    }
}

impl AzureAIConfig {
    /// Create a new Azure AI config with the given endpoint, API key, and model.
    pub fn new(
        endpoint: impl Into<String>,
        api_key: impl Into<String>,
        model: impl Into<String>,
    ) -> Self {
        Self { endpoint: endpoint.into(), api_key: api_key.into(), model: model.into() }
    }
}