Update _parse_proto_output function to include the entity_id as part …

…of the metadata returned. (#648)
langchain-ai · Dec 22, 2024 · 7382d23 · 7382d23
1 parent b60d953
commit 7382d23
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,
                     ),