Fix generative tests (#129833)

Author: luigidellaquila

muted-tests.yml

@@ -574,9 +574,6 @@ tests:
 - class: org.elasticsearch.search.query.RescoreKnnVectorQueryIT
   method: testKnnRetriever
   issue: https://github.com/elastic/elasticsearch/issues/129818
-- class: org.elasticsearch.xpack.esql.qa.single_node.GenerativeIT
-  method: test
-  issue: https://github.com/elastic/elasticsearch/issues/129819
 - class: org.elasticsearch.index.engine.ThreadPoolMergeExecutorServiceDiskSpaceTests
   method: testAbortingOrRunningMergeTaskHoldsUpBudget
   issue: https://github.com/elastic/elasticsearch/issues/129823

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/generative/EsqlQueryGenerator.java

@@ -184,7 +184,7 @@ public static String agg(List<Column> previousOutput) {
             };
         }
         // all types
-        name = randomName(previousOutput);
+        name = randomBoolean() ? randomStringField(previousOutput) : randomNumericOrDateField(previousOutput);
         if (name == null) {
             return "count(*)";
         }

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/generative/command/pipe/LookupJoinGenerator.java

@@ -55,8 +55,15 @@ public ValidationResult validateOutput(
         }
 
         // the -1 is for the additional RENAME, that could drop one column
-        if (previousColumns.size() - 1 > columns.size()) {
-            return new ValidationResult(false, "Expecting at least [" + previousColumns.size() + "] columns, got [" + columns.size() + "]");
+        int prevCols = previousColumns.size() - 1;
+
+        if (previousColumns.stream().anyMatch(x -> x.name().equals("<all-fields-projected>"))) {
+            // known bug https://github.com/elastic/elasticsearch/issues/121741
+            prevCols--;
+        }
+
+        if (prevCols > columns.size()) {
+            return new ValidationResult(false, "Expecting at least [" + prevCols + "] columns, got [" + columns.size() + "]");
         }
         return VALIDATION_OK;
     }