rig 0.41.0

An opinionated library for building LLM powered applications.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use futures::FutureExt;
use rig::providers::perplexity;
use std::future::Future;
use std::panic::AssertUnwindSafe;

use crate::cassettes::{CassetteSpec, ProviderCassette};

async fn perplexity_cassette(
    spec: impl Into<CassetteSpec>,
) -> (ProviderCassette, perplexity::Client) {
    let cassette = ProviderCassette::start("perplexity", spec, "https://api.perplexity.ai").await;
    let client = perplexity::Client::builder()
        .api_key(cassette.api_key("PERPLEXITY_API_KEY"))
        .base_url(cassette.base_url())
        .build()
        .expect("Perplexity cassette client should build");

    (cassette, client)
}

pub(super) async fn with_perplexity_cassette<F, Fut>(spec: impl Into<CassetteSpec>, test_body: F)
where
    F: FnOnce(perplexity::Client) -> Fut,
    Fut: Future<Output = ()>,
{
    let (cassette, client) = perplexity_cassette(spec).await;
    let result = AssertUnwindSafe(test_body(client)).catch_unwind().await;
    cassette.finish_after_test(result).await;
}