Skip to content

Commit 8746cfa

Browse files
committed
Add names back
1 parent d50a74b commit 8746cfa

File tree

12 files changed

+61
-10
lines changed

12 files changed

+61
-10
lines changed

server/src/main/java/org/elasticsearch/index/mapper/blockloader/BlockLoaderFunctionConfig.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,11 @@
1919
* transforming loaded values into blocks.
2020
*/
2121
public interface BlockLoaderFunctionConfig {
22+
/**
23+
* Name used in descriptions.
24+
*/
25+
String name();
26+
2227
record Named(String name, Warnings warnings) implements BlockLoaderFunctionConfig {
2328
@Override
2429
public int hashCode() {

server/src/main/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapper.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3199,6 +3199,11 @@ public VectorSimilarityFunctionConfig forByteVector() {
31993199
return this;
32003200
}
32013201

3202+
@Override
3203+
public String name() {
3204+
return similarityFunction.toString();
3205+
}
3206+
32023207
public byte[] vectorAsBytes() {
32033208
assert vectorAsBytes != null : "vectorAsBytes is null, call forByteVector() first";
32043209
return vectorAsBytes;

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/blockloader/BlockLoaderExpression.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,12 @@ public interface BlockLoaderExpression {
4545
* "fusing" the expression into the load. Or null if the fusion isn't possible.
4646
*/
4747
@Nullable
48-
Fuse tryFuse(SearchStats stats);
48+
FusedExpression tryFuse(SearchStats stats);
4949

5050
/**
5151
* Fused load configuration.
5252
* @param field the field whose load we're fusing into
5353
* @param config the fusion configuration
5454
*/
55-
record Fuse(FieldAttribute field, BlockLoaderFunctionConfig config) {}
55+
record FusedExpression(FieldAttribute field, BlockLoaderFunctionConfig config) {}
5656
}

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/scalar/string/Length.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,13 +98,13 @@ public ExpressionEvaluator.Factory toEvaluator(ToEvaluator toEvaluator) {
9898
}
9999

100100
@Override
101-
public Fuse tryFuse(SearchStats stats) {
101+
public FusedExpression tryFuse(SearchStats stats) {
102102
if (field instanceof FieldAttribute f) {
103103
if (stats.hasDocValues(f.fieldName()) == false) {
104104
return null;
105105
}
106106
BlockLoaderWarnings warnings = new BlockLoaderWarnings(DriverContext.WarningsMode.COLLECT, source());
107-
return new Fuse(f, new BlockLoaderFunctionConfig.Named("LENGTH", warnings));
107+
return new FusedExpression(f, new BlockLoaderFunctionConfig.Named("LENGTH", warnings));
108108
}
109109
return null;
110110
}

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/vector/CosineSimilarity.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,11 @@ public float calculateSimilarity(byte[] leftVector, byte[] rightVector) {
4040
public float calculateSimilarity(float[] leftVector, float[] rightVector) {
4141
return VectorUtil.cosine(leftVector, rightVector);
4242
}
43+
44+
@Override
45+
public String toString() {
46+
return "Cosine";
47+
}
4348
};
4449

4550
@FunctionInfo(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/vector/DotProduct.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,11 @@ public float calculateSimilarity(byte[] leftVector, byte[] rightVector) {
4141
public float calculateSimilarity(float[] leftVector, float[] rightVector) {
4242
return VectorUtil.dotProduct(leftVector, rightVector);
4343
}
44+
45+
@Override
46+
public String toString() {
47+
return "DotProduct";
48+
}
4449
};
4550

4651
@FunctionInfo(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/vector/Hamming.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,11 @@ public float calculateSimilarity(byte[] leftVector, byte[] rightVector) {
3636
public float calculateSimilarity(float[] leftVector, float[] rightVector) {
3737
throw new UnsupportedOperationException("Hamming distance is not supported for float vectors");
3838
}
39+
40+
@Override
41+
public String toString() {
42+
return "Hamming";
43+
}
3944
};
4045
public static final DenseVectorFieldMapper.SimilarityFunction EVALUATOR_SIMILARITY_FUNCTION =
4146
new DenseVectorFieldMapper.SimilarityFunction() {
@@ -56,6 +61,11 @@ public float calculateSimilarity(float[] leftVector, float[] rightVector) {
5661
}
5762
return Hamming.calculateSimilarity(a, b);
5863
}
64+
65+
@Override
66+
public String toString() {
67+
return "Hamming";
68+
}
5969
};
6070

6171
@FunctionInfo(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/vector/L1Norm.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,11 @@ public float calculateSimilarity(float[] leftVector, float[] rightVector) {
4949
}
5050
return result;
5151
}
52+
53+
@Override
54+
public String toString() {
55+
return "L1Norm";
56+
}
5257
};
5358

5459
@FunctionInfo(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/vector/L2Norm.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,11 @@ public float calculateSimilarity(byte[] leftVector, byte[] rightVector) {
3737
public float calculateSimilarity(float[] leftVector, float[] rightVector) {
3838
return (float) Math.sqrt(VectorUtil.squareDistance(leftVector, rightVector));
3939
}
40+
41+
@Override
42+
public String toString() {
43+
return "L1Norm";
44+
}
4045
};
4146

4247
@FunctionInfo(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/vector/VectorSimilarityFunction.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ public void close() {
208208
}
209209

210210
@Override
211-
public final Fuse tryFuse(SearchStats stats) {
211+
public final FusedExpression tryFuse(SearchStats stats) {
212212
// Bail if we're not directly comparing a field with a literal.
213213
Literal literal;
214214
FieldAttribute field;
@@ -233,7 +233,7 @@ public final Fuse tryFuse(SearchStats stats) {
233233
vectorArray[i] = ((Number) vectorList.get(i)).floatValue();
234234
}
235235

236-
return new Fuse(field, new DenseVectorFieldMapper.VectorSimilarityFunctionConfig(getSimilarityFunction(), vectorArray));
236+
return new FusedExpression(field, new DenseVectorFieldMapper.VectorSimilarityFunctionConfig(getSimilarityFunction(), vectorArray));
237237
}
238238

239239
interface VectorValueProvider extends Releasable {

0 commit comments

Comments
 (0)