use oris_runtime::embedding::{
    embedder_trait::Embedder,
    openai::openai_embedder::{AzureConfig, OpenAiEmbedder},
};

#[tokio::main]
async fn main() {
    let azure_config = AzureConfig::default()
        .with_api_key("REPLACE_ME_WITH_YOUR_API_KEY")
        .with_api_base("https://REPLACE_ME.openai.azure.com")
        .with_api_version("2023-05-15")
        .with_deployment_id("text-embedding-ada-002");

    let embedder = OpenAiEmbedder::new(azure_config);
    let result = embedder.embed_query("Why is the sky blue?").await.unwrap();
    println!("{:?}", result);
}
