3232import org .springframework .ai .chat .observation .ChatModelObservationDocumentation ;
3333import org .springframework .ai .chat .observation .DefaultChatModelObservationConvention ;
3434import org .springframework .ai .chat .prompt .Prompt ;
35+ import org .springframework .ai .google .genai .GoogleGenAiChatModel .ChatModel ;
3536import org .springframework .ai .observation .conventions .AiOperationType ;
3637import org .springframework .ai .observation .conventions .AiProvider ;
3738import org .springframework .beans .factory .annotation .Autowired ;
@@ -64,7 +65,7 @@ void beforeEach() {
6465 void observationForChatOperation () {
6566
6667 var options = GoogleGenAiChatOptions .builder ()
67- .model (GoogleGenAiChatModel . ChatModel .GEMINI_3_PRO_PREVIEW .getValue ())
68+ .model (ChatModel .GEMINI_2_5_PRO .getValue ())
6869 .temperature (0.7 )
6970 .stopSequences (List .of ("this-is-the-end" ))
7071 .maxOutputTokens (2048 )
@@ -86,7 +87,7 @@ void observationForChatOperation() {
8687 void observationForStreamingOperation () {
8788
8889 var options = GoogleGenAiChatOptions .builder ()
89- .model (GoogleGenAiChatModel . ChatModel .GEMINI_3_PRO_PREVIEW .getValue ())
90+ .model (ChatModel .GEMINI_2_5_PRO .getValue ())
9091 .temperature (0.7 )
9192 .stopSequences (List .of ("this-is-the-end" ))
9293 .maxOutputTokens (2048 )
@@ -126,7 +127,7 @@ private void validate(ChatResponseMetadata responseMetadata) {
126127 AiProvider .GOOGLE_GENAI_AI .value ())
127128 .hasLowCardinalityKeyValue (
128129 ChatModelObservationDocumentation .LowCardinalityKeyNames .REQUEST_MODEL .asString (),
129- GoogleGenAiChatModel . ChatModel .GEMINI_3_PRO_PREVIEW .getValue ())
130+ ChatModel .GEMINI_2_5_PRO .getValue ())
130131 .hasHighCardinalityKeyValue (
131132 ChatModelObservationDocumentation .HighCardinalityKeyNames .REQUEST_MAX_TOKENS .asString (), "2048" )
132133 .hasHighCardinalityKeyValue (
@@ -174,9 +175,7 @@ public GoogleGenAiChatModel vertexAiEmbedding(Client genAiClient, TestObservatio
174175 return GoogleGenAiChatModel .builder ()
175176 .genAiClient (genAiClient )
176177 .observationRegistry (observationRegistry )
177- .defaultOptions (GoogleGenAiChatOptions .builder ()
178- .model (GoogleGenAiChatModel .ChatModel .GEMINI_3_PRO_PREVIEW )
179- .build ())
178+ .defaultOptions (GoogleGenAiChatOptions .builder ().model (ChatModel .GEMINI_2_5_PRO ).build ())
180179 .build ();
181180 }
182181
0 commit comments