You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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:
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: