Gemini ChatModel Added

pom.xml

@@ -5,7 +5,7 @@
 	<modelVersion>4.0.0</modelVersion>
 	<groupId>com.mulesoft.connectors</groupId>
 	<artifactId>mule4-aichain-connector</artifactId>
-	<version>1.1.10-SNAPSHOT</version>
+	<version>1.1.28-SNAPSHOT</version>
 	<packaging>mule-extension</packaging>
 	<name>MuleSoft AI Chain Connector - Mule 4</name>
 
@@ -200,6 +200,7 @@
 			<artifactId>langchain4j-mistral-ai</artifactId>
 			<version>${langchain4j.version}</version>
 		</dependency>
+
 		<dependency>
 			<groupId>dev.langchain4j</groupId>
 			<artifactId>langchain4j-hugging-face</artifactId>
@@ -241,6 +242,11 @@
 			<artifactId>langchain4j-azure-open-ai</artifactId>
 			<version>${langchain4j.version}</version>
 		</dependency>
+		<dependency>
+			<groupId>dev.langchain4j</groupId>
+			<artifactId>langchain4j-google-ai-gemini</artifactId>
+			<version>${langchain4j.version}</version>
+		</dependency>
 		<dependency>
 			<groupId>dev.langchain4j</groupId>
 			<artifactId>langchain4j-easy-rag</artifactId>

src/main/java/org/mule/extension/mulechain/internal/config/LangchainLLMConfiguration.java

