allms 0.42.0

One Library to rule them aLLMs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use anyhow::Result;
use yup_oauth2::{read_service_account_key, ServiceAccountAuthenticator};

// Reusable function for Vertex API authentication
pub async fn get_vertex_token() -> Result<String> {
    let service_account_key = read_service_account_key("secrets/gcp_sa_key.json")
        .await
        .unwrap();

    let auth = ServiceAccountAuthenticator::builder(service_account_key)
        .build()
        .await
        .unwrap();

    let google_token = auth
        .token(&["https://www.googleapis.com/auth/cloud-platform"])
        .await
        .unwrap();

    Ok(google_token.token().unwrap().to_string())
}