Fix default dimensions for models

This commit is contained in:
Louis Dureuil 2024-02-07 11:48:19 +01:00
parent 74c180267e
commit 32ee05ccef
No known key found for this signature in database

View File

@ -68,8 +68,8 @@ impl EmbeddingModel {
pub fn default_dimensions(&self) -> usize { pub fn default_dimensions(&self) -> usize {
match self { match self {
EmbeddingModel::TextEmbeddingAda002 => 1536, EmbeddingModel::TextEmbeddingAda002 => 1536,
EmbeddingModel::TextEmbedding3Large => 1536, EmbeddingModel::TextEmbedding3Large => 3072,
EmbeddingModel::TextEmbedding3Small => 3072, EmbeddingModel::TextEmbedding3Small => 1536,
} }
} }