Skip to main content

reagent_rs/services/llm/models/
embedding.rs

1use serde::{Deserialize, Serialize};
2use std::collections::HashMap;
3
4#[derive(Serialize, Debug, Clone)]
5pub struct EmbeddingsRequest {
6    pub model: String,
7    pub input: Vec<String>,
8    #[serde(skip_serializing_if = "Option::is_none")]
9    pub options: Option<HashMap<String, serde_json::Value>>,
10    #[serde(skip_serializing_if = "Option::is_none")]
11    pub keep_alive: Option<String>,
12}
13
14#[derive(Deserialize, Debug, Clone)]
15pub struct EmbeddingsResponse {
16    pub embedding: Vec<f64>,
17    pub embeddings: Vec<Vec<f64>>,
18}