From 054e998ff49a1f3abbc0a474d9a8d23c0ed4b17b Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 21 Jun 2024 19:45:55 +0800 Subject: feat: serve embeddings api (#624) --- src/client/model.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/client/model.rs') diff --git a/src/client/model.rs b/src/client/model.rs index ebf1264..06f5577 100644 --- a/src/client/model.rs +++ b/src/client/model.rs @@ -8,7 +8,7 @@ use crate::config::Config; use crate::utils::{estimate_token_length, format_option_value}; use anyhow::{bail, Result}; -use serde::Deserialize; +use serde::{Deserialize, Serialize}; const PER_MESSAGES_TOKENS: usize = 5; const BASIS_TOKENS: usize = 2; @@ -242,7 +242,7 @@ impl Model { } } -#[derive(Debug, Clone, Default, Deserialize)] +#[derive(Debug, Clone, Default, Serialize, Deserialize)] pub struct ModelData { pub name: String, #[serde(default = "default_model_mode")] -- cgit v1.2.3