Skip to content

fix(telemetry): Record the chunks of a stream cut short before its aggregate - #7938

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
copybara/1012119875
Open

copybara-service[bot] wants to merge 1 commit into
mainfrom
copybara/1012119875

Conversation

@copybara-service

Copy link
Copy Markdown

fix(telemetry): Record the chunks of a stream cut short before its aggregate

Since the inference span records a stream's partial=False aggregate rather
than its partial=True chunks, a stream that failed before the aggregate
arrived recorded no output message at all. Previously its chunks were
recorded.

When the model call ends with the span still open, the skipped chunks are now
recorded, as one output message.

Reported in #7815 (comment).

Co-authored-by: Max Ind maxind@google.com

…gregate

Since the inference span records a stream's `partial=False` aggregate rather
than its `partial=True` chunks, a stream that failed before the aggregate
arrived recorded no output message at all. Previously its chunks were
recorded.

When the model call ends with the span still open, the skipped chunks are now
recorded, as one output message.

Reported in #7815 (comment).

Co-authored-by: Max Ind <maxind@google.com>
PiperOrigin-RevId: 1012119875
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant