Proto field additions for JiT flow

Author: rafal-hawrylak

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.

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 {

protos/jit.proto

@@ -126,6 +126,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.