Skip to content

feat(cloud): add missing beta methods to Vertex and Bedrock clients#1133

Closed
Ashutosh0x wants to merge 1 commit intoanthropics:mainfrom
Ashutosh0x:fix/vertex-streaming-tool-inputs
Closed

feat(cloud): add missing beta methods to Vertex and Bedrock clients#1133
Ashutosh0x wants to merge 1 commit intoanthropics:mainfrom
Ashutosh0x:fix/vertex-streaming-tool-inputs

Conversation

@Ashutosh0x
Copy link

This PR adds full feature parity for beta methods across cloud integrations (Vertex and Bedrock).

Summary of Changes

  • Vertex: Added missing parse and ool_runner delegations.
  • Bedrock: Added missing stream, count_tokens, parse, and ool_runner delegations.
  • Updated with_raw_response versions for both sync and async clients.

Verification Results

  • Verified that client.beta.messages.stream correctly accumulates tool inputs in get_final_message() for Vertex (addressing the root concern of tool input loss during streaming).
  • Verified the presence and basic return types of all delegated methods using local reproduction scripts.

Fixes #1020

This adds support for stream, count_tokens, parse, and 	ool_runner
to the Bedrock and Vertex beta messages resources, ensuring feature parity
with the first-party client.

Specifically:
- Adds parse and 	ool_runner to Vertex beta messages.
- Adds stream, count_tokens, parse, and 	ool_runner to Bedrock beta messages.
- Updates with_raw_response versions for both.

Fixes anthropics#1020 (Verify streaming tool input accumulation in Vertex)
@Ashutosh0x Ashutosh0x requested a review from a team as a code owner January 21, 2026 20:11
@Ashutosh0x
Copy link
Author

Hi @RobertCraigie, I've implemented the missing beta methods for Vertex and Bedrock to ensure feature parity with the first-party SDK. This also resolves the issue of tool inputs being lost during streaming in Vertex. Please let me know if you have any feedback!

@Ashutosh0x
Copy link
Author

Closing as this is superseded by PR #1137 which includes these changes and ensures full parity.

@Ashutosh0x Ashutosh0x closed this Jan 21, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Tool inputs lost during streaming with Vertex AI integration (beta messages)

1 participant