Skip to content
Merged
Changes from 34 commits
Commits
Show all changes
36 commits
Select commit Hold shift + click to select a range
0535410
VoyageAI embeddings and rerank:
fzowl Feb 3, 2025
07c39a0
VoyageAI embeddings and rerank:
fzowl Feb 3, 2025
91dee7f
VoyageAI embeddings and rerank:
fzowl Feb 3, 2025
6f11414
VoyageAI embeddings and rerank:
fzowl Feb 3, 2025
050d5b2
Adding initial tests
fzowl Feb 4, 2025
b94bc5f
Correcting the TransportVersions.java
fzowl Feb 5, 2025
be1e9cf
Correcting due to comments
fzowl Feb 5, 2025
71dfdc8
Adding BIT support
fzowl Feb 5, 2025
8f6e03b
Initial tests
fzowl Feb 7, 2025
d41538a
More tests
fzowl Feb 8, 2025
3f1a75a
More tests/corrections
fzowl Feb 9, 2025
a89fdf1
Removing warnings
fzowl Feb 9, 2025
b7cb871
Further tests
fzowl Feb 9, 2025
b9681af
Transport version correction
fzowl Feb 9, 2025
bcca709
Merge pull request #2 from voyage-ai/voyageai
fzowl Feb 9, 2025
c211583
Adding changelog and correcting TransportVersions
fzowl Feb 9, 2025
a14ea32
Merge branch 'main' into main
fzowl Feb 10, 2025
91de4c3
Spotless tests
fzowl Feb 11, 2025
cb9fd17
Changes due to the comments
fzowl Feb 16, 2025
01c7632
Merge branch 'main' into main
fzowl Feb 16, 2025
850ab5d
Changes due to the comments
fzowl Feb 18, 2025
d6042e7
Correcting QA tests
fzowl Feb 19, 2025
1ebe1a4
Correcting QA tests
fzowl Feb 19, 2025
8796053
Merge branch 'main' of github.com:elastic/elasticsearch into voyage-a…
jonathan-buttner Feb 20, 2025
0a5b7c0
Merge branch 'main' into main
fzowl Feb 20, 2025
29fd189
Merge branch 'main' into main
fzowl Feb 20, 2025
baf9e3a
Merge branch 'main' into main
fzowl Feb 20, 2025
2a27c1d
Merge branch 'main' into main
jonathan-buttner Feb 20, 2025
b1d5d7a
Merge branch 'elastic:main' into main
fzowl Feb 20, 2025
0efc61b
Merge branch 'elastic:main' into main
fzowl May 21, 2025
9abaa23
Adding VoyageAI's v3.5 models
fzowl May 21, 2025
d76e810
Merge pull request #3 from voyage-ai/voyageai-v3.5-models
fzowl May 21, 2025
80631b2
Merge branch 'main' into main
fzowl May 21, 2025
f5a1f10
Merge branch 'main' into main
fzowl May 21, 2025
41ef1f0
Adding VoyageAI's v3.5 models
fzowl May 26, 2025
b59c747
Adding VoyageAI's v3.5 models
fzowl May 26, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -21,33 +21,32 @@
import org.elasticsearch.xpack.inference.services.voyageai.action.VoyageAIActionVisitor;

import java.net.URI;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

public abstract class VoyageAIModel extends RateLimitGroupingModel {
private static final String DEFAULT_MODEL_FAMILY = "default_model_family";
private static final Map<String, String> MODEL_TO_MODEL_FAMILY = Map.of(
"voyage-multimodal-3",
"embed_multimodal",
"voyage-3-large",
"embed_large",
"voyage-code-3",
"embed_large",
"voyage-3",
"embed_medium",
"voyage-3-lite",
"embed_small",
"voyage-finance-2",
"embed_large",
"voyage-law-2",
"embed_large",
"voyage-code-2",
"embed_large",
"rerank-2",
"rerank_large",
"rerank-2-lite",
"rerank_small"
);
private static final Map<String, String> MODEL_TO_MODEL_FAMILY;

static {
Map<String, String> tempMap = new HashMap<>();
tempMap.put("voyage-3.5", "embed_medium");
tempMap.put("voyage-3.5-lite", "embed_small");
tempMap.put("voyage-multimodal-3", "embed_multimodal");
tempMap.put("voyage-3-large", "embed_large");
tempMap.put("voyage-code-3", "embed_large");
tempMap.put("voyage-3", "embed_medium");
tempMap.put("voyage-3-lite", "embed_small");
tempMap.put("voyage-finance-2", "embed_large");
tempMap.put("voyage-law-2", "embed_large");
tempMap.put("voyage-code-2", "embed_large");
tempMap.put("rerank-2", "rerank_large");
tempMap.put("rerank-2-lite", "rerank_small");

MODEL_TO_MODEL_FAMILY = Collections.unmodifiableMap(tempMap);
}

private final SecureString apiKey;
private final VoyageAIRateLimitServiceSettings rateLimitServiceSettings;
Expand Down