@@ -3667,9 +3667,10 @@ public void testResolveRerankFields() {
3667
3667
{
3668
3668
VerificationException ve = expectThrows (
3669
3669
VerificationException .class ,
3670
- () -> analyze (
3671
- "FROM books METADATA _score | RERANK \" italian food recipe\" ON missingField OPTIONS inferenceId=`reranking-inference-id`" ,
3672
- "mapping-books.json"
3670
+ () -> analyze ("""
3671
+ FROM books METADATA _score
3672
+ | RERANK \" italian food recipe\" ON missingField OPTIONS inferenceId=`reranking-inference-id`
3673
+ """ , "mapping-books.json"
3673
3674
)
3674
3675
3675
3676
);
@@ -3790,12 +3791,11 @@ public void testResolveCompletionInferenceIdResolutionError() {
3790
3791
}
3791
3792
3792
3793
public void testResolveCompletionTargetField () {
3793
- LogicalPlan plan = analyze (
3794
- """
3795
- FROM books METADATA _score
3796
- | COMPLETION CONCAT("Translate the following text in French\\ n", description) INTO translation OPTIONS inferenceId=`completion-inference-id`
3797
- """ ,
3798
- "mapping-books.json"
3794
+ LogicalPlan plan = analyze ("""
3795
+ FROM books METADATA _score
3796
+ | COMPLETION CONCAT("Translate the following text in French\\ n", description) INTO translation
3797
+ OPTIONS inferenceId=`completion-inference-id`
3798
+ """ ,"mapping-books.json"
3799
3799
);
3800
3800
3801
3801
Completion completion = as (as (plan , Limit .class ).child (), Completion .class );
@@ -3835,12 +3835,11 @@ public void testResolveCompletionPromptInvalidType() {
3835
3835
}
3836
3836
3837
3837
public void testResolveCompletionOutputField () {
3838
- LogicalPlan plan = analyze (
3839
- """
3840
- FROM books METADATA _score
3841
- | COMPLETION CONCAT("Translate the following text in French\\ n", description) INTO description OPTIONS inferenceId=`completion-inference-id`
3842
- """ ,
3843
- "mapping-books.json"
3838
+ LogicalPlan plan = analyze ("""
3839
+ FROM books METADATA _score
3840
+ | COMPLETION CONCAT("Translate the following text in French\\ n", description) INTO description
3841
+ OPTIONS inferenceId=`completion-inference-id`
3842
+ """ , "mapping-books.json"
3844
3843
);
3845
3844
3846
3845
Completion completion = as (as (plan , Limit .class ).child (), Completion .class );
0 commit comments