Fix to #35983 - Cosmos/FTS: update translation of FullTextScore to use multiple keywords rather than keyword array

Reacting to Cosmos changing the signature of FullTextScore function.

Fixes #35983

Author: maumar

src/EFCore.Cosmos/Query/Internal/CosmosShapedQueryCompilingExpressionVisitor.ParameterInliner.cs

@@ -80,12 +80,17 @@ protected override Expression VisitExtension(Expression expression)
                 }
 
                 // Inlines array parameter of full-text functions, transforming FullTextContainsAll(x, @keywordsArray) to FullTextContainsAll(x, keyword1, keyword2)) 
+                // we do this for FullTextContainsAll, FullTextContainsAny and FullTextScore
                 case SqlFunctionExpression
                 {
-                    Name: "FullTextContainsAny" or "FullTextContainsAll",
+                    Name: string name,
+                    IsScoringFunction: bool scoringFunction,
                     Arguments: [var property, SqlParameterExpression { TypeMapping: { ElementTypeMapping: var elementTypeMapping }, Type: Type type } keywords]
                 } fullTextContainsAllAnyFunction
-                when type == typeof(string[]):
+                when ((name == "FullTextContainsAny" && !scoringFunction)
+                    || (name == "FullTextContainsAll" && !scoringFunction)
+                    || (name == "FullTextScore" && scoringFunction))
+                    && type == typeof(string[]):
                 {
                     var keywordValues = new List<SqlExpression>();
                     foreach (var value in (IEnumerable)parametersValues[keywords.Name])
@@ -100,28 +105,6 @@ protected override Expression VisitExtension(Expression expression)
                         fullTextContainsAllAnyFunction.TypeMapping);
                 }
 
-                // Inlines array parameter of full-text score, transforming FullTextScore(x, @keywordsArray) to FullTextScore(x, [keyword1, keyword2])) 
-                case SqlFunctionExpression
-                {
-                    Name: "FullTextScore",
-                    IsScoringFunction: true,
-                    Arguments: [var property, SqlParameterExpression { TypeMapping: { ElementTypeMapping: not null } typeMapping } keywords]
-                } fullTextScoreFunction:
-                {
-                    var keywordValues = new List<string>();
-                    foreach (var value in (IEnumerable)parametersValues[keywords.Name])
-                    {
-                        keywordValues.Add((string)value);
-                    }
-
-                    return new SqlFunctionExpression(
-                        fullTextScoreFunction.Name,
-                        scoringFunction: true,
-                        [property, sqlExpressionFactory.Constant(keywordValues, typeMapping)],
-                        fullTextScoreFunction.Type,
-                        fullTextScoreFunction.TypeMapping);
-                }
-
                 default:
                     return expression;
             }

src/EFCore.Cosmos/Query/Internal/Translators/CosmosFullTextSearchTranslator.cs

@@ -41,12 +41,25 @@ public class CosmosFullTextSearchTranslator(ISqlExpressionFactory sqlExpressionF
                     typeMappingSource.FindMapping(typeof(bool))),
 
             nameof(CosmosDbFunctionsExtensions.FullTextScore)
-                when arguments is [_, var property, var keywords] => sqlExpressionFactory.ScoringFunction(
+                when arguments is [_, SqlExpression property, SqlConstantExpression { Type: var keywordClrType, Value: string[] values } keywords]
+                    && keywordClrType == typeof(string[]) => sqlExpressionFactory.ScoringFunction(
+                        "FullTextScore",
+                        [property, .. values.Select(x => sqlExpressionFactory.Constant(x))],
+                        typeof(double),
+                        typeMappingSource.FindMapping(typeof(double))),
+
+            nameof(CosmosDbFunctionsExtensions.FullTextScore)
+                when arguments is [_, SqlExpression property, SqlParameterExpression { Type: var keywordClrType } keywords]
+                    && keywordClrType == typeof(string[]) => sqlExpressionFactory.ScoringFunction(
+                        "FullTextScore",
+                        [property, keywords],
+                        typeof(double),
+                        typeMappingSource.FindMapping(typeof(double))),
+
+            nameof(CosmosDbFunctionsExtensions.FullTextScore)
+                when arguments is [_, SqlExpression property, ArrayConstantExpression keywords] => sqlExpressionFactory.ScoringFunction(
                     "FullTextScore",
-                    [
-                        property,
-                        keywords,
-                    ],
+                    [property, .. keywords.Items],
                     typeof(double),
                     typeMappingSource.FindMapping(typeof(double))),
 

test/EFCore.Cosmos.FunctionalTests/FullTextSearchCosmosTest.cs

@@ -292,7 +292,7 @@ public virtual async Task OrderByRank_FullTextScore_constant()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["otter"])
+ORDER BY RANK FullTextScore(c["Description"], "otter")
 """);
     }
 
@@ -309,7 +309,7 @@ public virtual async Task OrderByRank_FullTextScore_constants()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["otter","beaver"])
+ORDER BY RANK FullTextScore(c["Description"], "otter", "beaver")
 """);
     }
 
