snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-07-27 00:31:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ollama and openai impls to use the rest embedder internally

Browse Source
This commit is contained in:
Louis Dureuil
2024-03-19 15:41:37 +01:00
parent 8708cbef25
commit ac52c857e8
8 changed files with 394 additions and 779 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
milli/src/vector/settings.rs
View File

@ -204,7 +204,7 @@ impl From<EmbeddingConfig> for EmbeddingSettings {
},
super::EmbedderOptions::Ollama(options) => Self {
source: Setting::Set(EmbedderSource::Ollama),
model: Setting::Set(options.embedding_model.name().to_owned()),
model: Setting::Set(options.embedding_model.to_owned()),
revision: Setting::NotSet,
api_key: Setting::NotSet,
dimensions: Setting::NotSet,
@ -248,7 +248,7 @@ impl From<EmbeddingSettings> for EmbeddingConfig {
let mut options: ollama::EmbedderOptions =
super::ollama::EmbedderOptions::with_default_model();
if let Some(model) = model.set() {
options.embedding_model = super::ollama::EmbeddingModel::from_name(&model);
options.embedding_model = model;
}
this.embedder_options = super::EmbedderOptions::Ollama(options);
}