Skip to content

Commit e40e41e

Browse files
author
Max Hniebergall
committed
Register named writeables from UnifiedCompletionRequest
1 parent be3a459 commit e40e41e

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/InferenceNamedWriteablesProvider.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import org.elasticsearch.inference.SecretSettings;
1717
import org.elasticsearch.inference.ServiceSettings;
1818
import org.elasticsearch.inference.TaskSettings;
19+
import org.elasticsearch.inference.UnifiedCompletionRequest;
1920
import org.elasticsearch.xpack.core.inference.results.ChatCompletionResults;
2021
import org.elasticsearch.xpack.core.inference.results.ErrorChunkedInferenceResults;
2122
import org.elasticsearch.xpack.core.inference.results.InferenceChunkedSparseEmbeddingResults;
@@ -137,11 +138,18 @@ public static List<NamedWriteableRegistry.Entry> getNamedWriteables() {
137138
addEisNamedWriteables(namedWriteables);
138139
addAlibabaCloudSearchNamedWriteables(namedWriteables);
139140

141+
addUnifiedNamedWriteables(namedWriteables);
142+
140143
namedWriteables.addAll(StreamingTaskManager.namedWriteables());
141144

142145
return namedWriteables;
143146
}
144147

148+
private static void addUnifiedNamedWriteables(List<NamedWriteableRegistry.Entry> namedWriteables) {
149+
var writeables = UnifiedCompletionRequest.getNamedWriteables();
150+
namedWriteables.addAll(writeables);
151+
}
152+
145153
private static void addAmazonBedrockNamedWriteables(List<NamedWriteableRegistry.Entry> namedWriteables) {
146154
namedWriteables.add(
147155
new NamedWriteableRegistry.Entry(

0 commit comments

Comments
 (0)