3838import lombok .extern .slf4j .Slf4j ;
3939import lombok .val ;
4040import org .junit .jupiter .api .BeforeEach ;
41- import org .junit .jupiter .api .Disabled ;
4241import org .junit .jupiter .api .Test ;
4342import org .junit .jupiter .api .condition .DisabledIfSystemProperty ;
4443
@@ -399,15 +398,13 @@ void testResponseFormatText() {
399398 }
400399
401400 @ Test
402- @ Disabled ("This behaviour is not released to canary yet." )
403401 void testTemplateFromPromptRegistryById () {
404402 val result = service .templateFromPromptRegistryById ("Cloud ERP systems" ).getOriginalResponse ();
405403 val choices = (result .getFinalResult ()).getChoices ();
406404 assertThat (choices .get (0 ).getMessage ().getContent ()).isNotEmpty ();
407405 }
408406
409407 @ Test
410- @ Disabled ("This behaviour is not released to canary yet." )
411408 void testTemplateFromPromptRegistryByScenario () {
412409 val result =
413410 service .templateFromPromptRegistryByScenario ("Cloud ERP systems" ).getOriginalResponse ();
@@ -466,7 +463,6 @@ void testStreamingErrorHandlingMasking() {
466463 }
467464
468465 @ Test
469- @ Disabled ("This behaviour is not released to canary yet." )
470466 void testTranslation () {
471467 val result = service .translation ();
472468 val content = result .getContent ();
@@ -480,7 +476,8 @@ void testTranslation() {
480476 result .getOriginalResponse ().getIntermediateResults ().getOutputTranslation ();
481477 assertThat (inputTranslation ).isNotNull ();
482478 assertThat (outputTranslation ).isNotNull ();
483- assertThat (inputTranslation .getMessage ()).isEqualTo ("Input to LLM is translated successfully." );
479+ assertThat (inputTranslation .getMessage ())
480+ .isEqualTo ("Translated messages with roles: ['user']. " );
484481 assertThat (outputTranslation .getMessage ()).isEqualTo ("Output Translation successful" );
485482 }
486483
0 commit comments