Make confidenceInterval a float

Philippus · Jul 13, 2024 · 7db21fd · 7db21fd
1 parent 32c3749
commit 7db21fd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/elastic4s-domain/src/main/scala/com/sksamuel/elastic4s/fields/DenseVectorField.scala b/elastic4s-domain/src/main/scala/com/sksamuel/elastic4s/fields/DenseVectorField.scala
@@ -39,7 +39,9 @@ case object DotProduct extends Similarity { val name = "dot_product" }
 case object Cosine extends Similarity { val name = "cosine" }
 case object MaxInnerProduct extends Similarity { val name = "max_inner_product" }
 
-case class DenseVectorIndexOptions(`type`: DenseVectorField.KnnType, m: Option[Int] = None, efConstruction: Option[Int] = None, confidenceInterval: Option[Double] = None)
+case class DenseVectorIndexOptions(`type`: DenseVectorField.KnnType, m: Option[Int] = None, efConstruction: Option[Int] = None, confidenceInterval: Option[Float] = None) {
+
+}
 
 case class DenseVectorField(name: String,
                             elementType: Option[String] = None,

diff --git a/...ers/src/main/scala/com/sksamuel/elastic4s/handlers/fields/DenseVectorFieldBuilderFn.scala b/...ers/src/main/scala/com/sksamuel/elastic4s/handlers/fields/DenseVectorFieldBuilderFn.scala
@@ -23,7 +23,7 @@ object DenseVectorFieldBuilderFn {
         DenseVectorField.Int8Hnsw,
         values.get("m").map(_.asInstanceOf[Int]),
         values.get("ef_construction").map(_.asInstanceOf[Int]),
-        values.get("confidence_interval").map(_.asInstanceOf[Double])
+        values.get("confidence_interval").map(d => d.asInstanceOf[Double].toFloat)
       )
       case "flat" => DenseVectorIndexOptions(
         DenseVectorField.Flat
@@ -32,7 +32,7 @@ object DenseVectorFieldBuilderFn {
         DenseVectorField.Int8Flat,
         None,
         None,
-        values.get("confidence_interval").map(_.asInstanceOf[Double])
+        values.get("confidence_interval").map(d => d.asInstanceOf[Double].toFloat)
       )
     }