unify image model input

image model handles enum and string types.

Author: Barqawiz

core/com.intellijava.core/src/main/java/com/intellijava/core/controller/RemoteImageModel.java

@@ -23,44 +23,97 @@
 
 import com.intellijava.core.model.OpenaiImageResponse;
 import com.intellijava.core.model.OpenaiImageResponse.Data;
+import com.intellijava.core.model.SupportedImageModels;
 import com.intellijava.core.model.input.ImageModelInput;
 import com.intellijava.core.wrappers.OpenAIWrapper;
 
 /**
  * 
  * The RemoteImageModel class is used to generate images from text descriptions using an API key.
  * It currently supports OpenAI only. 
+ * 
  * The class uses the OpenAIWrapper to generate the images and returns a list of URLs for the generated images.
  *
  * @author github.com/Barqawiz
  */
 public class RemoteImageModel {
 
-	private String keyType;
+	private SupportedImageModels keyType;
 	private OpenAIWrapper openaiWrapper;
 
 	/**
 	 * 
 	 * Constructor for creating a new RemoteImageModel object.
 	 * 
-	 * Creates an instance of the class and sets up the API key and the key type.
+	 * Creates an instance of the class and set the API key value and type.
 	 * 
 	 * @param keyValue the API key.
-	 * @param keyType support openai only.
+	 * @param keyTypeString support openai only.
 	 * 
 	 * @throws IllegalArgumentException if the keyType passed is not "openai".
 	 * 
 	 */
-	public RemoteImageModel(String keyValue, String keyType) {
+	public RemoteImageModel(String keyValue, String keyTypeString) {
 
-		if (keyType.isEmpty() || keyType.equals("openai")) {
-			this.keyType = "openai";
-			openaiWrapper = new OpenAIWrapper(keyValue);
+		if (keyTypeString.isEmpty()) {
+			keyTypeString = SupportedImageModels.openai.toString();
+		}
+
+		List<String> supportedModels = this.getSupportedModels();
+
+		if (supportedModels.contains(keyTypeString)) {
+			this.initiate(keyValue, SupportedImageModels.valueOf(keyTypeString));
 		} else {
-			throw new IllegalArgumentException("This version support openai keyType only");
+			String models = String.join(" - ", supportedModels);
+			throw new IllegalArgumentException("The received keyValue not supported. Send any model from: " + models);
 		}
 	}
 
+	/**
+	 * 
+	 * Constructor for creating a new RemoteImageModel object.
+	 * 
+	 * Creates an instance of the class and set the API key value and type.
+	 * 
+	 * @param keyValue the API key.
+	 * @param keyType the model type from SupportedImageModels enum.
+	 */
+	public RemoteImageModel(String keyValue, SupportedImageModels keyType) { 
+		initiate(keyValue, keyType);
+	}
+	
+	/**
+	 * Get the supported models names as array of string
+	 * 
+	 * @return supportedModels
+	 */
+	public List<String> getSupportedModels() {
+		SupportedImageModels[] values = SupportedImageModels.values();
+		List<String> enumValues = new ArrayList<>();
+
+		for (int i = 0; i < values.length; i++) {
+			enumValues.add(values[i].name());
+		}
+
+		return enumValues;
+	}
+	
+	/**
+	 * Common function to initiate the class from any constructor.
+	 * 
+	 * @param keyValue the API key.
+	 * @param keyType enum of supported models.
+	 */
+	private void initiate(String keyValue, SupportedImageModels keyType) {
+		// set the model type
+		this.keyType = keyType;
+
+		// generate the related model
+		if (keyType.equals(SupportedImageModels.openai)) {
+			this.openaiWrapper = new OpenAIWrapper(keyValue);
+		} 
+	}
+	
 	/**
 	 * 
 	 * Generates images from a given text description.
@@ -71,7 +124,7 @@ public RemoteImageModel(String keyValue, String keyType) {
 	 */
 	public List<String> generateImages(ImageModelInput imageInput) throws IOException { 
 
-		if (this.keyType.equals("openai")) {
+		if (this.keyType == SupportedImageModels.openai) {
 			return this.generateOpenaiImage(imageInput.getPrompt(), 
 					imageInput.getNumberOfImages(), imageInput.getImageSize());
 		} else {

core/com.intellijava.core/src/main/java/com/intellijava/core/model/SupportedImageModels.java

@@ -0,0 +1,11 @@
+package com.intellijava.core.model;
+
+/**
+ * 
+ * Supported Image Models.
+ * 
+ * @author github.com/Barqawiz
+ */
+public enum SupportedImageModels {
+	/** openai model */openai
+}