From 74f9af1d0c0d6fd862ceaeab671d88934071fdb6 Mon Sep 17 00:00:00 2001
From: Rafal Hawrylak <hawrylak@google.com>
Date: Wed, 27 May 2026 22:13:32 +0000
Subject: [PATCH] Proto field additions for JiT flow

---
 protos/core.proto      |  3 ++-
 protos/execution.proto | 12 ++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/protos/core.proto b/protos/core.proto
index 91657f3cd..3e3b451a2 100644
--- a/protos/core.proto
+++ b/protos/core.proto
@@ -51,7 +51,8 @@ message CompileConfig {
   // Project configuration overrides.
   ProjectConfig project_config_override = 3;
 
-  // Override compilation timeout settings.
+  // Override top-level compilation timeout (the single compile worker that
+  // runs once per invocation).
   int32 timeout_millis = 6;
 
   Extension extension = 10;
diff --git a/protos/execution.proto b/protos/execution.proto
index b7578d17b..0263f34f2 100644
--- a/protos/execution.proto
+++ b/protos/execution.proto
@@ -14,8 +14,16 @@ message RunConfig {
   bool include_dependencies = 3;
   bool include_dependents = 11;
   bool full_refresh = 2;
+
+  // Wall-clock deadline for the entire run. When it fires, in-flight actions
+  // (including any running JiT compilation worker) are cancelled.
   int32 timeout_millis = 7;
 
+  // Per-model JiT compilation worker timeout. Each action with JiT code gets
+  // its own fresh deadline; independent of timeout_millis. When unset, no
+  // per-model cap is applied and only timeout_millis bounds JiT work.
+  int32 jit_timeout_millis = 10;
+
   // For internal use only, will be removed at a later date.
   bool disable_set_metadata = 9;
 
@@ -118,6 +126,10 @@ message TaskResult {
   string error_message = 2;
   Timing timing = 3;
   ExecutionMetadata metadata = 4;
+  // SQL that would be executed. Populated only during --dry-run so that
+  // dry-run output can include the query — particularly for JiT actions
+  // whose SQL is only available after runtime compilation.
+  string compiled_sql = 5;
 }
 
 message TestResult {