From 7db21fd89f9a98ce58e31b35a613048df42257be Mon Sep 17 00:00:00 2001 From: philippus Date: Sat, 13 Jul 2024 13:53:47 +0200 Subject: [PATCH] Make confidenceInterval a float --- .../com/sksamuel/elastic4s/fields/DenseVectorField.scala | 4 +++- .../elastic4s/handlers/fields/DenseVectorFieldBuilderFn.scala | 4 ++-- 2 files changed, 5 insertions(+), 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 index cb1e01e04..7c91c1e3c 100644 --- 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/elastic4s-handlers/src/main/scala/com/sksamuel/elastic4s/handlers/fields/DenseVectorFieldBuilderFn.scala b/elastic4s-handlers/src/main/scala/com/sksamuel/elastic4s/handlers/fields/DenseVectorFieldBuilderFn.scala index b9cfd6c19..1fa27c03d 100644 --- a/elastic4s-handlers/src/main/scala/com/sksamuel/elastic4s/handlers/fields/DenseVectorFieldBuilderFn.scala +++ b/elastic4s-handlers/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) ) }