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

Allow url parameter for ollama embedder

Browse Source
This commit is contained in:
Louis Dureuil
2024-03-25 11:13:21 +01:00
parent dfa5e41ea6
commit 58972f35cb
4 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
milli/src/vector/ollama.rs
View File

@ -12,15 +12,12 @@ pub struct Embedder {
#[derive(Debug, Clone, Hash, PartialEq, Eq, serde::Deserialize, serde::Serialize)]
pub struct EmbedderOptions {
pub embedding_model: String,
pub url: Option<String>,
}
impl EmbedderOptions {
pub fn with_default_model() -> Self {
Self { embedding_model: "nomic-embed-text".into() }
}
pub fn with_embedding_model(embedding_model: String) -> Self {
Self { embedding_model }
pub fn with_default_model(url: Option<String>) -> Self {
Self { embedding_model: "nomic-embed-text".into(), url }
}
}
@ -31,7 +28,7 @@ impl Embedder {
api_key: None,
distribution: None,
dimensions: None,
url: get_ollama_path(),
url: options.url.unwrap_or_else(get_ollama_path),
query: serde_json::json!({
"model": model,
}),