gRPC Metadata not being passed in VertexRagDataServiceClient when polling for LRO #5286
Labels
api: vertex-ai
Issues related to the googleapis/python-aiplatform API.
Content-Length: 216139 | pFad | http://github.com/googleapis/python-aiplatform/issues/5286
18Fetched URL: http://github.com/googleapis/python-aiplatform/issues/5286
Alternative Proxies:
Environment details
google-cloud-aiplatform
version: 1.91.0Steps to reproduce
Code example
Issue
While initializing Vertex AI with custom metadata, necessary for routing API calls through a proxy, I noticed that although the RAG corpus is created successfully, it appears to get stuck when polling for the status of the creation request.
Reason
This seems to be due to how the Operation future is currently being constructed in
VertexRagDataServiceClient
. At the moment, the gRPC metadata isn't being passed into the following code:Resolution
A small tweak here resolves the issue nicely:
A similar fix would be needed for other
VertexRagDataServiceClient
operations likedelete_rag_corpus
,update_rag_corpus
, etc.I’d be happy to open a PR with the changes, if the changes seems fine.
Thanks!
The text was updated successfully, but these errors were encountered: