From 6ad7fde100b9910f88f5c7bb3855699e43dad9cf Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 11 Jun 2026 10:26:17 -0700
Subject: [PATCH] Implement snapshot_session_id validation and override in
 GetSnapshot.

PiperOrigin-RevId: 930605458
---
 tsl/profiler/protobuf/profiler_service.proto | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tsl/profiler/protobuf/profiler_service.proto b/tsl/profiler/protobuf/profiler_service.proto
index 36cf8ce67..89095ebe3 100644
--- a/tsl/profiler/protobuf/profiler_service.proto
+++ b/tsl/profiler/protobuf/profiler_service.proto
@@ -107,7 +107,12 @@ message TerminateRequest {
 
 message TerminateResponse {}
 
-message GetSnapshotRequest {}
+message GetSnapshotRequest {
+  // Optional. Suffix for the snapshot directory sandboxed inside
+  // repository_root. E.g. "window_001". If empty, defaults to emit_xpace=true
+  // behavior. timestamp folder.
+  string snapshot_session_id = 1;
+}
 
 message ContinuousProfilingResponse {}