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
26
27

/// EmbeddingModel represents supported embedding models for generating embeddings.
/// It includes predefined models and allows specifying a custom model by name.
use crate::prelude::*;

#[derive(Debug, Clone, From, Eq, PartialEq, Serialize, Deserialize)]
pub enum EmbeddingModel {
    /// Predefined model: all-MiniLM-L6. This is a commonly used 
    // model for text embeddings.
    #[serde(rename = "text-embedding-all-minilm-l6-v2-embedding")]
    AllMiniLmL6,

    /// Custom model name as String. Allows users to specify any 
    /// model available in LM Studio.
    #[from]
    Custom(String)
}

/// Now we can .to_string() a model cleanly, e.g. for CLI arguments,
/// logging, or diagnostics.
impl std::fmt::Display for EmbeddingModel {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::AllMiniLmL6 => write!(f, "text-embedding-all-minilm-l6-v2-embedding"),
            Self::Custom(s) => write!(f, "{s}")
        }
    }
}