Skip to content

Commit e7e4593

Browse files
Add test for handling null user override in LlamaEmbeddingsModel
1 parent efeb10b commit e7e4593

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/llama/embeddings/LlamaEmbeddingsModelTests.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,15 @@ public void testOverrideWith_OverridesUser() {
7474
assertThat(overriddenModel, is(createEmbeddingsModel("model_name", "url", "api_key", "user_override", null, false)));
7575
}
7676

77+
public void testOverrideWith_OverridesNullUser() {
78+
var model = createEmbeddingsModel("model_name", "url", "api_key", null, null, false);
79+
var requestTaskSettingsMap = createRequestTaskSettingsMap("user_override");
80+
81+
var overriddenModel = LlamaEmbeddingsModel.of(model, requestTaskSettingsMap);
82+
83+
assertThat(overriddenModel, is(createEmbeddingsModel("model_name", "url", "api_key", "user_override", null, false)));
84+
}
85+
7786
public void testOverrideWith_EmptyMap() {
7887
var model = createEmbeddingsModel("model_name", "url", "api_key", "user", null, false);
7988

0 commit comments

Comments
 (0)