@@ -106,14 +106,14 @@ public void onError(Throwable throwable) {
106106 void provide_streaming_chat_model_with_property_values () {
107107 contextRunner .withPropertyValues (
108108 "langchain4j.google-ai-gemini.streaming-chat-model.api-key=" + API_KEY ,
109- "langchain4j.google-ai-gemini.streamingChatModel.modelName =gemini-2.5-flash-lite" ,
110- "langchain4j.google-ai-gemini.streamingChatModel .temperature=0.7" ,
111- "langchain4j.google-ai-gemini.streamingChatModel .topP=0.9" ,
112- "langchain4j.google-ai-gemini.streamingChatModel .topK=40" ,
113- "langchain4j.google-ai-gemini.streamingChatModel.maxOutputTokens =400" ,
114- "langchain4j.google-ai-gemini.streamingChatModel.safetySetting .HARM_CATEGORY_SEXUALLY_EXPLICIT=HARM_BLOCK_THRESHOLD_UNSPECIFIED" ,
115- "langchain4j.google-ai-gemini.streamingChatModel.functionCallingConfig .gemini-mode=NONE" ,
116- "langchain4j.google-ai-gemini.streamingChatModel.functionCallingConfig .allowed-function-names=allowCodeExecution,includeCodeExecutionOutput"
109+ "langchain4j.google-ai-gemini.streaming-chat-model.model-name =gemini-2.5-flash-lite" ,
110+ "langchain4j.google-ai-gemini.streaming-chat-model .temperature=0.7" ,
111+ "langchain4j.google-ai-gemini.streaming-chat-model .topP=0.9" ,
112+ "langchain4j.google-ai-gemini.streaming-chat-model .topK=40" ,
113+ "langchain4j.google-ai-gemini.streaming-chat-model.max-output-tokens =400" ,
114+ "langchain4j.google-ai-gemini.streaming-chat-model.safety-setting .HARM_CATEGORY_SEXUALLY_EXPLICIT=HARM_BLOCK_THRESHOLD_UNSPECIFIED" ,
115+ "langchain4j.google-ai-gemini.streaming-chat-model.function-calling-config .gemini-mode=NONE" ,
116+ "langchain4j.google-ai-gemini.streaming-chat-model.function-calling-config .allowed-function-names=allowCodeExecution,includeCodeExecutionOutput"
117117 )
118118 .run (context -> {
119119 StreamingChatModel streamingChatModel = context .getBean (StreamingChatModel .class );
@@ -144,27 +144,27 @@ public void onError(Throwable throwable) {
144144 void provide_embedding_model () {
145145 contextRunner .withPropertyValues (
146146 "langchain4j.google-ai-gemini.embedding-model.apiKey=" + API_KEY ,
147- "langchain4j.google-ai-gemini.embedding-model.model-name=text -embedding-004 "
147+ "langchain4j.google-ai-gemini.embedding-model.model-name=gemini -embedding-001 "
148148 ).run (context -> {
149149 EmbeddingModel embeddingModel = context .getBean (EmbeddingModel .class );
150150 assertThat (context .getBean (GoogleAiEmbeddingModel .class )).isSameAs (embeddingModel );
151151
152152 Response <Embedding > response = embeddingModel .embed ("Hi, I live in India" );
153- assertThat (response .content ().dimension ()).isEqualTo (768 );
153+ assertThat (response .content ().dimension ()).isEqualTo (3072 );
154154 });
155155 }
156156
157157 @ Test
158158 void provide_embedding_model_with_property_values () {
159159 contextRunner .withPropertyValues (
160160 "langchain4j.google-ai-gemini.embedding-model.apiKey=" + API_KEY ,
161- "langchain4j.google-ai-gemini.embeddingModel.titleMetadataKey =title-key" ,
162- "langchain4j.google-ai-gemini.embeddingModel.modelName=text -embedding-004 " ,
163- "langchain4j.google-ai-gemini.embeddingModel.logRequestsAndResponses =true" ,
164- "langchain4j.google-ai-gemini.embeddingModel.maxRetries =3" ,
165- "langchain4j.google-ai-gemini.embeddingModel.outputDimensionality =512" ,
166- "langchain4j.google-ai-gemini.embeddingModel.taskType =CLASSIFICATION" ,
167- "langchain4j.google-ai-gemini.embeddingModel .timeout=PT30S"
161+ "langchain4j.google-ai-gemini.embedding-model.title-metadata.key =title-key" ,
162+ "langchain4j.google-ai-gemini.embedding-model.model-name=gemini -embedding-001 " ,
163+ "langchain4j.google-ai-gemini.embedding-model.log-requests-and-responses =true" ,
164+ "langchain4j.google-ai-gemini.embedding-model.max-retries =3" ,
165+ "langchain4j.google-ai-gemini.embedding-model.output-dimensionality =512" ,
166+ "langchain4j.google-ai-gemini.embedding-model.task-type =CLASSIFICATION" ,
167+ "langchain4j.google-ai-gemini.embedding-model .timeout=PT30S"
168168 ).run (context -> {
169169 EmbeddingModel embeddingModel = context .getBean (GoogleAiEmbeddingModel .class );
170170 assertThat (embeddingModel ).isInstanceOf (EmbeddingModel .class );
0 commit comments