@@ -166,6 +166,33 @@ public void processLlmRequestWithGoogleSearchToolAddsToolToConfig() {
166166 Tool .builder ().googleSearch (GoogleSearch .builder ().build ()).build ());
167167 }
168168
169+ @ Test
170+ public void processLlmRequestWithLatestAliasAddsToolToConfig () {
171+ final GoogleSearchTool googleSearchTool = new GoogleSearchTool ();
172+ LlmRequest .Builder builder =
173+ LlmRequest .builder ().model ("gemini-flash-latest" ).build ().toBuilder ();
174+ Completable result = googleSearchTool .processLlmRequest (builder , null );
175+ result .test ().assertComplete ();
176+ assertThat (builder .build ().config ().get ().tools ().get ())
177+ .contains (Tool .builder ().googleSearch (GoogleSearch .builder ().build ()).build ());
178+ }
179+
180+ @ Test
181+ public void processLlmRequestWithUnsupportedModelReturnsError () {
182+ final GoogleSearchTool googleSearchTool = new GoogleSearchTool ();
183+ LlmRequest .Builder builder = LlmRequest .builder ().model ("text-bison-001" ).build ().toBuilder ();
184+ Completable result = googleSearchTool .processLlmRequest (builder , null );
185+ result .test ().assertError (IllegalArgumentException .class );
186+ }
187+
188+ @ Test
189+ public void processLlmRequest_WithNullModel_ReturnsError () {
190+ final GoogleSearchTool googleSearchTool = new GoogleSearchTool ();
191+ LlmRequest .Builder builder = LlmRequest .builder ().build ().toBuilder ();
192+ Completable result = googleSearchTool .processLlmRequest (builder , null );
193+ result .test ().assertError (IllegalArgumentException .class );
194+ }
195+
169196 @ Test
170197 public void processLlmRequestWithUrlContextToolAddsToolToConfig () {
171198 FunctionDeclaration functionDeclaration =
0 commit comments