searchengines-web/MeiliSearch
searchengines-web/MeiliSearch
1
0
Fork 0
mirror of https://github.com/meilisearch/MeiliSearch synced 2025-02-03 17:13:30 +01:00
Code Issues Releases Wiki Activity

Swap implementations of ollama

Browse Source
This commit is contained in:
Louis Dureuil 2025-01-20 22:22:22 +01:00
parent 0776217801
commit 4709c638ed
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
crates/milli/src/vector/ollama.rs
View File

@ -45,11 +45,6 @@ impl EmbedderOptions {
// **warning**: do not swap these two `if`s, as the second one is always true when the first one is. // **warning**: do not swap these two `if`s, as the second one is always true when the first one is.
let (request, response) = if url.ends_with("/api/embeddings") { let (request, response) = if url.ends_with("/api/embeddings") {
(
serde_json::json!({"model": model, "input": [super::rest::REQUEST_PLACEHOLDER, super::rest::REPEAT_PLACEHOLDER]}),
serde_json::json!({"embeddings": [super::rest::RESPONSE_PLACEHOLDER, super::rest::REPEAT_PLACEHOLDER]}),
)
} else if url.ends_with("/api/embed") {
( (
serde_json::json!({ serde_json::json!({
"model": model, "model": model,
@ -59,6 +54,11 @@ impl EmbedderOptions {
"embedding": super::rest::RESPONSE_PLACEHOLDER, "embedding": super::rest::RESPONSE_PLACEHOLDER,
}), }),
) )
} else if url.ends_with("/api/embed") {
(
serde_json::json!({"model": model, "input": [super::rest::REQUEST_PLACEHOLDER, super::rest::REPEAT_PLACEHOLDER]}),
serde_json::json!({"embeddings": [super::rest::RESPONSE_PLACEHOLDER, super::rest::REPEAT_PLACEHOLDER]}),
)
} else { } else {
return Err(NewEmbedderError::ollama_unsupported_url(url)); return Err(NewEmbedderError::ollama_unsupported_url(url));
}; };