@@ -326,7 +326,7 @@ public virtual async Task OrderByRank_FullTextScore_constant_array()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["otter","beaver"])
+ORDER BY RANK FullTextScore(c["Description"], "otter", "beaver")
 """);
     }
 
@@ -343,7 +343,7 @@ public virtual async Task OrderByRank_FullTextScore_constant_array_with_one_elem
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["otter"])
+ORDER BY RANK FullTextScore(c["Description"], "otter")
 """);
     }
 
@@ -361,7 +361,7 @@ public virtual async Task OrderByRank_FullTextScore_parameter_array()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["otter","beaver"])
+ORDER BY RANK FullTextScore(c["Description"], "otter", "beaver")
 """);
     }
 
@@ -384,7 +384,7 @@ public virtual async Task OrderByRank_FullTextScore_using_parameters()
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], [@otter, @beaver])
+ORDER BY RANK FullTextScore(c["Description"], @otter, @beaver)
 """);
     }
 
@@ -405,7 +405,7 @@ public virtual async Task OrderByRank_FullTextScore_using_one_parameter()
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], [@otter])
+ORDER BY RANK FullTextScore(c["Description"], @otter)
 """);
     }
 
@@ -426,7 +426,7 @@ public virtual async Task OrderByRank_FullTextScore_using_parameters_constant_mi
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["otter", @beaver])
+ORDER BY RANK FullTextScore(c["Description"], "otter", @beaver)
 """);
     }
 
@@ -447,7 +447,7 @@ public virtual async Task OrderByRank_FullTextScore_using_parameter()
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], [@otter])
+ORDER BY RANK FullTextScore(c["Description"], @otter)
 """);
     }
 
@@ -495,7 +495,7 @@ public virtual async Task OrderByRank_FullTextScore_on_non_FTS_property()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["PartitionKey"], ["taxonomy"])
+ORDER BY RANK FullTextScore(c["PartitionKey"], "taxonomy")
 """);
     }
 
@@ -514,7 +514,7 @@ public virtual async Task OrderByRank_with_RRF_using_two_FullTextScore_functions
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK RRF(FullTextScore(c["Description"], ["beaver"]), FullTextScore(c["Description"], ["otter","bat"]))
+ORDER BY RANK RRF(FullTextScore(c["Description"], "beaver"), FullTextScore(c["Description"], "otter", "bat"))
 """);
     }
 
@@ -584,7 +584,7 @@ public virtual async Task OrderByRank_Take()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["beaver"])
+ORDER BY RANK FullTextScore(c["Description"], "beaver")
 OFFSET 0 LIMIT 10
 """);
     }
@@ -603,7 +603,7 @@ public virtual async Task OrderByRank_Skip_Take()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["Description"], "beaver", "dolphin")
 OFFSET 1 LIMIT 20
 """);
     }
@@ -633,7 +633,7 @@ public virtual async Task OrderBy_scoring_function_overridden_by_another()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["beaver","dolphin","second"])
+ORDER BY RANK FullTextScore(c["Description"], "beaver", "dolphin", "second")
 """);
     }
 
@@ -667,7 +667,7 @@ public virtual async Task Regular_OrderBy_overridden_by_OrderBy_using_scoring_fu
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Description"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["Description"], "beaver", "dolphin")
 """);
     }
 
@@ -725,7 +725,7 @@ public virtual async Task OrderByRank_Where()
 SELECT VALUE c
 FROM root c
 WHERE ((c["PartitionKey"] || "Foo") = "habitatFoo")
-ORDER BY RANK FullTextScore(c["Description"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["Description"], "beaver", "dolphin")
 """);
     }
 
