|
|
@@ -32,7 +32,6 @@ import org.elasticsearch.xpack.core.inference.action.InferenceAction;
|
|
|
import org.elasticsearch.xpack.inference.chunking.ChunkingSettingsBuilder;
|
|
|
import org.elasticsearch.xpack.inference.chunking.EmbeddingRequestChunker;
|
|
|
import org.elasticsearch.xpack.inference.external.action.SenderExecutableAction;
|
|
|
-import org.elasticsearch.xpack.inference.external.action.openai.OpenAiActionCreator;
|
|
|
import org.elasticsearch.xpack.inference.external.http.retry.ResponseHandler;
|
|
|
import org.elasticsearch.xpack.inference.external.http.sender.EmbeddingsInput;
|
|
|
import org.elasticsearch.xpack.inference.external.http.sender.GenericRequestManager;
|
|
|
@@ -46,6 +45,7 @@ import org.elasticsearch.xpack.inference.services.ConfigurationParseContext;
|
|
|
import org.elasticsearch.xpack.inference.services.SenderService;
|
|
|
import org.elasticsearch.xpack.inference.services.ServiceComponents;
|
|
|
import org.elasticsearch.xpack.inference.services.ServiceUtils;
|
|
|
+import org.elasticsearch.xpack.inference.services.openai.action.OpenAiActionCreator;
|
|
|
import org.elasticsearch.xpack.inference.services.openai.completion.OpenAiChatCompletionModel;
|
|
|
import org.elasticsearch.xpack.inference.services.openai.embeddings.OpenAiEmbeddingsModel;
|
|
|
import org.elasticsearch.xpack.inference.services.openai.embeddings.OpenAiEmbeddingsServiceSettings;
|
|
|
@@ -60,7 +60,6 @@ import java.util.Map;
|
|
|
import java.util.Set;
|
|
|
|
|
|
import static org.elasticsearch.xpack.inference.external.action.ActionUtils.constructFailedToSendRequestMessage;
|
|
|
-import static org.elasticsearch.xpack.inference.external.action.openai.OpenAiActionCreator.COMPLETION_ERROR_PREFIX;
|
|
|
import static org.elasticsearch.xpack.inference.services.ServiceFields.DIMENSIONS;
|
|
|
import static org.elasticsearch.xpack.inference.services.ServiceFields.MODEL_ID;
|
|
|
import static org.elasticsearch.xpack.inference.services.ServiceUtils.createInvalidModelException;
|
|
|
@@ -72,6 +71,7 @@ import static org.elasticsearch.xpack.inference.services.ServiceUtils.throwIfNot
|
|
|
import static org.elasticsearch.xpack.inference.services.ServiceUtils.useChatCompletionUrlMessage;
|
|
|
import static org.elasticsearch.xpack.inference.services.openai.OpenAiServiceFields.EMBEDDING_MAX_BATCH_SIZE;
|
|
|
import static org.elasticsearch.xpack.inference.services.openai.OpenAiServiceFields.ORGANIZATION;
|
|
|
+import static org.elasticsearch.xpack.inference.services.openai.action.OpenAiActionCreator.COMPLETION_ERROR_PREFIX;
|
|
|
|
|
|
public class OpenAiService extends SenderService {
|
|
|
public static final String NAME = "openai";
|