diff --git a/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiApi.java b/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiApi.java index 751a63aee2..8c4248ba8f 100644 --- a/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiApi.java +++ b/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiApi.java @@ -28,14 +28,10 @@ import com.fasterxml.jackson.annotation.JsonInclude; import com.fasterxml.jackson.annotation.JsonInclude.Include; import com.fasterxml.jackson.annotation.JsonProperty; +import org.springframework.ai.model.*; import reactor.core.publisher.Flux; import reactor.core.publisher.Mono; -import org.springframework.ai.model.ApiKey; -import org.springframework.ai.model.ChatModelDescription; -import org.springframework.ai.model.ModelOptionsUtils; -import org.springframework.ai.model.NoopApiKey; -import org.springframework.ai.model.SimpleApiKey; import org.springframework.ai.openai.api.common.OpenAiApiConstants; import org.springframework.ai.retry.RetryUtils; import org.springframework.core.ParameterizedTypeReference; @@ -62,6 +58,7 @@ * @author Thomas Vitale * @author David Frizelle * @author Alexandros Pappas + * @author Chen Yang */ public class OpenAiApi { @@ -622,7 +619,7 @@ public enum ChatCompletionFinishReason { * OpenAI Embeddings Models: * Embeddings. */ - public enum EmbeddingModel { + public enum EmbeddingModel implements EmbeddingModelDescription { /** * Most capable embedding model for both english and non-english tasks. DIMENSION: @@ -651,6 +648,10 @@ public String getValue() { return this.value; } + @Override + public String getName() { + return this.value; + } } /**