update image and language input to become more flexible

Author: Barqawiz

core/com.intellijava.core/src/main/java/com/intellijava/core/model/input/ImageModelInput.java

@@ -37,7 +37,25 @@ private ImageModelInput(Builder builder) {
         this.numberOfImages = builder.numberOfImages;
         this.imageSize = builder.imageSize;
     }
+    
+    
     /**
+     * ImageModelInput default constructor.
+     * 
+     * @param prompt
+     * @param numberOfImages
+     * @param imageSize
+     */
+    public ImageModelInput(String prompt, int numberOfImages, String imageSize) {
+		super();
+		this.prompt = prompt;
+		this.numberOfImages = numberOfImages;
+		this.imageSize = imageSize;
+	}
+
+
+
+	/**
      * 
      * Builder class for ImageModelInput
      */
@@ -114,5 +132,35 @@ public int getNumberOfImages() {
     public String getImageSize() {
         return imageSize;
     }
+
+
+    /**
+     * Setter for prompt.
+     * 
+     * @param prompt
+     */
+	public void setPrompt(String prompt) {
+		this.prompt = prompt;
+	}
+
+
+	/**
+	 * Setter for numberOfImages.
+	 * @param numberOfImages
+	 */
+	public void setNumberOfImages(int numberOfImages) {
+		this.numberOfImages = numberOfImages;
+	}
+
+
+	/**
+	 * Setter for imageSize.
+	 * 
+	 * @param imageSize
+	 */
+	public void setImageSize(String imageSize) {
+		this.imageSize = imageSize;
+	}
+    
 }
 

core/com.intellijava.core/src/main/java/com/intellijava/core/model/input/LanguageModelInput.java

@@ -30,7 +30,29 @@ private LanguageModelInput(Builder builder) {
         this.maxTokens = builder.maxTokens;
         this.numberOfOutputs = builder.numberOfOutputs;
     }
+    
+    
     /**
+     * LanguageModelInput default constructor.
+     * 
+     * @param model
+     * @param prompt
+     * @param temperature
+     * @param maxTokens
+     * @param numberOfOutputs
+     */
+    public LanguageModelInput(String model, String prompt, float temperature, int maxTokens, int numberOfOutputs) {
+		super();
+		this.model = model;
+		this.prompt = prompt;
+		this.temperature = temperature;
+		this.maxTokens = maxTokens;
+		this.numberOfOutputs = numberOfOutputs;
+	}
+
+
+
+	/**
      * 
      * Builder class for LanguageModelInput.
      * 
@@ -85,7 +107,7 @@ public Builder setTemperature(float temperature) {
         }
 
         /**
-         *  Setter for maxTokens
+         *  Setter for maxTokens.
          * @param maxTokens maximum size of the model input and output. 
          * @return instance of Builder
          */
@@ -157,7 +179,53 @@ public int getMaxTokens() {
 	public int getNumberOfOutputs() {
 		return numberOfOutputs;
 	}
-    
+
+	/**
+	 * Setter for model.
+	 * 
+	 * @param model
+	 */
+	public void setModel(String model) {
+		this.model = model;
+	}
+
+	
+	/**
+	 * Setter for prompt.
+	 * 
+	 * @param prompt
+	 */
+	public void setPrompt(String prompt) {
+		this.prompt = prompt;
+	}
+
+	
+	/**
+	 * Setter for temperature.
+	 * 
+	 * @param temperature
+	 */
+	public void setTemperature(float temperature) {
+		this.temperature = temperature;
+	}
+
+	/**
+	 * Setter for maxTokens.
+	 * 
+	 * @param maxTokens
+	 */
+	public void setMaxTokens(int maxTokens) {
+		this.maxTokens = maxTokens;
+	}
+
+	/**
+	 * Setter for numberOfOutputs.
+	 * 
+	 * @param numberOfOutputs
+	 */
+	public void setNumberOfOutputs(int numberOfOutputs) {
+		this.numberOfOutputs = numberOfOutputs;
+	}
 
 }