community[patch]: BigQueryVectorSearch JSON type unsupported for metadatas (#18234)

This commit is contained in:
Ashley Xu
2024-02-28 08:19:53 -08:00
committed by GitHub
parent 92c34d4803
commit e3211c2b3d

View File

@@ -404,7 +404,8 @@ class BigQueryVectorSearch(VectorStore):
if self.metadata_field:
metadata = row[self.metadata_field]
if metadata:
metadata = json.loads(metadata)
if not isinstance(metadata, dict):
metadata = json.loads(metadata)
else:
metadata = {}
metadata["__id"] = row[self.doc_id_field]
@@ -544,7 +545,8 @@ class BigQueryVectorSearch(VectorStore):
for row in job:
metadata = row[self.metadata_field]
if metadata:
metadata = json.loads(metadata)
if not isinstance(metadata, dict):
metadata = json.loads(metadata)
else:
metadata = {}
metadata["__id"] = row[self.doc_id_field]