Fix stats reporting in scripting requests

1 files changed, 7 insertions, 3 deletions

diff --git a/ydb/core/kqp/session_actor/kqp_session_actor.cpp b/ydb/core/kqp/session_actor/kqp_session_actor.cpp
index 0b1f8175af6..5425ddd5319 100644
--- a/ydb/core/kqp/session_actor/kqp_session_actor.cpp
+++ b/ydb/core/kqp/session_actor/kqp_session_actor.cpp
@@ -206,8 +206,12 @@ struct TKqpQueryState {
         return cStats;
     }
 
-    bool CollectStatsDefined() const {
-        return GetStatsMode() != Ydb::Table::QueryStatsCollection::STATS_COLLECTION_NONE;
+    bool ReportStats() const {
+        return GetStatsMode() != Ydb::Table::QueryStatsCollection::STATS_COLLECTION_NONE
+            // always report stats for scripting subrequests
+            || GetType() == NKikimrKqp::QUERY_TYPE_AST_DML
+            || GetType() == NKikimrKqp::QUERY_TYPE_AST_SCAN
+        ;
     }
 
     bool HasPreparedQuery() const {
@@ -1449,7 +1453,7 @@ public:
                 [this]() { return this->QueryState->ExtractQueryText(); });
         }
 
-        if (QueryState->CollectStatsDefined()) {
+        if (QueryState->ReportStats()) {
             response->SetQueryPlan(SerializeAnalyzePlan(*stats));
             response->MutableQueryStats()->Swap(stats);
         }