From 7382d23e0877d6f45d1aedda5fc871fbff642761 Mon Sep 17 00:00:00 2001 From: Elizabeth Wu Date: Sun, 22 Dec 2024 19:20:16 +1100 Subject: [PATCH] Update _parse_proto_output function to include the entity_id as part of the metadata returned. (#648) --- .../bq_storage_vectorstores/featurestore.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py b/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py index 7f5eb388..5738c45b 100644 --- a/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py +++ b/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py @@ -406,6 +406,7 @@ def _parse_proto_output( documents.append( [ Document( + id=result.entity_id, page_content=content, metadata=metadata, ),