diff --git a/pom.xml b/pom.xml
index c01065a..f048854 100644
--- a/pom.xml
+++ b/pom.xml
@@ -176,37 +176,37 @@
dev.langchain4j
langchain4j-open-ai
- 0.34.0
+ 0.36.0
dev.langchain4j
langchain4j-ollama
- 0.34.0
+ 0.36.0
dev.langchain4j
langchain4j-local-ai
- 0.34.0
+ 0.36.0
dev.langchain4j
langchain4j-mistral-ai
- 0.34.0
+ 0.36.0
dev.langchain4j
langchain4j-anthropic
- 0.34.0
+ 0.36.0
dev.langchain4j
langchain4j-google-ai-gemini
- 0.34.0
+ 0.36.0
dev.langchain4j
langchain4j
- 0.34.0
+ 0.36.0
org.commonmark
diff --git a/src/main/java/io/github/jeddict/ai/lang/JeddictChatModel.java b/src/main/java/io/github/jeddict/ai/lang/JeddictChatModel.java
index cc380c9..2de3e16 100644
--- 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();