snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-22 18:17:39 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

refactor(vector): improve ollama support

Browse Source
This commit is contained in:
Guillaume Loulier 2024-10-04 11:28:03 +02:00
parent 386ca86297
commit 29e3b47017
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -47,10 +47,10 @@ impl Embedder {
url: options.url.unwrap_or_else(get_ollama_path), url: options.url.unwrap_or_else(get_ollama_path),
request: serde_json::json!({ request: serde_json::json!({
"model": model, "model": model,
"prompt": super::rest::REQUEST_PLACEHOLDER, "input": super::rest::REQUEST_PLACEHOLDER,
}), }),
response: serde_json::json!({ response: serde_json::json!({
"embedding": super::rest::RESPONSE_PLACEHOLDER, "embeddings": super::rest::RESPONSE_PLACEHOLDER,
}), }),
headers: Default::default(), headers: Default::default(),
}, },
@ -118,6 +118,6 @@ impl Embedder {
} }
fn get_ollama_path() -> String { fn get_ollama_path() -> String {
// Important: Hostname not enough, has to be entire path to embeddings endpoint // Important: Hostname not enough, has to be entire path to embeddings endpoint (endpoint can be found in the API documentation: https://github.com/ollama/ollama/blob/main/docs/api.md#generate-embeddings)
std::env::var("MEILI_OLLAMA_URL").unwrap_or("http://localhost:11434/api/embeddings".to_string()) std::env::var("MEILI_OLLAMA_URL").unwrap_or("http://localhost:11434/api/embed".to_string())
} }