33import dev .langchain4j .data .embedding .Embedding ;
44import dev .langchain4j .model .chat .ChatModel ;
55import dev .langchain4j .model .chat .StreamingChatModel ;
6+ import dev .langchain4j .model .chat .request .ChatRequestParameters ;
67import dev .langchain4j .model .chat .response .ChatResponse ;
78import dev .langchain4j .model .chat .response .StreamingChatResponseHandler ;
89import 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