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();