YQL-15435 Fix DictFromKeys constraints.

1 files changed, 6 insertions, 0 deletions

diff --git a/ydb/library/yql/core/yql_expr_constraint.cpp b/ydb/library/yql/core/yql_expr_constraint.cpp
index 025b503c4f..2969939aef 100644
--- a/ydb/library/yql/core/yql_expr_constraint.cpp
+++ b/ydb/library/yql/core/yql_expr_constraint.cpp
@@ -1733,6 +1733,9 @@ private:
     }
 
     TStatus DictWrap(const TExprNode::TPtr& input, TExprNode::TPtr& output, TExprContext& ctx) const {
+        const std::vector<std::string_view> k(1U, ctx.GetIndexAsString(0U));
+        input->AddConstraint(ctx.MakeConstraint<TUniqueConstraintNode>(k));
+        input->AddConstraint(ctx.MakeConstraint<TDistinctConstraintNode>(k));
         if (input->ChildrenSize() == 1) {
             return FromEmpty(input, output, ctx);
         }
@@ -1743,6 +1746,9 @@ private:
         if (input->Child(1)->ChildrenSize() == 0) {
             input->AddConstraint(ctx.MakeConstraint<TEmptyConstraintNode>());
         }
+        const std::vector<std::string_view> k(1U, ctx.GetIndexAsString(0U));
+        input->AddConstraint(ctx.MakeConstraint<TUniqueConstraintNode>(k));
+        input->AddConstraint(ctx.MakeConstraint<TDistinctConstraintNode>(k));
         return TStatus::Ok;
     }