Google Gemini Chat Stream Support

jeddict · Nov 17, 2024 · cbf614d · cbf614d
1 parent c595b54
commit cbf614d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/pom.xml b/pom.xml
@@ -176,37 +176,37 @@
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-open-ai</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-ollama</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-local-ai</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-mistral-ai</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-anthropic</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-google-ai-gemini</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j</artifactId>
-            <version>0.34.0</version>
+            <version>0.36.0</version>
         </dependency>
         <dependency>
             <groupId>org.commonmark</groupId>

diff --git a/src/main/java/io/github/jeddict/ai/lang/JeddictChatModel.java b/src/main/java/io/github/jeddict/ai/lang/JeddictChatModel.java
@@ -30,6 +30,7 @@
 import dev.langchain4j.model.chat.ChatLanguageModel;
 import dev.langchain4j.model.chat.StreamingChatLanguageModel;
 import dev.langchain4j.model.googleai.GoogleAiGeminiChatModel;
+import dev.langchain4j.model.googleai.GoogleAiGeminiStreamingChatModel;
 import dev.langchain4j.model.localai.LocalAiChatModel;
 import dev.langchain4j.model.localai.LocalAiStreamingChatModel;
 import dev.langchain4j.model.mistralai.MistralAiChatModel;
@@ -86,7 +87,7 @@ public JeddictChatModel(StreamingResponseHandler handler) {
             if (preferencesManager.isStreamEnabled() && handler != null) {
                 switch (preferencesManager.getProvider()) {
                     case GOOGLE ->
-                        model = GoogleAiGeminiChatModel.builder()
+                        streamModel = GoogleAiGeminiStreamingChatModel.builder()
                                 .apiKey(preferencesManager.getApiKey())
                                 .modelName(preferencesManager.getModelName())
                                 .build();