From e2d053d717efd1352c6218b0678c6a0331074194 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      |  9 ++++++++-
 protos/execution.proto | 11 +++++++++++
 protos/jit.proto       |  2 ++
 3 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/protos/core.proto b/protos/core.proto
index 91657f3cd..10481548e 100644
--- a/protos/core.proto
+++ b/protos/core.proto
@@ -51,9 +51,16 @@ 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). Does not bound per-model JiT compilation;
+  // see jit_timeout_millis.
   int32 timeout_millis = 6;
 
+  // Per-model JiT compilation worker timeout. Each model with JiT code gets
+  // its own fresh budget; this is per-model, not a shared budget across the
+  // compile.
+  int32 jit_timeout_millis = 13;
+
   Extension extension = 10;
 
   // Whether to emit compilation debug logs.
diff --git a/protos/execution.proto b/protos/execution.proto
index b7578d17b..5c84526eb 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 hit, all in-flight actions are
+  // cancelled. Does not bound per-model JiT compilation; see jit_timeout_millis.
   int32 timeout_millis = 7;
 
+  // Per-model JiT compilation worker timeout. Each action with JiT code gets
+  // its own fresh budget; this is per-model, not a shared budget across the
+  // run. Falls back to DEFAULT_JIT_COMPILATION_TIMEOUT_MILLIS when unset.
+  int32 jit_timeout_millis = 10;
+
   // For internal use only, will be removed at a later date.
   bool disable_set_metadata = 9;
 
@@ -46,6 +54,8 @@ message ExecutionAction {
 
   string jit_code = 12;
 
+  bool disabled = 13;
+
   reserved 1, 3, 7;
 }
 
@@ -118,6 +128,7 @@ message TaskResult {
   string error_message = 2;
   Timing timing = 3;
   ExecutionMetadata metadata = 4;
+  string compiled_sql = 5;
 }
 
 message TestResult {
diff --git a/protos/jit.proto b/protos/jit.proto
index 4807c2f28..d25441482 100644
--- a/protos/jit.proto
+++ b/protos/jit.proto
@@ -130,6 +130,8 @@ message JitCompilationRequest {
   repeated string file_paths = 6;
   // Current execution information for introspection.
   RunningExecutionData execution_data = 7;
+  // File name where the target is defined.
+  string file_name = 8;
 }
 
 // JiT compilation response.