lm-studio-api-extended 0.1.3

Unofficial Rust client for LM Studio with text embedding support.
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

use crate::prelude::*;

// A message role
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum Role {
    #[serde(rename = "system")]
    System,
    #[serde(rename = "user")]
    User,
    #[serde(rename = "assistant")]
    Assistant,
}

impl TryFrom<&str> for Role {
    type Error = String;

    fn try_from(value: &str) -> std::result::Result<Self, Self::Error> {
        let value = value.trim().to_lowercase();

        if &value == "system" { Ok(Self::System) }
        else if &value == "user" { Ok(Self::User) }
        else if &value == "assistant" { Ok(Self::Assistant) }
        else { Err("Invalid role was provided".to_string()) }
    }
}