Skip to content

Commit 91dee7f

Browse files
committed
VoyageAI embeddings and rerank:
- embeddings works, tested - rerank works, tested (https://www.elastic.co/search-labs/blog/elasticsearch-cohere-rerank) What's missing: - unit and integration tests
1 parent 07c39a0 commit 91dee7f

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed

server/src/main/java/org/elasticsearch/TransportVersions.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,14 +172,14 @@ static TransportVersion def(int id) {
172172
public static final TransportVersion TIMEOUT_GET_PARAM_FOR_RESOLVE_CLUSTER = def(8_838_0_00);
173173
public static final TransportVersion INFERENCE_REQUEST_ADAPTIVE_RATE_LIMITING = def(8_839_0_00);
174174
public static final TransportVersion ML_INFERENCE_IBM_WATSONX_RERANK_ADDED = def(8_840_0_00);
175-
public static final TransportVersion COHERE_BIT_EMBEDDING_TYPE_SUPPORT_ADDED_BACKPORT_8_X = def(8_840_0_01);
176175
public static final TransportVersion ELASTICSEARCH_9_0 = def(9_000_0_00);
177176
public static final TransportVersion COHERE_BIT_EMBEDDING_TYPE_SUPPORT_ADDED = def(9_001_0_00);
178177
public static final TransportVersion REMOVE_SNAPSHOT_FAILURES = def(9_002_0_00);
179178
public static final TransportVersion TRANSPORT_STATS_HANDLING_TIME_REQUIRED = def(9_003_0_00);
180179

181180
public static final TransportVersion REMOVE_DESIRED_NODE_VERSION = def(9_004_0_00);
182181

182+
public static final TransportVersion VOYAGE_AI_INTEGRATION_ADDED = def(9_002_0_00);
183183
/*
184184
* STOP! READ THIS FIRST! No, really,
185185
* ____ _____ ___ ____ _ ____ _____ _ ____ _____ _ _ ___ ____ _____ ___ ____ ____ _____ _

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/voyageai/VoyageAIService.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ static SimilarityMeasure defaultSimilarity() {
346346

347347
@Override
348348
public TransportVersion getMinimalSupportedVersion() {
349-
return TransportVersions.JINA_AI_INTEGRATION_ADDED;
349+
return TransportVersions.VOYAGE_AI_INTEGRATION_ADDED;
350350
}
351351

352352
public static class Configuration {

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/voyageai/VoyageAIServiceSettings.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ public XContentBuilder toXContentFragmentOfExposedFields(XContentBuilder builder
131131

132132
@Override
133133
public TransportVersion getMinimalSupportedVersion() {
134-
return TransportVersions.JINA_AI_INTEGRATION_ADDED;
134+
return TransportVersions.VOYAGE_AI_INTEGRATION_ADDED;
135135
}
136136

137137
@Override

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/voyageai/embeddings/VoyageAIEmbeddingType.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
import java.util.Map;
1919

2020
/**
21-
* Defines the type of embedding that the cohere api should return for a request.
21+
* Defines the type of embedding that the VoyageAI api should return for a request.
2222
*
2323
* <p>
2424
* <a href="https://docs.voyageai.com/reference/embeddings-api">See api docs for details.</a>
@@ -89,7 +89,7 @@ public static VoyageAIEmbeddingType fromElementType(DenseVectorFieldMapper.Eleme
8989

9090
throw new IllegalArgumentException(
9191
Strings.format(
92-
"Element type [%s] does not map to a Cohere embedding value, must be one of [%s]",
92+
"Element type [%s] does not map to a VoyageAI embedding value, must be one of [%s]",
9393
elementType,
9494
String.join(", ", validElementTypes)
9595
)

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/voyageai/rerank/VoyageAIRerankServiceSettings.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ protected XContentBuilder toXContentFragmentOfExposedFields(XContentBuilder buil
9090

9191
@Override
9292
public TransportVersion getMinimalSupportedVersion() {
93-
return TransportVersions.JINA_AI_INTEGRATION_ADDED;
93+
return TransportVersions.VOYAGE_AI_INTEGRATION_ADDED;
9494
}
9595

9696
@Override

0 commit comments

Comments
 (0)