[ESQL] Fix SpatialDocValuesExtraction rule replacing TimeSeries agg node

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/optimizer/rules/physical/local/SpatialDocValuesExtraction.java

@@ -97,15 +97,7 @@ && allowedForDocValues(fieldAttribute, ctx.searchStats(), agg, foundAttributes))
                     }
                 }
                 if (changedAggregates) {
-                    exec = new AggregateExec(
-                        agg.source(),
-                        agg.child(),
-                        agg.groupings(),
-                        orderedAggregates,
-                        agg.getMode(),
-                        agg.intermediateAttributes(),
-                        agg.estimatedRowSize()
-                    );
+                    exec = agg.withAggregates(orderedAggregates);
                 }
             }
             if (exec instanceof EvalExec evalExec) {

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/physical/AggregateExec.java

@@ -120,6 +120,10 @@ public List<? extends NamedExpression> aggregates() {
         return aggregates;
     }
 
+    public AggregateExec withAggregates(List<? extends NamedExpression> newAggregates) {
+        return new AggregateExec(source(), child(), groupings, newAggregates, mode, intermediateAttributes, estimatedRowSize);
+    }
+
     public AggregateExec withMode(AggregatorMode newMode) {
         return new AggregateExec(source(), child(), groupings, aggregates, newMode, intermediateAttributes, estimatedRowSize);
     }

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/physical/TimeSeriesAggregateExec.java

@@ -97,6 +97,21 @@ public TimeSeriesAggregateExec replaceChild(PhysicalPlan newChild) {
         );
     }
 
+    @Override
+    public AggregateExec withAggregates(List<? extends NamedExpression> newAggregates) {
+        return new TimeSeriesAggregateExec(
+            source(),
+            child(),
+            groupings(),
+            newAggregates,
+            getMode(),
+            intermediateAttributes(),
+            estimatedRowSize(),
+            timeBucket
+        );
+    }
+
+    @Override
     public TimeSeriesAggregateExec withMode(AggregatorMode newMode) {
         return new TimeSeriesAggregateExec(
             source(),