diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-21 19:45:55 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-21 19:45:55 +0800 |
| commit | 054e998ff49a1f3abbc0a474d9a8d23c0ed4b17b (patch) | |
| tree | f1134647bea4b524babfa1508f8e0ce74c6ad3ca /src/client | |
| parent | 97c82e565fc1d2e2b6a95b60beacbbe3c4708a39 (diff) | |
| download | aichat-054e998ff49a1f3abbc0a474d9a8d23c0ed4b17b.tar.gz | |
feat: serve embeddings api (#624)
Diffstat (limited to 'src/client')
| -rw-r--r-- | src/client/model.rs | 4 |
1 files changed, 2 insertions, 2 deletions
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")] |