@@ -779,7 +779,7 @@ public virtual async Task OrderByRank_with_FullTextScore_on_nested_owned_type()
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Owned"]["NestedReference"]["AnotherDescription"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["Owned"]["NestedReference"]["AnotherDescription"], "beaver", "dolphin")
 """);
     }
 
@@ -816,7 +816,7 @@ public virtual async Task OrderByRank_with_FullTextScore_on_nested_owned_collect
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Owned"]["NestedCollection"][0]["AnotherDescription"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["Owned"]["NestedCollection"][0]["AnotherDescription"], "beaver", "dolphin")
 """);
     }
 
@@ -832,7 +832,7 @@ public virtual async Task OrderBy_scoring_function_on_property_with_modified_jso
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["CustomDecription"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["CustomDecription"], "beaver", "dolphin")
 """);
     }
 
@@ -849,7 +849,7 @@ public virtual async Task OrderByRank_with_FullTextScore_on_nested_owned_type_wi
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["Owned"]["CustomNestedReference"]["AnotherDescription"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["Owned"]["CustomNestedReference"]["AnotherDescription"], "beaver", "dolphin")
 """);
     }
 
@@ -866,7 +866,7 @@ public virtual async Task OrderByRank_with_FullTextScore_on_property_without_ind
 """
 SELECT VALUE c
 FROM root c
-ORDER BY RANK FullTextScore(c["DescriptionNoIndex"], ["beaver","dolphin"])
+ORDER BY RANK FullTextScore(c["DescriptionNoIndex"], "beaver", "dolphin")
 """);
     }
 

test/EFCore.Cosmos.FunctionalTests/HybridSearchCosmosTest.cs

@@ -39,7 +39,7 @@ public virtual async Task Hybrid_search_vector_distance_and_FullTextScore_in_Ord
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK RRF(FullTextScore(c["Description"], ["beaver","otter"]), VectorDistance(c["SBytes"], @inputVector, false, {'distanceFunction':'dotproduct', 'dataType':'int8'}))
+ORDER BY RANK RRF(FullTextScore(c["Description"], "beaver", "otter"), VectorDistance(c["SBytes"], @inputVector, false, {'distanceFunction':'dotproduct', 'dataType':'int8'}))
 """);
     }
 
@@ -62,7 +62,7 @@ public virtual async Task Hybrid_search_vector_distance_and_FullTextScore_with_s
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK RRF(FullTextScore(c["Description"], ["beaver"]), VectorDistance(c["SBytes"], @inputVector, false, {'distanceFunction':'dotproduct', 'dataType':'int8'}))
+ORDER BY RANK RRF(FullTextScore(c["Description"], "beaver"), VectorDistance(c["SBytes"], @inputVector, false, {'distanceFunction':'dotproduct', 'dataType':'int8'}))
 """);
     }
 
@@ -84,7 +84,7 @@ public virtual async Task Hybrid_search_vector_distance_and_FullTextScore_in_Ord
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK RRF(FullTextScore(c["Owned"]["AnotherDescription"], ["beaver"]), VectorDistance(c["Owned"]["Singles"], @inputVector, false, {'distanceFunction':'cosine', 'dataType':'float32'}))
+ORDER BY RANK RRF(FullTextScore(c["Owned"]["AnotherDescription"], "beaver"), VectorDistance(c["Owned"]["Singles"], @inputVector, false, {'distanceFunction':'cosine', 'dataType':'float32'}))
 """);
     }
 
@@ -107,7 +107,7 @@ public virtual async Task Hybrid_search_vector_distance_and_FullTextScore_in_Ord
 
 SELECT VALUE c
 FROM root c
-ORDER BY RANK RRF(VectorDistance(c["Owned"]["Singles"], @inputVector, false, {'distanceFunction':'cosine', 'dataType':'float32'}), FullTextScore(c["Owned"]["AnotherDescription"], ["beaver","otter"]))
+ORDER BY RANK RRF(VectorDistance(c["Owned"]["Singles"], @inputVector, false, {'distanceFunction':'cosine', 'dataType':'float32'}), FullTextScore(c["Owned"]["AnotherDescription"], "beaver", "otter"))
 """);
     }