Skip to content

Commit 69ba3ef

Browse files
committed
fixing flaky tests
1 parent 97db5e6 commit 69ba3ef

File tree

1 file changed

+11
-11
lines changed
  • langchain4j-google-ai-gemini-spring-boot-starter/src/test/java/dev/langchain4j/googleaigemini/spring

1 file changed

+11
-11
lines changed

langchain4j-google-ai-gemini-spring-boot-starter/src/test/java/dev/langchain4j/googleaigemini/spring/AutoConfigIT.java

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import dev.langchain4j.data.embedding.Embedding;
44
import dev.langchain4j.model.chat.ChatModel;
55
import dev.langchain4j.model.chat.StreamingChatModel;
6+
import dev.langchain4j.model.chat.request.ChatRequestParameters;
67
import dev.langchain4j.model.chat.response.ChatResponse;
78
import dev.langchain4j.model.chat.response.StreamingChatResponseHandler;
89
import dev.langchain4j.model.embedding.EmbeddingModel;
@@ -36,6 +37,7 @@ void provide_chat_model() {
3637
)
3738
.run(context -> {
3839
ChatModel chatModel = context.getBean(ChatModel.class);
40+
assertThat(chatModel).isInstanceOf(GoogleAiGeminiChatModel.class);
3941
assertThat(context.getBean(GoogleAiGeminiChatModel.class)).isSameAs(chatModel);
4042

4143
String response = chatModel.chat("What is the capital of India");
@@ -50,22 +52,20 @@ void provide_chat_model() {
5052
void provide_chat_model_with_property_values() {
5153
contextRunner.withPropertyValues(
5254
"langchain4j.google-ai-gemini.chat-model.api-key=" + API_KEY,
53-
"langchain4j.google-ai-gemini.chatModel.modelName=gemini-2.5-flash-lite",
54-
"langchain4j.google-ai-gemini.chatModel.temperature=0.7",
55-
"langchain4j.google-ai-gemini.chatModel.topP=0.9",
56-
"langchain4j.google-ai-gemini.chatModel.topK=40",
57-
"langchain4j.google-ai-gemini.chatModel.maxOutputTokens=800",
58-
"langchain4j.google-ai-gemini.chatModel.safetySetting.HARM_CATEGORY_DANGEROUS_CONTENT=BLOCK_LOW_AND_ABOVE",
59-
"langchain4j.google-ai-gemini.chatModel.functionCallingConfig.gemini-mode=NONE",
60-
"langchain4j.google-ai-gemini.chatModel.functionCallingConfig.allowed-function-names=allowCodeExecution,includeCodeExecutionOutput"
55+
"langchain4j.google-ai-gemini.chat-model.model-name=gemini-2.5-flash-lite",
56+
"langchain4j.google-ai-gemini.chat-model.temperature=0.7"
6157
)
6258
.run(context -> {
6359
ChatModel chatModel = context.getBean(ChatModel.class);
64-
assertThat(chatModel).isInstanceOf(ChatModel.class);
60+
assertThat(chatModel).isInstanceOf(GoogleAiGeminiChatModel.class);
61+
assertThat(context.getBean(GoogleAiGeminiChatModel.class)).isSameAs(chatModel);
62+
63+
ChatRequestParameters defaultParameters = chatModel.defaultRequestParameters();
64+
assertThat(defaultParameters.modelName()).isEqualTo("gemini-2.5-flash-lite");
65+
assertThat(defaultParameters.temperature()).isEqualTo(0.7);
66+
6567
String response = chatModel.chat("What is the capital of India");
6668
assertThat(response).contains("Delhi");
67-
String newResponse = chatModel.chat("Calculate the Fibonacci of 22 and give me the result as an integer value along with the code. ");
68-
assertThat(newResponse).contains("17711");
6969
});
7070
}
7171

0 commit comments

Comments
 (0)