@@ -145,7 +145,6 @@ protected static void cleanupClass() {
145145 }
146146
147147 @ Test
148- @ Disabled ("Requires further resource deployment" )
149148 public void createKnowledgeBaseSync () {
150149 // Test creating a knowledge knowledgebase.
151150 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
@@ -172,7 +171,6 @@ public void createKnowledgeBaseSync() {
172171 }
173172
174173 @ Test
175- @ Disabled ("Requires further resource deployment" )
176174 public void createKnowledgeBaseAsync () {
177175 // Test creating a knowledge knowledgebase.
178176 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -200,7 +198,6 @@ public void createKnowledgeBaseAsync() {
200198 }
201199
202200 @ Test
203- @ Disabled ("Requires further resource deployment" )
204201 public void getKnowledgeBaseSync () {
205202 // Test getting a knowledge knowledgebase.
206203 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
@@ -228,7 +225,6 @@ public void getKnowledgeBaseSync() {
228225 }
229226
230227 @ Test
231- @ Disabled ("Requires further resource deployment" )
232228 public void getKnowledgeBaseAsync () {
233229 // Test getting a knowledge knowledgebase.
234230 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -259,7 +255,6 @@ public void getKnowledgeBaseAsync() {
259255 }
260256
261257 @ Test
262- @ Disabled ("Requires further resource deployment" )
263258 public void listKnowledgeBasesSync () {
264259 // Test listing knowledge knowledgebases.
265260 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
@@ -282,7 +277,6 @@ public void listKnowledgeBasesSync() {
282277 }
283278
284279 @ Test
285- @ Disabled ("Requires further resource deployment" )
286280 public void listKnowledgeBasesAsync () {
287281 // Test listing knowledge knowledgebases.
288282 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -310,7 +304,6 @@ public void listKnowledgeBasesAsync() {
310304 }
311305
312306 @ Test
313- @ Disabled ("Requires further resource deployment" )
314307 public void deleteKnowledgeBaseSync () {
315308 // Test deleting a knowledge knowledgebase.
316309 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
@@ -324,7 +317,6 @@ public void deleteKnowledgeBaseSync() {
324317 }
325318
326319 @ Test
327- @ Disabled ("Requires further resource deployment" )
328320 public void deleteKnowledgeBaseAsync () {
329321 // Test deleting a knowledge base.
330322 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -345,7 +337,6 @@ public void deleteKnowledgeBaseAsync() {
345337 }
346338
347339 @ Test
348- @ Disabled ("Requires further resource deployment" )
349340 public void updateKnowledgeBaseSync () {
350341 // Test updating a knowledge base.
351342 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
@@ -354,13 +345,12 @@ public void updateKnowledgeBaseSync() {
354345 searchIndexClient .createKnowledgeBase (knowledgeBase );
355346 String newDescription = "Updated description" ;
356347 knowledgeBase .setDescription (newDescription );
357- searchIndexClient .createKnowledgeBase (knowledgeBase );
348+ searchIndexClient .createOrUpdateKnowledgeBase (knowledgeBase );
358349 KnowledgeBase retrieved = searchIndexClient .getKnowledgeBase (knowledgeBase .getName ());
359350 assertEquals (newDescription , retrieved .getDescription ());
360351 }
361352
362353 @ Test
363- @ Disabled ("Requires further resource deployment" )
364354 public void updateKnowledgeBaseAsync () {
365355 // Test updating a knowledge base.
366356 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -369,7 +359,11 @@ public void updateKnowledgeBaseAsync() {
369359 String newDescription = "Updated description" ;
370360
371361 Mono <KnowledgeBase > createUpdateAndGetMono = searchIndexClient .createKnowledgeBase (knowledgeBase )
372- .flatMap (created -> searchIndexClient .createKnowledgeBase (created .setDescription (newDescription )))
362+ .flatMap (created -> searchIndexClient .deleteKnowledgeBase (created .getName ())
363+ .then (searchIndexClient
364+ .createKnowledgeBase (new KnowledgeBase (knowledgeBase .getName (), KNOWLEDGE_SOURCE_REFERENCE )
365+ .setModels (KNOWLEDGE_BASE_MODEL )
366+ .setDescription (newDescription ))))
373367 .flatMap (updated -> searchIndexClient .getKnowledgeBase (updated .getName ()));
374368
375369 StepVerifier .create (createUpdateAndGetMono )
@@ -378,26 +372,25 @@ public void updateKnowledgeBaseAsync() {
378372 }
379373
380374 @ Test
381- @ Disabled ("Requires further resource deployment" )
382375 public void basicRetrievalSync () {
383376 // Test knowledge base retrieval functionality.
384377 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
385378 KnowledgeBase knowledgeBase
386379 = new KnowledgeBase (randomKnowledgeBaseName (), KNOWLEDGE_SOURCE_REFERENCE ).setModels (KNOWLEDGE_BASE_MODEL );
387380 searchIndexClient .createKnowledgeBase (knowledgeBase );
388381
389- KnowledgeBaseRetrievalClient knowledgeBaseClient = getKnowledgeBaseRetrievalClientBuilder (true ).buildClient ();
382+ KnowledgeBaseRetrievalClient knowledgeBaseClient
383+ = getKnowledgeBaseRetrievalClientBuilder (true ).knowledgeBaseName (knowledgeBase .getName ()).buildClient ();
390384
391385 KnowledgeBaseRetrievalOptions retrievalRequest = new KnowledgeBaseRetrievalOptions ()
392386 .setIntents (new KnowledgeRetrievalSemanticIntent ("What are the pet policies at the hotel?" ));
393387
394- KnowledgeBaseRetrievalResult response = knowledgeBaseClient .retrieve (knowledgeBase . getName (), retrievalRequest );
388+ KnowledgeBaseRetrievalResult response = knowledgeBaseClient .retrieve (retrievalRequest );
395389 assertNotNull (response );
396390 assertNotNull (response .getResponse ());
397391 }
398392
399393 @ Test
400- @ Disabled ("Requires further resource deployment" )
401394 public void basicRetrievalAsync () {
402395 // Test knowledge base retrieval functionality.
403396 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -407,12 +400,13 @@ public void basicRetrievalAsync() {
407400 Mono <KnowledgeBaseRetrievalResult > createAndRetrieveMono
408401 = searchIndexClient .createKnowledgeBase (knowledgeBase ).flatMap (created -> {
409402 KnowledgeBaseRetrievalAsyncClient knowledgeBaseClient
410- = getKnowledgeBaseRetrievalClientBuilder (false ).buildAsyncClient ();
403+ = getKnowledgeBaseRetrievalClientBuilder (false ).knowledgeBaseName (created .getName ())
404+ .buildAsyncClient ();
411405
412406 KnowledgeBaseRetrievalOptions retrievalRequest = new KnowledgeBaseRetrievalOptions ()
413407 .setIntents (new KnowledgeRetrievalSemanticIntent ("What are the pet policies at the hotel?" ));
414408
415- return knowledgeBaseClient .retrieve (created . getName (), retrievalRequest );
409+ return knowledgeBaseClient .retrieve (retrievalRequest );
416410 });
417411
418412 StepVerifier .create (createAndRetrieveMono ).assertNext (response -> {
@@ -422,27 +416,26 @@ public void basicRetrievalAsync() {
422416 }
423417
424418 @ Test
425- @ Disabled ("Requires further resource deployment" )
426419 public void basicRetrievalWithReasoningEffortSync () {
427420 // Test knowledge base retrieval functionality.
428421 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
429422 KnowledgeBase knowledgeBase
430423 = new KnowledgeBase (randomKnowledgeBaseName (), KNOWLEDGE_SOURCE_REFERENCE ).setModels (KNOWLEDGE_BASE_MODEL );
431424 searchIndexClient .createKnowledgeBase (knowledgeBase );
432425
433- KnowledgeBaseRetrievalClient knowledgeBaseClient = getKnowledgeBaseRetrievalClientBuilder (true ).buildClient ();
426+ KnowledgeBaseRetrievalClient knowledgeBaseClient
427+ = getKnowledgeBaseRetrievalClientBuilder (true ).knowledgeBaseName (knowledgeBase .getName ()).buildClient ();
434428
435429 KnowledgeBaseRetrievalOptions retrievalRequest = new KnowledgeBaseRetrievalOptions ()
436430 .setIntents (new KnowledgeRetrievalSemanticIntent ("What are the pet policies at the hotel?" ));
437431 // .setRetrievalReasoningEffort(KnowledgeRetrievalReasoningEffortKind.MEDIUM); // TODO: Missing enum
438432
439- KnowledgeBaseRetrievalResult response = knowledgeBaseClient .retrieve (knowledgeBase . getName (), retrievalRequest );
433+ KnowledgeBaseRetrievalResult response = knowledgeBaseClient .retrieve (retrievalRequest );
440434 assertNotNull (response );
441435 assertNotNull (response .getResponse ());
442436 }
443437
444438 @ Test
445- @ Disabled ("Requires further resource deployment" )
446439 public void basicRetrievalWithReasoningEffortAsync () {
447440 // Test knowledge base retrieval functionality.
448441 SearchIndexAsyncClient searchIndexClient = getSearchIndexClientBuilder (false ).buildAsyncClient ();
@@ -452,13 +445,14 @@ public void basicRetrievalWithReasoningEffortAsync() {
452445 Mono <KnowledgeBaseRetrievalResult > createAndRetrieveMono
453446 = searchIndexClient .createKnowledgeBase (knowledgeBase ).flatMap (created -> {
454447 KnowledgeBaseRetrievalAsyncClient knowledgeBaseClient
455- = getKnowledgeBaseRetrievalClientBuilder (false ).buildAsyncClient ();
448+ = getKnowledgeBaseRetrievalClientBuilder (false ).knowledgeBaseName (created .getName ())
449+ .buildAsyncClient ();
456450
457451 KnowledgeBaseRetrievalOptions retrievalRequest = new KnowledgeBaseRetrievalOptions ()
458452 .setIntents (new KnowledgeRetrievalSemanticIntent ("What are the pet policies at the hotel?" ));
459453 // .setRetrievalReasoningEffort(KnowledgeRetrievalReasoningEffortKind.MEDIUM); // TODO: Missing enum
460454
461- return knowledgeBaseClient .retrieve (created . getName (), retrievalRequest );
455+ return knowledgeBaseClient .retrieve (retrievalRequest );
462456 });
463457
464458 StepVerifier .create (createAndRetrieveMono ).assertNext (response -> {
@@ -480,7 +474,6 @@ public void answerSynthesisRetrievalAsync() {
480474 }
481475
482476 @ Test
483- @ Disabled ("Requires further resource deployment" )
484477 public void knowledgeBaseObjectHasNoAgentReferences () throws IOException {
485478 SearchIndexClient searchIndexClient = getSearchIndexClientBuilder (true ).buildClient ();
486479 KnowledgeBase knowledgeBase
@@ -498,7 +491,6 @@ public void knowledgeBaseObjectHasNoAgentReferences() throws IOException {
498491 }
499492
500493 @ Test
501- @ Disabled ("Requires further resource deployment" )
502494 public void knowledgeBaseEndpointsUseKnowledgeBasesPath () {
503495 SearchIndexClient client = getSearchIndexClientBuilder (true )
504496 .httpLogOptions (new HttpLogOptions ().setLogLevel (HttpLogDetailLevel .BODY_AND_HEADERS ))
@@ -540,7 +532,6 @@ public void legacyKnowledgeAgentsListedAsKnowledgeBases() {
540532 }
541533
542534 @ Test
543- @ Disabled ("Requires further resource deployment" )
544535 public void knowledgeSourcesEndpointUnchanged () {
545536 SearchIndexClient client = getSearchIndexClientBuilder (true ).buildClient ();
546537
@@ -559,7 +550,6 @@ public void knowledgeSourcesEndpointUnchanged() {
559550 }
560551
561552 @ Test
562- @ Disabled ("Requires further resource deployment" )
563553 public void knowledgeBaseTypeNamesContainNoAgentReferences () {
564554 SearchIndexClient client = getSearchIndexClientBuilder (true ).buildClient ();
565555
0 commit comments