Skip to content

feat: Pass state from first bidi_stream_query request to async_create_session#6547

Merged
copybara-service[bot] merged 1 commit intomainfrom
copybara_894311208
Apr 6, 2026
Merged

feat: Pass state from first bidi_stream_query request to async_create_session#6547
copybara-service[bot] merged 1 commit intomainfrom
copybara_894311208

Conversation

@copybara-service
Copy link
Copy Markdown
Contributor

feat: Pass state from first bidi_stream_query request to async_create_session

The bidi_stream_query method in AdkApp now correctly extracts the "state" field from the initial request in the stream and forwards it to async_create_session when a new session is being created. This ensures that session state is properly initialized for bidi-directional streaming calls. A unit test is added to verify this behavior.

@copybara-service copybara-service bot requested a review from a team as a code owner April 4, 2026 00:07
@product-auto-label product-auto-label bot added size: s Pull request size is small. api: vertex-ai Issues related to the googleapis/python-aiplatform API. labels Apr 4, 2026
@copybara-service copybara-service bot force-pushed the copybara_894311208 branch from c269b70 to 2dc3698 Compare April 6, 2026 15:56
…_session

The `bidi_stream_query` method in AdkApp now correctly extracts the "state" field from the initial request in the stream and forwards it to `async_create_session` when a new session is being created. This ensures that session state is properly initialized for bidi-directional streaming calls. A unit test is added to verify this behavior.

PiperOrigin-RevId: 895368130
@copybara-service copybara-service bot force-pushed the copybara_894311208 branch from 2dc3698 to 37b5a0f Compare April 6, 2026 16:42
@copybara-service copybara-service bot merged commit 37b5a0f into main Apr 6, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

api: vertex-ai Issues related to the googleapis/python-aiplatform API. google-contributor size: s Pull request size is small.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant