ep: Fix pull examples query (#49610)

Oleksiy Syvokon created 1 month ago

Release Notes:

- N/A

Change summary

crates/edit_prediction_cli/src/pull_examples.rs | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)

Detailed changes

crates/edit_prediction_cli/src/pull_examples.rs

@@ -517,7 +517,7 @@ pub async fn fetch_rejected_examples_after(
                 AND rej.event_type = ?
                 AND req.event_properties:version = 'V3'
                 AND rej.event_properties:was_shown = true
-                AND req.event_properties:can_collect_data = true
+                AND req.event_properties:input:can_collect_data = true
                 AND req.time > TRY_TO_TIMESTAMP_NTZ(?)
                 AND (? IS NULL OR (
                     TRY_CAST(SPLIT_PART(req.event_properties:zed_version::string, '.', 2) AS INTEGER) > ?
@@ -671,7 +671,7 @@ pub async fn fetch_requested_examples_after(
             FROM events req
             WHERE req.event_type = ?
                 AND req.event_properties:version = 'V3'
-                AND req.event_properties:can_collect_data = true
+                AND req.event_properties:input:can_collect_data = true
                 AND req.time > TRY_TO_TIMESTAMP_NTZ(?)
                 AND (? IS NULL OR (
                     TRY_CAST(SPLIT_PART(req.event_properties:zed_version::string, '.', 2) AS INTEGER) > ?