enable test_insert for v2

1 files changed, 9 insertions, 3 deletions

diff --git a/ydb/library/yql/providers/s3/provider/yql_s3_datasink_type_ann.cpp b/ydb/library/yql/providers/s3/provider/yql_s3_datasink_type_ann.cpp
index 9000aaef8e..cabe474f27 100644
--- a/ydb/library/yql/providers/s3/provider/yql_s3_datasink_type_ann.cpp
+++ b/ydb/library/yql/providers/s3/provider/yql_s3_datasink_type_ann.cpp
@@ -124,13 +124,19 @@ private:
             }
         }
 
+        const auto format = tgt.Format();
+        const bool isSingleRowPerFileFormat = IsIn({TStringBuf("raw"), TStringBuf("json_list")}, format);
+
+        const TTypeAnnotationNode* listItemType = ctx.MakeType<TDataExprType>(EDataSlot::String);
+        if (!isSingleRowPerFileFormat) {
+            listItemType = ctx.MakeType<TOptionalExprType>(listItemType);
+        }
+
         input->SetTypeAnn(
             ctx.MakeType<TTupleExprType>(
                 TTypeAnnotationNode::TListType{
                     ctx.MakeType<TListExprType>(
-                        ctx.MakeType<TOptionalExprType>(
-                            ctx.MakeType<TDataExprType>(EDataSlot::String)
-                        )
+                        listItemType
                     )
                 }
             )