Fix input type for embedding

This commit is contained in:
Dongri Jin
2024-11-09 08:44:15 +09:00
parent 0a9f65168b
commit 51b9491de9
2 changed files with 12 additions and 3 deletions

View File

@@ -11,10 +11,18 @@ pub struct EmbeddingData {
pub index: i32,
}
#[derive(Debug, Serialize, Deserialize, Clone)]
#[serde(rename_all = "lowercase")]
pub enum EncodingFormat {
Float,
Base64,
}
#[derive(Debug, Serialize, Clone, Deserialize)]
pub struct EmbeddingRequest {
pub model: String,
pub input: String,
pub input: Vec<String>,
pub encoding_format: Option<EncodingFormat>,
#[serde(skip_serializing_if = "Option::is_none")]
pub dimensions: Option<i32>,
#[serde(skip_serializing_if = "Option::is_none")]
@@ -22,10 +30,11 @@ pub struct EmbeddingRequest {
}
impl EmbeddingRequest {
pub fn new(model: String, input: String) -> Self {
pub fn new(model: String, input: Vec<String>) -> Self {
Self {
model,
input,
encoding_format: None,
dimensions: None,
user: None,
}