|
@@ -26,6 +26,8 @@ import org.elasticsearch.xpack.core.inference.results.SparseEmbeddingResults;
|
|
|
import org.elasticsearch.xpack.core.inference.results.TextEmbeddingByteResults;
|
|
|
import org.elasticsearch.xpack.core.inference.results.TextEmbeddingResults;
|
|
|
import org.elasticsearch.xpack.inference.services.azureopenai.AzureOpenAiSecretSettings;
|
|
|
+import org.elasticsearch.xpack.inference.services.azureopenai.completion.AzureOpenAiCompletionServiceSettings;
|
|
|
+import org.elasticsearch.xpack.inference.services.azureopenai.completion.AzureOpenAiCompletionTaskSettings;
|
|
|
import org.elasticsearch.xpack.inference.services.azureopenai.embeddings.AzureOpenAiEmbeddingsServiceSettings;
|
|
|
import org.elasticsearch.xpack.inference.services.azureopenai.embeddings.AzureOpenAiEmbeddingsTaskSettings;
|
|
|
import org.elasticsearch.xpack.inference.services.cohere.CohereServiceSettings;
|
|
@@ -237,6 +239,21 @@ public class InferenceNamedWriteablesProvider {
|
|
|
)
|
|
|
);
|
|
|
|
|
|
+ namedWriteables.add(
|
|
|
+ new NamedWriteableRegistry.Entry(
|
|
|
+ ServiceSettings.class,
|
|
|
+ AzureOpenAiCompletionServiceSettings.NAME,
|
|
|
+ AzureOpenAiCompletionServiceSettings::new
|
|
|
+ )
|
|
|
+ );
|
|
|
+ namedWriteables.add(
|
|
|
+ new NamedWriteableRegistry.Entry(
|
|
|
+ TaskSettings.class,
|
|
|
+ AzureOpenAiCompletionTaskSettings.NAME,
|
|
|
+ AzureOpenAiCompletionTaskSettings::new
|
|
|
+ )
|
|
|
+ );
|
|
|
+
|
|
|
return namedWriteables;
|
|
|
}
|
|
|
}
|