From 9182be03219bfbe53077c378673be4cd31fbdb0b Mon Sep 17 00:00:00 2001
From: Sannidhya Chauhan <sannidhya@google.com>
Date: Thu, 11 Dec 2025 03:53:14 -0800
Subject: [PATCH] Add Python bindings for continuous profiling and snapshot
 retrieval.

PiperOrigin-RevId: 843151720
---
 tsl/profiler/protobuf/profiler_service.proto | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tsl/profiler/protobuf/profiler_service.proto b/tsl/profiler/protobuf/profiler_service.proto
index 2419d8290..0382e763b 100644
--- a/tsl/profiler/protobuf/profiler_service.proto
+++ b/tsl/profiler/protobuf/profiler_service.proto
@@ -17,6 +17,11 @@ service ProfilerService {
   rpc Terminate(TerminateRequest) returns (TerminateResponse) {}
   // Collects profiling data and returns user-friendly metrics.
   rpc Monitor(MonitorRequest) returns (MonitorResponse) {}
+  // Starts a continuous profiling session.
+  rpc StartContinuousProfiling(ProfileRequest)
+      returns (ContinuousProfilingResponse) {}
+  // Gets a snapshot of an ongoing profiling session.
+  rpc GetSnapshot(GetSnapshotRequest) returns (ProfileResponse) {}
 }
 
 message ToolRequestOptions {
@@ -99,6 +104,10 @@ message TerminateRequest {
 
 message TerminateResponse {}
 
+message GetSnapshotRequest {}
+
+message ContinuousProfilingResponse {}
+
 // Next-ID: 4
 message MonitorRequest {
   // Duration for which to profile between each update.