YQL-19967: Properly handle optional args for curried function

Follows up e13229863b189b9ad804f0c6772204399430179e
Follows up 5b56b1b4416cf5b7abcae727a61e9402e10af067
commit_hash:c14af47d9a4960f42df3fad23da4b03721d27f7c

(cherry picked from commit 896fcc7)

Author: igormunkin

yql/essentials/minikql/comp_nodes/mkql_udf.cpp

@@ -322,17 +322,30 @@ IComputationNode* WrapUdf(TCallable& callable, const TComputationNodeFactoryCont
                                            << TruncateTypeDiff(diff)).c_str());
         }
 
+        const auto callableFuncType = AS_TYPE(TCallableType, funcInfo.FunctionType);
+        const auto callableNodeType = AS_TYPE(TCallableType, callable.GetType()->GetReturnType());
+
+        const auto callableType = runConfigNodeType->IsVoid()
+                                ? callableNodeType : callableFuncType;
+        const auto runConfigType = runConfigNodeType->IsVoid()
+                                 ? runConfigFuncType : runConfigNodeType;
+
         // If so, check the following invariants:
         // * The first argument of the head function in the sequence
         //   of the curried functions has to be the same as the
         //   run config type.
+        // * All other arguments of the head function in the sequence
+        //   of the curried function have to be optional.
         // * The type of the resulting callable has to be the same
         //   as the function type.
-        const auto firstArgType = runConfigNodeType->IsVoid()
-                                ? callable.GetType()->GetArgumentType(0)
-                                : funcInfo.FunctionType->GetArgumentType(0);
-        const auto runConfigType = runConfigNodeType->IsVoid()
-                                 ? runConfigFuncType : runConfigNodeType;
+        if (callableType->GetArgumentsCount() - callableType->GetOptionalArgumentsCount() != 1U) {
+            UdfTerminate((TStringBuilder() << pos
+                                           << " Udf Function '"
+                                           << funcName
+                                           << "' wrapper has more than one required argument: "
+                                           << PrintNode(callableType)).c_str());
+        }
+        const auto firstArgType = callableType->GetArgumentType(0);
         if (!runConfigType->IsSameType(*firstArgType)) {
             TString diff = TStringBuilder()
                 << "type mismatch, expected run config type: "
@@ -345,18 +358,18 @@ IComputationNode* WrapUdf(TCallable& callable, const TComputationNodeFactoryCont
                                            << "' "
                                            << TruncateTypeDiff(diff)).c_str());
         }
-        const auto callableFuncType = runConfigNodeType->IsVoid()
-                                    ? funcInfo.FunctionType
-                                    : funcInfo.FunctionType->GetReturnType();
-        const auto callableNodeType = runConfigNodeType->IsVoid()
-                                    ? AS_TYPE(TCallableType, callable.GetType()->GetReturnType())->GetReturnType()
-                                    : callable.GetType()->GetReturnType();
-        if (!callableNodeType->IsSameType(*callableFuncType)) {
+        const auto closureFuncType = runConfigNodeType->IsVoid()
+                                   ? callableFuncType
+                                   : AS_TYPE(TCallableType, callableFuncType)->GetReturnType();
+        const auto closureNodeType = runConfigNodeType->IsVoid()
+                                   ? AS_TYPE(TCallableType, callableNodeType)->GetReturnType()
+                                   : callableNodeType;
+        if (!closureNodeType->IsSameType(*closureFuncType)) {
             TString diff = TStringBuilder()
                 << "type mismatch, expected return type: "
-                << PrintNode(callableNodeType, true)
+                << PrintNode(closureNodeType, true)
                 << ", actual: "
-                << PrintNode(callableFuncType, true);
+                << PrintNode(closureFuncType, true);
             UdfTerminate((TStringBuilder() << pos
                                            << " Udf Function '"
                                            << funcName