minne/evaluations/src/openai.rs

use anyhow::{Context, Result};
use async_openai::{config::OpenAIConfig, Client};

const DEFAULT_BASE_URL: &str = "https://api.openai.com/v1";

pub fn build_client_from_env() -> Result<(Client<OpenAIConfig>, String)> {
    let api_key = std::env::var("OPENAI_API_KEY")
        .context("OPENAI_API_KEY must be set to run retrieval evaluations")?;
    let base_url =
        std::env::var("OPENAI_BASE_URL").unwrap_or_else(|_| DEFAULT_BASE_URL.to_string());

    let config = OpenAIConfig::new()
        .with_api_key(api_key)
        .with_api_base(&base_url);
    Ok((Client::with_config(config), base_url))
}