Small adjustments to respect the spec

This commit is contained in:
Louis Dureuil 2023-12-13 23:25:38 +01:00
parent 5b51cb04af
commit a4536b1381
No known key found for this signature in database
2 changed files with 4 additions and 2 deletions

View File

@ -35,6 +35,8 @@ pub struct EmbedderOptions {
#[deserr(rename_all = camelCase, deny_unknown_fields)] #[deserr(rename_all = camelCase, deny_unknown_fields)]
pub enum EmbeddingModel { pub enum EmbeddingModel {
#[default] #[default]
#[serde(rename = "text-embedding-ada-002")]
#[deserr(rename = "text-embedding-ada-002")]
TextEmbeddingAda002, TextEmbeddingAda002,
} }

View File

@ -243,8 +243,8 @@ pub struct OpenAiEmbedderSettings {
#[serde(default, skip_serializing_if = "Setting::is_not_set")] #[serde(default, skip_serializing_if = "Setting::is_not_set")]
#[deserr(default)] #[deserr(default)]
pub api_key: Setting<String>, pub api_key: Setting<String>,
#[serde(default, skip_serializing_if = "Setting::is_not_set")] #[serde(default, skip_serializing_if = "Setting::is_not_set", rename = "model")]
#[deserr(default)] #[deserr(default, rename = "model")]
pub embedding_model: Setting<crate::vector::openai::EmbeddingModel>, pub embedding_model: Setting<crate::vector::openai::EmbeddingModel>,
} }