Skip to content

Commit 19b2c0c

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 f163c20 commit 19b2c0c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/external/request/voyageai/VoyageAIEmbeddingsRequestEntity.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import java.util.List;
1818
import java.util.Objects;
1919

20-
import static org.elasticsearch.xpack.inference.services.cohere.embeddings.CohereEmbeddingsTaskSettings.invalidInputTypeMessage;
20+
import static org.elasticsearch.xpack.inference.services.voyageai.embeddings.VoyageAIEmbeddingsTaskSettings.invalidInputTypeMessage;
2121

2222
public record VoyageAIEmbeddingsRequestEntity(
2323
List<String> input,

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
@@ -38,7 +38,7 @@ public class VoyageAIServiceSettings extends FilteredXContentObject implements S
3838
public static final String NAME = "voyageai_service_settings";
3939
public static final String MODEL_ID = "model_id";
4040
private static final Logger logger = LogManager.getLogger(VoyageAIServiceSettings.class);
41-
// See https://jina.ai/contact-sales/#rate-limit
41+
// See https://docs.voyageai.com/docs/rate-limits
4242
public static final RateLimitSettings DEFAULT_RATE_LIMIT_SETTINGS = new RateLimitSettings(2_000);
4343

4444
public static VoyageAIServiceSettings fromMap(Map<String, Object> map, ConfigurationParseContext context) {

0 commit comments

Comments
 (0)