prepare for 0.6.1 to return multi text output

prepare the RemoteLanguageModel to support multiple output responses.

Author: Barqawiz

core/com.intellijava.core/pom.xml

@@ -6,7 +6,7 @@
 
   <groupId>io.github.barqawiz</groupId>
   <artifactId>intellijava.core</artifactId>
-  <version>0.6.0</version>
+  <version>0.6.1</version>
 
   <name>Intellijava</name>
   <description>IntelliJava allows java developers to easily integrate with the latest language models, image generation, and deep learning frameworks.</description>

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

@@ -21,8 +21,11 @@
 import java.util.List;
 import java.util.Map;
 import com.intellijava.core.model.CohereLanguageResponse;
+import com.intellijava.core.model.CohereLanguageResponse.Generation;
 import com.intellijava.core.model.OpenaiLanguageResponse;
+import com.intellijava.core.model.OpenaiLanguageResponse.Choice;
 import com.intellijava.core.model.SupportedLangModels;
+import com.intellijava.core.model.OpenaiImageResponse.Data;
 import com.intellijava.core.model.input.LanguageModelInput;
 import com.intellijava.core.wrappers.CohereAIWrapper;
 import com.intellijava.core.wrappers.OpenAIWrapper;
@@ -143,11 +146,13 @@ private void initiate(String keyValue, SupportedLangModels keyType) {
 	public String generateText(LanguageModelInput langInput) throws IOException {
 
 		if (this.keyType.equals(SupportedLangModels.openai)) {
-			return this.generateOpenaiText(langInput.getModel(), langInput.getPrompt(), langInput.getTemperature(),
-					langInput.getMaxTokens());
+			return this.generateOpenaiText(langInput.getModel(), 
+					langInput.getPrompt(), langInput.getTemperature(),
+					langInput.getMaxTokens(), langInput.getNumberOfOutputs()).get(0);
 		} else if (this.keyType.equals(SupportedLangModels.cohere)) {
-			return this.generateCohereText(langInput.getModel(), langInput.getPrompt(), langInput.getTemperature(),
-					langInput.getMaxTokens());
+			return this.generateCohereText(langInput.getModel(), 
+					langInput.getPrompt(), langInput.getTemperature(),
+					langInput.getMaxTokens(), langInput.getNumberOfOutputs()).get(0);
 		} else {
 			throw new IllegalArgumentException("This version support openai keyType only");
 		}
@@ -163,11 +168,13 @@ public String generateText(LanguageModelInput langInput) throws IOException {
 	 * @param prompt      text of the required action or the question.
 	 * @param temperature higher values means more risks and creativity.
 	 * @param maxTokens   maximum size of the model input and output.
+	 * @param numberOfOutputs   number of model outputs.
 	 * @return string model response.
 	 * @throws IOException if there is an error when connecting to the OpenAI API.
 	 * 
 	 */
-	private String generateOpenaiText(String model, String prompt, float temperature, int maxTokens)
+	private List<String> generateOpenaiText(String model, String prompt, float temperature, 
+			int maxTokens, int numberOfOutputs)
 			throws IOException {
 
 		if (model.equals(""))
@@ -178,10 +185,16 @@ private String generateOpenaiText(String model, String prompt, float temperature
 		params.put("prompt", prompt);
 		params.put("temperature", temperature);
 		params.put("max_tokens", maxTokens);
+		params.put("n", numberOfOutputs);
 
 		OpenaiLanguageResponse resModel = (OpenaiLanguageResponse) openaiWrapper.generateText(params);
 
-		return resModel.getChoices().get(0).getText();
+		List<String> outputs = new ArrayList<>();
+		for (Choice item : resModel.getChoices()) {
+			outputs.add(item.getText());
+		}
+		
+		return outputs;
 
 	}
 
@@ -192,11 +205,13 @@ private String generateOpenaiText(String model, String prompt, float temperature
 	 * @param prompt      text of the required action or the question.
 	 * @param temperature higher values means more risks and creativity.
 	 * @param maxTokens   maximum size of the model input and output.
+	 * @param numberOfOutputs   number of model outputs.
 	 * @return string model response.
 	 * @throws IOException if there is an error when connecting to the API.
 	 * 
 	 */
-	private String generateCohereText(String model, String prompt, float temperature, int maxTokens)
+	private List<String> generateCohereText(String model, String prompt, float temperature, 
+			int maxTokens, int numberOfOutputs)
 			throws IOException {
 
 		if (model.equals(""))
@@ -207,10 +222,16 @@ private String generateCohereText(String model, String prompt, float temperature
 		params.put("prompt", prompt);
 		params.put("temperature", temperature);
 		params.put("max_tokens", maxTokens);
+		params.put("num_generations", numberOfOutputs);
 
 		CohereLanguageResponse resModel = (CohereLanguageResponse) cohereWrapper.generateText(params);
-
-		return resModel.getGenerations().get(0).getText();
+		
+		List<String> outputs = new ArrayList<>();
+		for (Generation item: resModel.getGenerations()) {
+			outputs.add(item.getText());
+		}
+		
+		return outputs;
 
 	}
 }

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

@@ -17,6 +17,7 @@ public class LanguageModelInput {
 	private String prompt; 
 	private float temperature; 
 	private int maxTokens;
+	private int numberOfOutputs = 1;
 
     /**
      * Private Constructor for the Builder.
@@ -27,6 +28,7 @@ private LanguageModelInput(Builder builder) {
         this.prompt = builder.prompt;
         this.temperature = builder.temperature;
         this.maxTokens = builder.maxTokens;
+        this.maxTokens = builder.numberOfOutputs;
     }
     /**
      * 
@@ -38,6 +40,7 @@ public static class Builder {
         private String prompt;
         private float temperature;
         private int maxTokens;
+        private int numberOfOutputs = 1;
 
         /**
          * Language input Constructor.
@@ -90,6 +93,22 @@ public Builder setMaxTokens(int maxTokens) {
             this.maxTokens = maxTokens;
             return this;
         }
+        
+        /**
+        *  Setter for numberOfOutputs
+        * @param numberOfOutputs number of model outputs, default value is 1. 
+        * 
+        * Cohere maximum value is five.
+        * 
+        * @return instance of Builder
+        */
+        public Builder setNumberOfOutputs(int numberOfOutputs) {
+        	if (this.numberOfOutputs < 0)
+        		this.numberOfOutputs = 0;
+        	
+    		this.numberOfOutputs = numberOfOutputs;
+    		return this;
+    	}
 
         /**
          * Build the final LanguageModelInput object.
@@ -130,5 +149,15 @@ public float getTemperature() {
     public int getMaxTokens() {
         return maxTokens;
     }
+
+    /**
+     * Getter for number of model outputs.
+     * @return numberOfOutputs
+     */
+	public int getNumberOfOutputs() {
+		return numberOfOutputs;
+	}
+    
+    
 }