use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize)]
#[cfg_attr(test, derive(Default))]
pub struct Embedding {
#[serde(rename = "model")]
pub model: String,
#[serde(rename = "dimension")]
pub dimension: i64,
#[serde(rename = "embedding")]
pub embedding: Vec<f64>,
#[serde(rename = "error")]
pub error: String,
}
impl Embedding {
pub fn model(&self) -> &String {
&self.model
}
pub fn dimension(&self) -> &i64 {
&self.dimension
}
pub fn embedding(&self) -> &Vec<f64> {
&self.embedding
}
pub fn error(&self) -> &String {
&self.error
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_embedding_creation() {
let _model = <Embedding as Default>::default();
let _ = _model.model();
let _ = _model.dimension();
let _ = _model.embedding();
let _ = _model.error();
}
#[test]
fn test_embedding_serialization() {
let model = <Embedding as Default>::default();
let json = serde_json::to_string(&model);
assert!(json.is_ok());
let deserialized: Result<Embedding, _> = serde_json::from_str(&json.unwrap());
assert!(deserialized.is_ok());
}
}