From dad94ea78a55ec62b95762bf87d49737e9fd9605 Mon Sep 17 00:00:00 2001
From: Florian Bernd <florian.bernd@elastic.co>
Date: Fri, 29 Sep 2023 13:21:28 +0200
Subject: [PATCH] Add `Similarity` to `KnnQuery`

---
 .../_Generated/Types/KnnQuery.g.cs            | 38 +++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/src/Elastic.Clients.Elasticsearch/_Generated/Types/KnnQuery.g.cs b/src/Elastic.Clients.Elasticsearch/_Generated/Types/KnnQuery.g.cs
index a0d239f009f..f0400fa5ba0 100644
--- a/src/Elastic.Clients.Elasticsearch/_Generated/Types/KnnQuery.g.cs
+++ b/src/Elastic.Clients.Elasticsearch/_Generated/Types/KnnQuery.g.cs
@@ -70,6 +70,12 @@ public sealed partial class KnnQuery
 	/// </summary>
 	[JsonInclude, JsonPropertyName("query_vector_builder")]
 	public Elastic.Clients.Elasticsearch.QueryVectorBuilder? QueryVectorBuilder { get; set; }
+
+	/// <summary>
+	/// <para>The minimum similarity for a vector to be considered a match</para>
+	/// </summary>
+	[JsonInclude, JsonPropertyName("similarity")]
+	public float? Similarity { get; set; }
 }
 
 public sealed partial class KnnQueryDescriptor<TDocument> : SerializableDescriptor<KnnQueryDescriptor<TDocument>>
@@ -92,6 +98,7 @@ public KnnQueryDescriptor() : base()
 	private Elastic.Clients.Elasticsearch.QueryVectorBuilder? QueryVectorBuilderValue { get; set; }
 	private QueryVectorBuilderDescriptor QueryVectorBuilderDescriptor { get; set; }
 	private Action<QueryVectorBuilderDescriptor> QueryVectorBuilderDescriptorAction { get; set; }
+	private float? SimilarityValue { get; set; }
 
 	/// <summary>
 	/// <para>Filters for the kNN search query</para>
@@ -213,6 +220,15 @@ public KnnQueryDescriptor<TDocument> QueryVectorBuilder(Action<QueryVectorBuilde
 		return Self;
 	}
 
+	/// <summary>
+	/// <para>The minimum similarity for a vector to be considered a match</para>
+	/// </summary>
+	public KnnQueryDescriptor<TDocument> Similarity(float? similarity)
+	{
+		SimilarityValue = similarity;
+		return Self;
+	}
+
 	protected override void Serialize(Utf8JsonWriter writer, JsonSerializerOptions options, IElasticsearchClientSettings settings)
 	{
 		writer.WriteStartObject();
@@ -279,6 +295,12 @@ protected override void Serialize(Utf8JsonWriter writer, JsonSerializerOptions o
 			JsonSerializer.Serialize(writer, QueryVectorBuilderValue, options);
 		}
 
+		if (SimilarityValue.HasValue)
+		{
+			writer.WritePropertyName("similarity");
+			writer.WriteNumberValue(SimilarityValue.Value);
+		}
+
 		writer.WriteEndObject();
 	}
 }
@@ -303,6 +325,7 @@ public KnnQueryDescriptor() : base()
 	private Elastic.Clients.Elasticsearch.QueryVectorBuilder? QueryVectorBuilderValue { get; set; }
 	private QueryVectorBuilderDescriptor QueryVectorBuilderDescriptor { get; set; }
 	private Action<QueryVectorBuilderDescriptor> QueryVectorBuilderDescriptorAction { get; set; }
+	private float? SimilarityValue { get; set; }
 
 	/// <summary>
 	/// <para>Filters for the kNN search query</para>
@@ -433,6 +456,15 @@ public KnnQueryDescriptor QueryVectorBuilder(Action<QueryVectorBuilderDescriptor
 		return Self;
 	}
 
+	/// <summary>
+	/// <para>The minimum similarity for a vector to be considered a match</para>
+	/// </summary>
+	public KnnQueryDescriptor Similarity(float? similarity)
+	{
+		SimilarityValue = similarity;
+		return Self;
+	}
+
 	protected override void Serialize(Utf8JsonWriter writer, JsonSerializerOptions options, IElasticsearchClientSettings settings)
 	{
 		writer.WriteStartObject();
@@ -499,6 +531,12 @@ protected override void Serialize(Utf8JsonWriter writer, JsonSerializerOptions o
 			JsonSerializer.Serialize(writer, QueryVectorBuilderValue, options);
 		}
 
+		if (SimilarityValue.HasValue)
+		{
+			writer.WritePropertyName("similarity");
+			writer.WriteNumberValue(SimilarityValue.Value);
+		}
+
 		writer.WriteEndObject();
 	}
 }
\ No newline at end of file