Background

When using URL context as a tool in streaming, Google sends urlContextMetadata in the first chunk. And because of that, when we try to access it in the last chunk and add in the provider data, we get undefined.

Summary

Save the urlContextMetadata from the first chunk and adding only if we get a new urlContextMetadata

Verification

Tested manually with streamText in ai-core examples

Tasks

• Tests have been added / updated (for bug fixes / features)
• Documentation has been added / updated (for bug fixes / features)
• A patch changeset for relevant packages has been added (for bug fixes / features - run pnpm changeset in the project root)
• Formatting issues have been fixed (run pnpm prettier-fix in the project root)

Future Work

Currently, when using google_search and url_context tools combined, we are getting the result, but we are not getting the groundingChunks and the groundingSupports for the URL context in streaming.

This might be a bug from Google's streaming API.

Related Issues