Skip to content

Commit 18ee182

Browse files
Update embeddings handler to use HuggingFace response entity
1 parent ad009c6 commit 18ee182

File tree

2 files changed

+2
-37
lines changed

2 files changed

+2
-37
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/llama/action/LlamaActionCreator.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,13 @@
1818
import org.elasticsearch.xpack.inference.external.http.sender.Sender;
1919
import org.elasticsearch.xpack.inference.external.http.sender.UnifiedChatInput;
2020
import org.elasticsearch.xpack.inference.services.ServiceComponents;
21+
import org.elasticsearch.xpack.inference.services.huggingface.response.HuggingFaceEmbeddingsResponseEntity;
2122
import org.elasticsearch.xpack.inference.services.llama.completion.LlamaChatCompletionModel;
2223
import org.elasticsearch.xpack.inference.services.llama.completion.LlamaCompletionResponseHandler;
2324
import org.elasticsearch.xpack.inference.services.llama.embeddings.LlamaEmbeddingsModel;
2425
import org.elasticsearch.xpack.inference.services.llama.embeddings.LlamaEmbeddingsResponseHandler;
2526
import org.elasticsearch.xpack.inference.services.llama.request.completion.LlamaChatCompletionRequest;
2627
import org.elasticsearch.xpack.inference.services.llama.request.embeddings.LlamaEmbeddingsRequest;
27-
import org.elasticsearch.xpack.inference.services.llama.response.embeddings.LlamaEmbeddingsResponseEntity;
2828
import org.elasticsearch.xpack.inference.services.openai.response.OpenAiChatCompletionResponseEntity;
2929

3030
import java.util.Objects;
@@ -44,7 +44,7 @@ public class LlamaActionCreator implements LlamaActionVisitor {
4444

4545
private static final ResponseHandler EMBEDDINGS_HANDLER = new LlamaEmbeddingsResponseHandler(
4646
"llama text embedding",
47-
LlamaEmbeddingsResponseEntity::fromResponse
47+
HuggingFaceEmbeddingsResponseEntity::fromResponse
4848
);
4949
private static final ResponseHandler COMPLETION_HANDLER = new LlamaCompletionResponseHandler(
5050
"llama completion",

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/llama/response/embeddings/LlamaEmbeddingsResponseEntity.java

Lines changed: 0 additions & 35 deletions
This file was deleted.

0 commit comments

Comments
 (0)