aleph-alpha-api 0.1.0

Interact with large language models provided by the Aleph Alpha API in Rust code
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

use serde::{Deserialize, Serialize};

#[derive(Serialize)]
pub struct TokenizationRequest {
    /// Name of the model tasked with completing the prompt. E.g. `luminous-base`.
    pub model: String,
    /// String to tokenize.
    pub prompt: String,
    /// Set this value to `true` to return text-tokens.
    pub tokens: bool,
    /// Set this value to `true to return numeric token-ids.
    pub token_ids: bool,
}

#[derive(Deserialize)]
pub struct TokenizationResponse {
    pub tokens: Option<Vec<String>>,
    pub token_ids: Option<Vec<u32>>,
}

#[derive(Serialize, Debug)]
pub struct DetokenizationRequest {
    /// Name of the model tasked with completing the prompt. E.g. `luminous-base"`.
    pub model: String,
    /// List of ids to detokenize.
    pub token_ids: Vec<u32>,
}

#[derive(Deserialize, Debug)]
pub struct DetokenizationResponse {
    pub result: String,
}