From 76968179defd9f575a89ff04c685624c13d134d5 Mon Sep 17 00:00:00 2001
From: Kerollmops <clement@meilisearch.com>
Date: Tue, 11 Mar 2025 15:33:29 +0100
Subject: [PATCH] Fix the upgrade arroy calls

---
 crates/meilitool/src/upgrade/v1_11.rs    | 4 ++--
 crates/milli/src/update/upgrade/v1_14.rs | 9 +++++----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/crates/meilitool/src/upgrade/v1_11.rs b/crates/meilitool/src/upgrade/v1_11.rs
index dea768dd5..8b653a55e 100644
--- a/crates/meilitool/src/upgrade/v1_11.rs
+++ b/crates/meilitool/src/upgrade/v1_11.rs
@@ -78,9 +78,9 @@ pub fn v1_10_to_v1_11(
 
         meilisearch_types::milli::arroy::upgrade::cosine_from_0_4_to_0_5(
             &index_rtxn,
-            index_read_database,
+            index_read_database.remap_types(),
             &mut index_wtxn,
-            index_write_database,
+            index_write_database.remap_types(),
         )?;
 
         index_wtxn.commit()?;
diff --git a/crates/milli/src/update/upgrade/v1_14.rs b/crates/milli/src/update/upgrade/v1_14.rs
index 832f9ec83..aba95a655 100644
--- a/crates/milli/src/update/upgrade/v1_14.rs
+++ b/crates/milli/src/update/upgrade/v1_14.rs
@@ -1,3 +1,4 @@
+use arroy::distances::Cosine;
 use heed::RwTxn;
 
 use super::UpgradeIndex;
@@ -24,11 +25,11 @@ impl UpgradeIndex for Latest_V1_13_To_Latest_V1_14 {
         progress.update_progress(VectorStore::UpdateInternalVersions);
 
         let rtxn = index.read_txn()?;
-        arroy::upgrade::cosine_from_0_5_to_0_6(
+        arroy::upgrade::from_0_5_to_0_6::<Cosine>(
             &rtxn,
-            index.vector_arroy,
-            &mut wtxn,
-            index.vector_arroy,
+            index.vector_arroy.remap_data_type(),
+            wtxn,
+            index.vector_arroy.remap_data_type(),
         )?;
 
         Ok(true)