The EmbeddingModel enum class of OpenAI implements the EmbeddingModel…

…Description interface.
spring-projects · Feb 19, 2025 · 600a337 · 600a337
1 parent 8e23422
commit 600a337
Showing 1 changed file with 7 additions and 6 deletions.
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
@@ -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:
 	 * <a href="https://platform.openai.com/docs/models/embeddings">Embeddings</a>.
 	 */
-	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;
+		}
 	}
 
 	/**