Skip to content

Commit bdcdc7f

Browse files
committed
check for prefix
1 parent 4f0b5d7 commit bdcdc7f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sky/client/sdk.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2106,7 +2106,9 @@ def stream_and_get(
21062106
stream_request_id: Optional[server_common.RequestId[
21072107
T]] = server_common.get_stream_request_id(response)
21082108
if request_id is not None and stream_request_id is not None:
2109-
assert request_id == stream_request_id
2109+
assert stream_request_id.startswith(request_id), (
2110+
f'Stream request ID {stream_request_id} does not start with '
2111+
f'request ID {request_id}')
21102112
if request_id is None:
21112113
request_id = stream_request_id
21122114
elif response.status_code != 200:

0 commit comments

Comments
 (0)