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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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]