transformrs 1.0.0

An interface for AI APIs
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
30
31
32
33

extern crate transformrs;

use std::error::Error;
use transformrs::models::models;
use transformrs::models::Models;
use transformrs::Provider;

async fn test_models(provider: Provider) -> Result<Models, Box<dyn Error + Send + Sync>> {
    let keys = transformrs::load_keys(".env");
    let key = keys.for_provider(&provider).unwrap();
    let resp = models(&provider, &key).await;
    let resp = match resp {
        Ok(resp) => resp,
        Err(e) => {
            return Err(e);
        }
    };
    println!("{:?}", resp.raw());
    let resp = resp.structured();
    Ok(resp.unwrap())
}

#[tokio::test]
async fn test_models_groq() {
    let models = test_models(Provider::Groq).await.unwrap();
    assert!(models.contains("llama3-8b-8192"));
}

#[tokio::test]
async fn test_models_openai() {
    let models = test_models(Provider::OpenAI).await.unwrap();
    assert!(models.contains("gpt-4o"));
}