use kernelx_core::prelude::*;

#[tokio::main]
async fn main() -> Result<()> {
    let provider = OpenAI::from_env()?; // Checks env for $OPENAI_API_KEY

    let model = provider.get_model::<dyn Embed>(OpenAIModels::TextEmbedding3Small)?;

    let res = model
        .embed_data(vec![
            "Lorem ipsum dolor sit amet, consectetur adipiscing elit.".to_string(),
            "Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.".to_string(),
            "Excepteur sint occaecat cupidatat non proident, sunt in culpa.".to_string(),
        ])
        .await?;

    println!("{:?}", res); // Vectors

    Ok(())
}