github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-02 03:15:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/vectorstores/bigquery_vector_search.py
View File

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