@@ -68,19 +68,25 @@ public class LangchainLLMConfiguration implements Initialisable {
 
   @Parameter
   @Placement(order = 6, tab = Placement.DEFAULT_TAB)
+  @Optional(defaultValue = "0.95")
+  private double topP = 0.95;
+
+
+  @Parameter
+  @Placement(order = 7, tab = Placement.DEFAULT_TAB)
   @Optional(defaultValue = "60")
   @DisplayName("LLM timeout")
   private int llmTimeout = 60;
 
   @Parameter
   @Optional(defaultValue = "SECONDS")
-  @Placement(order = 7, tab = Placement.DEFAULT_TAB)
+  @Placement(order = 8, tab = Placement.DEFAULT_TAB)
   @DisplayName("LLM timeout unit")
   @Summary("Time unit to be used in the LLM Timeout")
   private TimeUnit llmTimeoutUnit = TimeUnit.SECONDS;
 
   @Parameter
-  @Placement(order = 8, tab = Placement.DEFAULT_TAB)
+  @Placement(order = 9, tab = Placement.DEFAULT_TAB)
   @Expression(ExpressionSupport.SUPPORTED)
   @Optional(defaultValue = "500")
   private int maxTokens = 500;
@@ -109,6 +115,11 @@ public double getTemperature() {
     return temperature;
   }
 
+  public double getTopP() {
+    return topP;
+  }
+
+
   public int getLlmTimeout() {
     return llmTimeout;
   }

src/main/java/org/mule/extension/mulechain/internal/config/util/LangchainLLMInitializerUtil.java

@@ -5,6 +5,7 @@
 
 import dev.langchain4j.model.anthropic.AnthropicChatModel;
 import dev.langchain4j.model.azure.AzureOpenAiChatModel;
+import dev.langchain4j.model.googleai.GoogleAiGeminiChatModel;
 import dev.langchain4j.model.huggingface.HuggingFaceChatModel;
 import dev.langchain4j.model.mistralai.MistralAiChatModel;
 import dev.langchain4j.model.ollama.OllamaChatModel;
@@ -26,6 +27,7 @@ public static OpenAiChatModel createOpenAiChatModel(ConfigExtractor configExtrac
         .modelName(configuration.getModelName())
         .maxTokens(configuration.getMaxTokens())
         .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
         .timeout(ofSeconds(durationInSec))
         .logRequests(true)
         .logResponses(true)
@@ -43,6 +45,7 @@ public static OpenAiChatModel createGroqOpenAiChatModel(ConfigExtractor configEx
         .modelName(configuration.getModelName())
         .maxTokens(configuration.getMaxTokens())
         .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
         .timeout(ofSeconds(durationInSec))
         .logRequests(true)
         .logResponses(true)
@@ -61,6 +64,7 @@ public static MistralAiChatModel createMistralAiChatModel(ConfigExtractor config
         .modelName(configuration.getModelName())
         .maxTokens(configuration.getMaxTokens())
         .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
         .timeout(ofSeconds(durationInSec))
         .logRequests(true)
         .logResponses(true)
@@ -75,6 +79,7 @@ public static OllamaChatModel createOllamaChatModel(ConfigExtractor configExtrac
         .baseUrl(ollamaBaseUrl)
         .modelName(configuration.getModelName())
         .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
         .timeout(ofSeconds(durationInSec))
         .build();
   }
@@ -95,6 +100,21 @@ public static HuggingFaceChatModel createHuggingFaceChatModel(ConfigExtractor co
   }
 
 
+  public static GoogleAiGeminiChatModel createGoogleGeminiChatModel(ConfigExtractor configExtractor,
+                                                                    LangchainLLMConfiguration configuration) {
+    String geminiAiKey = configExtractor.extractValue("GEMINI_AI_KEY");
+    long durationInSec = configuration.getLlmTimeoutUnit().toSeconds(configuration.getLlmTimeout());
+    return GoogleAiGeminiChatModel.builder()
+        .apiKey(geminiAiKey)
+        .modelName(configuration.getModelName())
+        .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
+        .maxOutputTokens(configuration.getMaxTokens())
+        .logRequestsAndResponses(true)
+        .build();
+  }
+
+
   public static AnthropicChatModel createAnthropicChatModel(ConfigExtractor configExtractor,
                                                             LangchainLLMConfiguration configuration) {
     String anthropicApiKey = configExtractor.extractValue("ANTHROPIC_API_KEY");
@@ -105,6 +125,7 @@ public static AnthropicChatModel createAnthropicChatModel(ConfigExtractor config
         .modelName(configuration.getModelName())
         .maxTokens(configuration.getMaxTokens())
         .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
         .timeout(ofSeconds(durationInSec))
         .logRequests(true)
         .logResponses(true)
@@ -124,6 +145,7 @@ public static AzureOpenAiChatModel createAzureOpenAiChatModel(ConfigExtractor co
         .deploymentName(azureOpenaiDeploymentName)
         .maxTokens(configuration.getMaxTokens())
         .temperature(configuration.getTemperature())
+        .topP(configuration.getTopP())
         .timeout(ofSeconds(durationInSec))
         .logRequestsAndResponses(true)
         .build();

src/main/java/org/mule/extension/mulechain/internal/constants/MuleChainConstants.java

@@ -18,6 +18,7 @@ private MuleChainConstants() {}
   public static final String DB_FILE_PATH = "dbFilePath";
   public static final String MAX_MESSAGES = "maxMessages";
   public static final String TOOLS_USED = "toolsUsed";
+  public static final String TOOL_EXECUTION_REQUESTS = "toolExecutionRequests";
   public static final String STATUS = "status";
   public static final String STORE_NAME = "storeName";
   public static final String CREATED = "created";

src/main/java/org/mule/extension/mulechain/internal/helpers/ResponseHelper.java

@@ -44,6 +44,7 @@ public static Result<InputStream, LLMResponseAttributes> createLLMResponse(Strin
   public static Result<InputStream, LLMResponseAttributes> createLLMResponse(String response,
                                                                              TokenUsage tokenUsage,
                                                                              Map<String, String> responseAttributes) {
+
     return Result.<InputStream, LLMResponseAttributes>builder()
         .attributes(new LLMResponseAttributes(tokenUsage, (HashMap<String, String>) responseAttributes))
         .attributesMediaType(org.mule.runtime.api.metadata.MediaType.APPLICATION_JAVA)

src/main/java/org/mule/extension/mulechain/internal/llm/type/LangchainLLMType.java

@@ -26,7 +26,9 @@ public enum LangchainLLMType {
                       "AZURE_OPENAI", OPENAI.getModelNameStream(),
                       LangchainLLMInitializerUtil::createAzureOpenAiChatModel), HUGGING_FACE(
                           "HUGGING_FACE", getHuggingFaceModelNameStream(),
-                          LangchainLLMInitializerUtil::createHuggingFaceChatModel);
+                          LangchainLLMInitializerUtil::createHuggingFaceChatModel), GEMINI_AI("GEMINI_AI",
+                              getGoogleGeminiModelNameStream(),
+                              LangchainLLMInitializerUtil::createGoogleGeminiChatModel);
 
   private final String value;
   private final Stream<String> modelNameStream;
@@ -69,6 +71,11 @@ private static Stream<String> getHuggingFaceModelNameStream() {
     return Arrays.stream(HuggingFaceModelName.values()).map(String::valueOf);
   }
 
+  private static Stream<String> getGoogleGeminiModelNameStream() {
+    return Arrays.stream(GeminiModelName.values()).map(String::valueOf);
+  }
+
+
   private static Stream<String> getAnthropicModelNameStream() {
     return Arrays.stream(AnthropicChatModelName.values()).map(String::valueOf);
   }
@@ -111,4 +118,20 @@ public String toString() {
     }
   }
 
+  enum GeminiModelName {
+    GEMINI_1_5_FLASH("gemini-1.5-flash"), GEMINI_1_5_PRO("gemini-1.5-pro"), GEMINI_1_PRO(
+        "gemini-1.0-pro");
+
+    private final String value;
+
+    GeminiModelName(String value) {
+      this.value = value;
+    }
+
+    @Override
+    public String toString() {
+      return this.value;
+    }
+  }
+
 }